fix automatic po grab

This commit is contained in:
insert 2025-01-25 09:39:48 -05:00
parent db27864f03
commit ffcf1b594c
Signed by: insert
GPG key ID: A70775C389ACF105

View file

@ -45,7 +45,6 @@ con = sqlite3.connect(os.getenv("QUEUE_PATH","queue.db"), check_same_thread=Fals
cur = con.cursor()
options = Options()
options.add_argument("--headless")
options.profile = webdriver.FirefoxProfile(os.getenv("PROFILE_PATH"))
driver = webdriver.Firefox(options=options)
print("Getting potoken from firefox...")
@ -58,6 +57,7 @@ for r in driver.requests:
if "https://www.youtube.com/youtubei/v1/player" in r.url:
potoken = json.loads(r.body)['serviceIntegrityDimensions']["poToken"]
break
driver.get("https://youtube.com")
cookiesstr = to_netscape_string(driver.get_cookies())
with open("cookies.txt", 'w') as f:
f.write(f'# Netscape HTTP Cookie File\n# https://curl.haxx.se/rfc/cookie_spec.html\n# This is a generated file! Do not edit.\n\n{cookiesstr}')