From 6d70fe03a39f49aeb7464cfae803e695cfdd5fbc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=81=BC=E7=9C=BC=E8=80=85?= <821986597@qq.com>
Date: Mon, 29 Apr 2024 10:24:06 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0JDcookies=E5=90=8E=EF=BC=8C?=
 =?UTF-8?q?=E8=87=AA=E5=8A=A8=E5=90=AF=E7=94=A8JDcookies?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 api/qinglong.py | 22 ++++++++++++++++++++++
 main.py         |  7 +++++++
 2 files changed, 29 insertions(+)

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()