diff --git a/api/qinglong.py b/api/qinglong.py index 8f88a39..1b5ec9d 100644 --- a/api/qinglong.py +++ b/api/qinglong.py @@ -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}") \ No newline at end of file diff --git a/main.py b/main.py index 1798e8d..742894a 100644 --- a/main.py +++ b/main.py @@ -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()