Cleanup

1 files changed, 5 insertions, 14 deletions

diff --git a/playlist_generator.py b/playlist_generator.py
index b31e472..a9cbab2 100644
--- a/playlist_generator.py
+++ b/playlist_generator.py
@@ -11,19 +11,6 @@ class PlaylistGenerator:
             lines = f.readlines()
             return [line.strip() for line in lines if len(line.strip()) > 0 and not line.startswith("#")]
 
-    # regex = r"v=([a-zA-Z-_\d]*)"
-    def extract_dict(self, urls: list) -> dict[dict[str]]:
-        infos = {}
-        ydl_opts = {}
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            for url in urls:
-                info = ydl.extract_info(url, download=False, )
-
-                # ℹ️ ydl.sanitize_info makes the info json-serializable
-                # print(json.dumps(ydl.sanitize_info(info)))
-                infos[info["id"]] = {"title": info["title"]}
-        return infos
-
     def extract(self, urls: list) -> list[dict[str]]:
         info = []
         ydl_opts = {}
@@ -69,7 +56,7 @@ class PlaylistGenerator:
         return field.replace('"', r'\"')
 
 
-if __name__ == "__main__":
+def main():
     parser = argparse.ArgumentParser()
     parser.add_argument("urls_file")
     parser.add_argument("output_file")
@@ -82,3 +69,7 @@ if __name__ == "__main__":
     js_playlist = generator.build_playlist_string(video_info)
     with open(args.output_file, 'w') as f:
         f.write(js_playlist)
+
+
+if __name__ == "__main__":
+    main()