feat: Allow users to specify a cookie when querying youtube.

See #263.
2021-04-23 23:01:20 +08:00
parent b7232b8bae
commit 61748c199a
3 changed files with 11 additions and 1 deletions
--- a/util.py
+++ b/util.py
@ -321,7 +321,9 @@ def youtube_search(query):
    import json

    try:
-        r = requests.get("https://www.youtube.com/results", params={'search_query': query}, timeout=5)
+        cookie = json.loads(var.config.get('bot', 'youtube_query_cookie', default='{}'))
+        r = requests.get("https://www.youtube.com/results", cookie=cookie,
+                         params={'search_query': query}, timeout=5)
        result_json_match = re.findall(r">var ytInitialData = (.*?);</script>", r.text)

        if not len(result_json_match):