86 lines
2.3 KiB
Python
Executable File
86 lines
2.3 KiB
Python
Executable File
#!/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())
|