From 1cdf78b0a6bfef43804ebe70d15e44a61a685ed4 Mon Sep 17 00:00:00 2001 From: Christopher Usher Date: Tue, 5 Nov 2024 16:28:15 -0800 Subject: [PATCH] Fun with inheritance --- sheetsync/sheetsync/main.py | 17 ++++++----------- 1 file changed, 6 insertions(+), 11 deletions(-) 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,