db_media mam_search arg_utils

Author: chapmanjacobd

xklb/db_media.py

@@ -382,6 +382,8 @@ def get_sibling_media(args, media):
                 new_media.append(d)
         media = new_media
 
+    # TODO: all-if>10, each-if=10
+
     return media
 
 

xklb/scratch/mam_search.py

@@ -23,6 +23,8 @@ def parse_args():
     parser.add_argument("--musicology", action="store_true")
     parser.add_argument("--radio", action="store_true")
 
+    parser.add_argument("--search-in", default='torrents')
+
     parser.add_argument("--cookie", required=True)
     parser.add_argument("--cookies", help="path to a Netscape formatted cookies file")
     parser.add_argument("--cookies-from-browser", metavar="BROWSER[+KEYRING][:PROFILE][::CONTAINER]")
@@ -36,7 +38,6 @@ def parse_args():
     log.info(objects.dict_filter_bool(args.__dict__))
     return args
 
-
 def get_page(args, query_data):
     import pandas as pd
 
@@ -100,7 +101,7 @@ def mam_search():
                 "description": args.description,
             },
             "searchType": "all",  # fl-VIP, fl, VIP, all
-            "searchIn": "torrents",
+            "searchIn": args.search_in,
             "browseFlagsHideVsShow": 0,
             "cat": [],
             "sortType": "dateDesc",

xklb/utils/arg_utils.py

@@ -1,3 +1,4 @@
+import operator
 import argparse, random, sys
 from ast import literal_eval
 from pathlib import Path
@@ -218,3 +219,17 @@ def override_config(parser, extractor_config, args):
     overridden_args = {k: v for k, v in args.__dict__.items() if default_args.get(k) != v}
     args_env = argparse.Namespace(**{**default_args, **extractor_config, **overridden_args})
     return args_env
+
+ops = {
+    '<': operator.lt,
+    '<=': operator.le,
+    '==': operator.eq,
+    '!=': operator.ne,
+    '>=': operator.ge,
+    '>': operator.gt
+}
+
+def cmp(arg1, op, arg2):
+    operation = ops.get(op)
+    return operation(arg1, arg2)
+