From 95f604cc62a96b31e2ac0797e447e09b9158efb3 Mon Sep 17 00:00:00 2001 From: Pinapelz Date: Thu, 4 May 2023 16:33:28 -0700 Subject: Implemented ability to handle multiple Holodex Organizations --- main.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) (limited to 'main.py') diff --git a/main.py b/main.py index ee2f8b6..8f11942 100644 --- a/main.py +++ b/main.py @@ -103,11 +103,15 @@ def main(mode=0): server.create_table(HISTORICAL_TABLE_NAME, TABLE_COLUMNS) server.create_table(DAY_DIFF_TABLE_NAME, DIFF_COLUMNS) excluded_channels = [] + data = [] if mode == 0: - print("Running Holodex API Data Collection") - hldex = HolodexAPI(fs.get_api_key("holodex_api_key"), member_count=ORG_MEMBER_COUNT, organization=HOLODEX_ORG) - data = hldex.get_data_all_channels() - excluded_channels += hldex.get_exclude_channels() + holodex_organizations = HOLODEX_ORG.split(",") + print("Running Holodex API Data Collection\n"+"Found "+str(len(holodex_organizations))+" organizations") + for org in holodex_organizations: + hldex = HolodexAPI(fs.get_api_key("holodex_api_key"), member_count = ORG_MEMBER_COUNT, organization = org) + data += hldex.get_data_all_channels() + excluded_channels += hldex.get_exclude_channels() + elif mode == 1: print("Running YouTube API Data Collection") ytapi = YouTubeAPI(fs.get_api_key("youtube_api_key")) @@ -148,7 +152,7 @@ def generate_channel_files(): if __name__ == "__main__": - MODE = 1 + MODE = 0 if len(sys.argv) > 1: MODE = int(sys.argv[1]) ROOT_STORAGE_PATH = sys.argv[2] -- cgit v1.2.3