import time import json from watchdog.observers import Observer from watchdog.events import FileSystemEventHandler from jellyfin_accounts.email import Mailgun, Smtp from jellyfin_accounts.web_api import jf from __main__ import config from __main__ import email_log as log class Watcher: def __init__(self, dir): self.observer = Observer() self.dir = str(dir) def run(self): event_handler = Handler() self.observer.schedule(event_handler, self.dir, recursive=True) try: self.observer.start() except NotADirectoryError: log.error(f'Directory {self.dir} does not exist') try: while True: time.sleep(5) except: self.observer.stop() log.info('Watchdog stopped') class Handler(FileSystemEventHandler): @staticmethod def on_any_event(event): if event.is_directory: return None elif ((event.event_type == 'modified' or event.event_type == 'created') and 'passwordreset' in event.src_path): log.debug(f'Password reset file: {event.src_path}') time.sleep(1) with open(event.src_path, 'r') as f: reset = json.load(f) log.info(f'New password reset for {reset["UserName"]}') try: with open(config['files']['emails'], 'r') as f: emails = json.load(f) id = jf.getUsers(reset['UserName'], public=False)['Id'] address = emails[id] method = config['email']['method'] if method == 'mailgun': email = Mailgun(address) elif method == 'smtp': email = Smtp(address) if email.construct_reset(reset): email.send() except (FileNotFoundError, json.decoder.JSONDecodeError, IndexError) as e: err = f'{address}: Failed: ' + type(e).__name__ log.error(err) def start(): log.info(f'Monitoring {config["password_resets"]["watch_directory"]}') w = Watcher(config['password_resets']['watch_directory']) w.run()