From 1ee91422f481f2aba3ed71b94efba7d715a728ec Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=80=E4=BC=91?=
 <48150701+yixiu001@users.noreply.github.com>
Date: Tue, 9 Jul 2024 10:47:23 +0800
Subject: [PATCH] Update login_script.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

修改脚本，同时支持ct8与serv00的自动化保号
---
 login_script.py | 23 ++++++++++++++---------
 1 file changed, 14 insertions(+), 9 deletions(-)

diff --git a/login_script.py b/login_script.py
index d485f97e..5b197872 100644
--- a/login_script.py
+++ b/login_script.py
@@ -20,17 +20,20 @@ async def delay_time(ms):
 # 全局浏览器实例
 browser = None
 
-async def login(username, password, panelnum):
+async def login(username, password, panel):
     global browser
 
     page = None  # 确保 page 在任何情况下都被定义
-
+    serviceName = 'serv00'
+    if (panel.includes('ct8')) {
+        serviceName = 'ct8';
+    }
     try:
         if not browser:
             browser = await launch(headless=True, args=['--no-sandbox', '--disable-setuid-sandbox'])
 
         page = await browser.newPage()
-        url = f'https://panel{panelnum}.serv00.com/login/?next=/'
+        url = f'https://{panel}/login/?next=/'
         await page.goto(url)
 
         username_input = await page.querySelector('#id_username')
@@ -56,7 +59,8 @@ async def login(username, password, panelnum):
         return is_logged_in
 
     except Exception as e:
-        print(f'serv00账号 {username} 登录时出现错误: {e}')
+        print(f'{serviceName}账号 {username} 登录时出现错误: {e}')
+        
         return False
 
     finally:
@@ -71,23 +75,24 @@ async def main():
     for account in accounts:
         username = account['username']
         password = account['password']
-        panelnum = account['panelnum']
+        panel = account['panel']
 
-        is_logged_in = await login(username, password, panelnum)
+        is_logged_in = await login(username, password, panel)
 
         if is_logged_in:
             now_utc = format_to_iso(datetime.utcnow())
             now_beijing = format_to_iso(datetime.utcnow() + timedelta(hours=8))
-            success_message = f'serv00账号 {username} 于北京时间 {now_beijing}（UTC时间 {now_utc}）登录成功！'
+            success_message = f'{serviceName}账号 {username} 于北京时间 {now_beijing}（UTC时间 {now_utc}）登录成功！'
+
             print(success_message)
             send_telegram_message(success_message)
         else:
-            print(f'serv00账号 {username} 登录失败，请检查serv00账号和密码是否正确。')
+            print(f'{serviceName}账号 {username} 登录失败，请检查{serviceName}账号和密码是否正确。')
 
         delay = random.randint(1000, 8000)
         await delay_time(delay)
 
-    print('所有serv00账号登录完成！')
+    print('所有{serviceName}账号登录完成！')
 
 # 发送Telegram消息
 def send_telegram_message(message):