done
This commit is contained in:
parent
26a585ee2c
commit
57cc50a44c
85
main.py
Executable file
85
main.py
Executable file
@ -0,0 +1,85 @@
|
||||
#!/usr/bin/python
|
||||
|
||||
import time
|
||||
import sys
|
||||
import os
|
||||
from watchdog.observers import Observer
|
||||
from watchdog.events import FileSystemEventHandler
|
||||
|
||||
from minio import Minio
|
||||
from minio.commonconfig import Tags
|
||||
from minio.error import S3Error
|
||||
|
||||
DIRECTORY_TO_WATCH = "/media/hd/memes/"
|
||||
client = None
|
||||
S3_BUCKET = None
|
||||
|
||||
def uploadFile(file_name):
|
||||
pass
|
||||
|
||||
class Watcher:
|
||||
def __init__(self):
|
||||
self.observer = Observer()
|
||||
|
||||
def run(self):
|
||||
event_handler = Handler()
|
||||
self.observer.schedule(event_handler, DIRECTORY_TO_WATCH, recursive=True)
|
||||
self.observer.start()
|
||||
try:
|
||||
while True:
|
||||
time.sleep(5)
|
||||
except:
|
||||
self.observer.stop()
|
||||
print("Error!")
|
||||
self.observer.join()
|
||||
|
||||
class Handler(FileSystemEventHandler):
|
||||
@staticmethod
|
||||
def on_any_event(event):
|
||||
if event.is_directory:
|
||||
None
|
||||
elif event.event_type == 'created':
|
||||
print("Made file " + event.src_path)
|
||||
tags = Tags.new_object_tags()
|
||||
tags["uploader"] = "Tyler"
|
||||
head, filename = os.path.split(event.src_path)
|
||||
client.fput_object(bucket_name=S3_BUCKET,
|
||||
object_name=filename,
|
||||
file_path=event.src_path,
|
||||
tags=tags,
|
||||
content_type=None)
|
||||
os.remove(event.src_path)
|
||||
print("Uploaded file!")
|
||||
|
||||
def main():
|
||||
global client
|
||||
global S3_BUCKET
|
||||
|
||||
if "S3_URL" not in os.environ:
|
||||
raise Exception("S3_URL is not set!")
|
||||
S3_URL = os.environ["S3_URL"]
|
||||
if "S3_UN" not in os.environ:
|
||||
raise Exception("S3_UN is not set!")
|
||||
S3_UN = os.environ["S3_UN"]
|
||||
if "S3_PW" not in os.environ:
|
||||
raise Exception("S3_PW is not set!")
|
||||
S3_PW = os.environ["S3_PW"]
|
||||
if "S3_BUCKET" not in os.environ:
|
||||
raise Exception("S3_BUCKET is not set!")
|
||||
S3_BUCKET = os.environ["S3_BUCKET"]
|
||||
|
||||
# override defaults
|
||||
if "S3_TLS" in os.environ:
|
||||
S3_TLS = os.environ["S3_TLS"].lower() in ("yes", "true", "1", "t")
|
||||
|
||||
client = Minio(S3_URL,
|
||||
access_key=S3_UN,
|
||||
secret_key=S3_PW,
|
||||
secure=S3_TLS)
|
||||
|
||||
w = Watcher()
|
||||
w.run()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
sys.exit(main())
|
Loading…
Reference in New Issue
Block a user