piair/MsRewards
1
0
Fork 0
mirror of https://gitea.augustin64.fr/piair/MsRewards-Reborn.git synced 2025-08-09 15:06:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

ptite update

Browse Source
This commit is contained in:
piair
2023-04-25 21:56:29 +02:00
parent 710c272659
commit ed1e91b304
9 changed files with 72 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
modules/driver_tools.py
View File

@ -21,11 +21,23 @@ def rgpd_popup(driver) -> None:
# save webdriver cookies
def save_cookies(driver):
pickle.dump(driver.get_cookies(), open(f"{'/'.join(__file__.split('/')[:-2])}/user_data/cookies/{g._mail}.pkl", "wb"))
if g.dev:
f = open(f"{'/'.join(__file__.split('/')[:-2])}/user_data/cookies/{g._mail}_unsafe.pkl", "w")
for i in driver.get_cookies():
f.write(str(i) + "\n")
f.close()
else :
pickle.dump(driver.get_cookies(), open(f"{'/'.join(__file__.split('/')[:-2])}/user_data/cookies/{g._mail}.pkl", "wb"))
# load cookies previously saved to the driver
def load_cookies(driver):
cookies = pickle.load(open(f"{'/'.join(__file__.split('/')[:-2])}/user_data/cookies/{g._mail}.pkl", "rb"))
if g.dev:
f = open(f"{'/'.join(__file__.split('/')[:-2])}/user_data/cookies/{g._mail}_unsafe.pkl", "r")
lines = f.readlines()
f.close()
cookies = [literal_eval(x) for x in lines]
else :
cookies = pickle.load(open(f"{'/'.join(__file__.split('/')[:-2])}/user_data/cookies/{g._mail}.pkl", "rb"))
for cookie in cookies:
driver.add_cookie(cookie)