diff --git a/sheetsync/sheetsync/main.py b/sheetsync/sheetsync/main.py index cdfa66e..4a66ca9 100644 --- a/sheetsync/sheetsync/main.py +++ b/sheetsync/sheetsync/main.py @@ -481,8 +481,12 @@ def main(dbconnect, sync_configs, metrics_port=8005, backdoor_port=0, media_dir= client_secret=creds['client_secret'], refresh_token=creds['refresh_token'], ) - if config["type"] == "events": - middleware = SheetsEventsMiddleware( + if config["type"] in ("events", "archive"): + middleware_cls = { + "events": SheetsEventsMiddleware, + "archive": SheetsArchiveMiddleware, + }[config["type"]] + middleware = middleware_cls( client, config["sheet_id"], config["worksheets"], @@ -491,15 +495,6 @@ def main(dbconnect, sync_configs, metrics_port=8005, backdoor_port=0, media_dir= shifts, allocate_ids, ) - elif config["type"] == "archive": - middleware = SheetsArchiveMiddleware( - client, - config["sheet_id"], - config["worksheets"], - common.dateutil.parse(config["bustime_start"]), - config["edit_url"], - allocate_ids, - ) elif config["type"] == "playlists": middleware = SheetsPlaylistsMiddleware( client,