jellyfin-accounts/jellyfin_accounts/pw_reset.py

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 jellyfin_accounts import config, data_store
from jellyfin_accounts 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' 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:
                    id = jf.getUsers(reset['UserName'], public=False)['Id']
                    address = data_store.emails[id]
                    if address != '':
                        method = config['email']['method']
                        if method == 'mailgun':
                            email = Mailgun(address)
                        elif method == 'smtp':
                            email = Smtp(address)
                        if email.construct_reset(reset):
                            email.send()
                    else:
                        raise IndexError
                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()