diff --git a/export.py b/export.py index 6b87618..ef8f0aa 100644 --- a/export.py +++ b/export.py @@ -138,4 +138,55 @@ def subprrun(jobs, mysession): jobs.task_done() - return True \ No newline at end of file + return True + +if __name__ == "__main__": + from os import environ, mkdir + from os.path import isfile + from json import loads + #HSID, SSID, SID cookies required + if "HSID" in environ.keys() and "SSID" in environ.keys() and "SID" in environ.keys(): + cookies = {"HSID": environ["HSID"], "SSID": environ["SSID"], "SID": environ["SID"]} + elif isfile("config.json"): + cookies = loads(open("config.json").read()) + else: + print("HSID, SSID, and SID cookies from youtube.com are required. Specify in config.json or as environment variables.") + assert False + if not (cookies["HSID"] and cookies["SSID"] and cookies["SID"]): + print("HSID, SSID, and SID cookies from youtube.com are required. Specify in config.json or as environment variables.") + assert False + + mysession = requests.session() + mysession.headers.update({"cookie": "HSID="+cookies["HSID"]+"; SSID="+cookies["SSID"]+"; SID="+cookies["SID"], "Accept-Language": "en-US",}) + del cookies + from sys import argv + from queue import Queue + langs = ['ab', 'aa', 'af', 'sq', 'ase', 'am', 'ar', 'arc', 'hy', 'as', 'ay', 'az', 'bn', 'ba', 'eu', 'be', 'bh', 'bi', 'bs', 'br', + 'bg', 'yue', 'yue-HK', 'ca', 'chr', 'zh-CN', 'zh-HK', 'zh-Hans', 'zh-SG', 'zh-TW', 'zh-Hant', 'cho', 'co', 'hr', 'cs', 'da', 'nl', + 'nl-BE', 'nl-NL', 'dz', 'en', 'en-CA', 'en-IN', 'en-IE', 'en-GB', 'en-US', 'eo', 'et', 'fo', 'fj', 'fil', 'fi', 'fr', 'fr-BE', + 'fr-CA', 'fr-FR', 'fr-CH', 'ff', 'gl', 'ka', 'de', 'de-AT', 'de-DE', 'de-CH', 'el', 'kl', 'gn', 'gu', 'ht', 'hak', 'hak-TW', 'ha', + 'iw', 'hi', 'hi-Latn', 'ho', 'hu', 'is', 'ig', 'id', 'ia', 'ie', 'iu', 'ik', 'ga', 'it', 'ja', 'jv', 'kn', 'ks', 'kk', 'km', 'rw', + 'tlh', 'ko', 'ku', 'ky', 'lo', 'la', 'lv', 'ln', 'lt', 'lb', 'mk', 'mg', 'ms', 'ml', 'mt', 'mni', 'mi', 'mr', 'mas', 'nan', + 'nan-TW', 'lus', 'mo', 'mn', 'my', 'na', 'nv', 'ne', 'no', 'oc', 'or', 'om', 'ps', 'fa', 'fa-AF', 'fa-IR', 'pl', 'pt', 'pt-BR', + 'pt-PT', 'pa', 'qu', 'ro', 'rm', 'rn', 'ru', 'ru-Latn', 'sm', 'sg', 'sa', 'sc', 'gd', 'sr', 'sr-Cyrl', 'sr-Latn', 'sh', 'sdp', 'sn', + 'scn', 'sd', 'si', 'sk', 'sl', 'so', 'st', 'es', 'es-419', 'es-MX', 'es-ES', 'es-US', 'su', 'sw', 'ss', 'sv', 'tl', 'tg', 'ta', + 'tt', 'te', 'th', 'bo', 'ti', 'tpi', 'to', 'ts', 'tn', 'tr', 'tk', 'tw', 'uk', 'ur', 'uz', 'vi', 'vo', 'vor', 'cy', 'fy', 'wo', + 'xh', 'yi', 'yo', 'zu'] + vidl = argv + vidl.pop(0) + + try: + mkdir("out") + except: + pass + + jobs = Queue() + for video in vidl: + try: + mkdir("out/"+video.strip()) + except: + pass + for lang in langs: + jobs.put((lang, video)) + + subprrun(jobs, mysession) \ No newline at end of file