From e318a6aaffe3361ad64681c23e208194cda5fb65 Mon Sep 17 00:00:00 2001 From: Jesse Bannon Date: Fri, 10 Sep 2021 21:30:55 -0700 Subject: [PATCH] black, add argparse --- requirements.txt | 1 + ytdl_subscribe/main.py | 2 ++ ytdl_subscribe/subscriptions.py | 8 +++++--- 3 files changed, 8 insertions(+), 3 deletions(-) diff --git a/requirements.txt b/requirements.txt index 7c447aba..581a17f5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,5 @@ black +argparse==1.4.0 dicttoxml==1.7.4 mergedeep==1.3.4 music-tag==0.4.3 diff --git a/ytdl_subscribe/main.py b/ytdl_subscribe/main.py index ab019330..4a8b39c6 100644 --- a/ytdl_subscribe/main.py +++ b/ytdl_subscribe/main.py @@ -1,5 +1,7 @@ from ytdl_subscribe.parse import parse_subscriptions +parser = argparse.ArgumentParser(description="Process some integers.") + if __name__ == "__main__": # execute only if run as a script diff --git a/ytdl_subscribe/subscriptions.py b/ytdl_subscribe/subscriptions.py index 82a30501..df1f686b 100644 --- a/ytdl_subscribe/subscriptions.py +++ b/ytdl_subscribe/subscriptions.py @@ -247,7 +247,7 @@ class YoutubeSubscription(Subscription): return entry def extract_info(self): - playlist_id = self.options['playlist_id'] + playlist_id = self.options["playlist_id"] url = f"https://youtube.com/playlist?list={playlist_id}" track_ytdl_opts = { "download_archive": self.WORKING_DIRECTORY + "/ytdl-download-archive.txt", @@ -263,8 +263,10 @@ class YoutubeSubscription(Subscription): # Load the entries from info.json, ignore the playlist entry entries = [] for file_name in os.listdir(self.WORKING_DIRECTORY): - if file_name.endswith('.info.json') and not file_name.startswith(playlist_id): - with open(self.WORKING_DIRECTORY + '/' + file_name, 'r') as f: + if file_name.endswith(".info.json") and not file_name.startswith( + playlist_id + ): + with open(self.WORKING_DIRECTORY + "/" + file_name, "r") as f: entries.append(json.load(f)) entries = [self.parse_entry(e) for e in entries]