更新JDcookies后,自动启用JDcookies

This commit is contained in:
灼眼者 2024-04-29 10:24:06 +08:00
parent db901f3446
commit 6d70fe03a3
2 changed files with 29 additions and 0 deletions

View File

@ -9,6 +9,8 @@ logger.add("main.log", format="{time:YYYY-MM-DD HH:mm:ss} {level} {message}", le
class QlUri(Enum):
user_login = "api/user/login"
envs = "api/envs"
envs_enable = "api/envs/enable"
envs_disable = "api/envs/disable"
class QlApi(object):
@ -60,3 +62,23 @@ class QlApi(object):
return data
else:
logger.error(f"Set Envs failed. Status code: {response.status}")
async def envs_enable(self, data: bytes):
async with aiohttp.ClientSession() as session:
async with session.put(f"{self.url}/{QlUri.envs_enable.value}", data=data, headers=self.headers) as response:
if response.status == 200:
logger.info("enable Envs successful. Token obtained.")
data = await response.json()
return data
else:
logger.error(f"enable Envs failed. Status code: {response.status}")
async def envs_disable(self, data: bytes):
async with aiohttp.ClientSession() as session:
async with session.put(f"{self.url}/{QlUri.envs_disable.value}", data=data, headers=self.headers) as response:
if response.status == 200:
logger.info("disable Envs successful. Token obtained.")
data = await response.json()
return data
else:
logger.error(f"disable Envs failed. Status code: {response.status}")

View File

@ -130,6 +130,13 @@ async def main():
else:
logger.error(f"{user} update fail")
data = bytes(f"[{user_info['id']}]", 'utf-8')
response = await qlapi.envs_enable(data=data)
if response['code'] == 200:
logger.info(f"{user} envs_enable sucess")
else:
logger.error(f"{user} envs_enable fail")
except Exception as e:
traceback.print_exc()