From 69e1ed4485d636879d4eaac65532809a6672576a Mon Sep 17 00:00:00 2001 From: Fitti Date: Mon, 1 Mar 2021 10:12:15 +0100 Subject: [PATCH] Add category filter --- clipper.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/clipper.py b/clipper.py index e45b09f..da40fd2 100644 --- a/clipper.py +++ b/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")