Add category filter

clipper.py

@@ -45,7 +45,7 @@ def get_gdrive_files(credentials, clips, staging):
     return files, staging_folder, drive
 
 
-def get_urls(twitch, start, end, b_id, pagination=None, clipper=None):
+def get_urls(twitch, start, end, b_id, pagination=None, clipper=None, category=None):
     clips_list = []
     global game_ids
 
@@ -71,6 +71,8 @@ def get_urls(twitch, start, end, b_id, pagination=None, clipper=None):
         title += " _ " + creator + " _ " + clip["id"]
         if clipper and clipper.lower() != creator.lower():
             pass
+        elif category and category.lower() != game.lower():
+            pass
         else:
             clips_list.append([title, clip_url])
 
@@ -120,6 +122,13 @@ def dl_progress(count, block_size, total_size):
                         help="only download clips made by this person",
                         metavar="username",
                         type=str)
+    parser.add_argument("--category",
+                        help="only download clips from this category/game " +
+                        "(some non-game categories like don't get " +
+                        "reported by the API, so type \"NOGAME\" " +
+                        "for these if you notice they're missing)",
+                        metavar="game",
+                        type=str)
     args = parser.parse_args()
 
     filepath = realpath(__file__)
@@ -216,7 +225,8 @@ def dl_progress(count, block_size, total_size):
                                             end=start + timedelta(days=1),
                                             b_id=b_id,
                                             pagination=pagination,
-                                            clipper=args.clipper)
+                                            clipper=args.clipper,
+                                            category=args.category)
             all_urls += new_urls
             print(f"Clips created on {datestring}: " + str(len(all_urls)),
                   end="\r")