From 76abe4d8088892c2b1da2881468de0e24e973ab6 Mon Sep 17 00:00:00 2001 From: hlt1995 <93117676+hlt1995@users.noreply.github.com> Date: Tue, 4 Nov 2025 14:19:04 +0800 Subject: [PATCH] Update and rename Backup/xiaomiWallet_A.py to XiaomiWallet.py --- Backup/xiaomiWallet_A.py => XiaomiWallet.py | 35 +++++++++++++++------ 1 file changed, 26 insertions(+), 9 deletions(-) rename Backup/xiaomiWallet_A.py => XiaomiWallet.py (91%) diff --git a/Backup/xiaomiWallet_A.py b/XiaomiWallet.py similarity index 91% rename from Backup/xiaomiWallet_A.py rename to XiaomiWallet.py index afd1611..5eb0cb4 100644 --- a/Backup/xiaomiWallet_A.py +++ b/XiaomiWallet.py @@ -3,6 +3,7 @@ #每天两次视频任务 #环境变量 xmqb ，格式为：备注名&passToken&userId，多账号用@隔开 +#环境变量 xmqb_UA ，用于自定义User-Agent，不设置则使用默认UA #浏览器打开https://account.xiaomi.com/ 登陆 import os @@ -30,6 +31,19 @@ os.environ["PUSH_SWITCH"] = PUSH_SWITCH import notify urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning) +# 默认UA +DEFAULT_UA = 'Mozilla/5.0 (Linux; U; Android 14; zh-CN; M2012K11AC Build/UKQ1.230804.001; AppBundle/com.mipay.wallet; AppVersionName/6.89.1.5275.2323; AppVersionCode/20577595; MiuiVersion/stable-V816.0.13.0.UMNCNXM; DeviceId/alioth; NetworkType/WIFI; mix_version; WebViewVersion/118.0.0.0) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Mobile Safari/537.36 XiaoMi/MiuiBrowser/4.3' + +def get_custom_ua(): + """获取自定义UA，如果没有设置则使用默认UA""" + custom_ua = os.environ.get("xmqb_UA", "").strip() + if custom_ua: + print(f"✅ 读取到xmqb_UA变量，使用自定义UA: {custom_ua}") + return custom_ua + else: + print(f"✅ 未配置xmqb_UA变量，使用默认UA: {DEFAULT_UA[:50]}...") # 只打印前50个字符避免过长 + return DEFAULT_UA + def get_ck_from_env(): ck_str = os.environ.get("xmqb", "") @@ -61,11 +75,11 @@ def get_ck_from_env(): class RnlRequest: - def __init__(self, cookies: Union[str, dict]): + def __init__(self, cookies: Union[str, dict], user_agent: str): self.session = requests.Session() self._base_headers = { 'Host': 'm.jr.airstarfinance.net', - 'User-Agent': 'Mozilla/5.0 (Linux; U; Android 14; zh-CN; M2012K11AC Build/UKQ1.230804.001; AppBundle/com.mipay.wallet; AppVersionName/6.89.1.5275.2323; AppVersionCode/20577595; MiuiVersion/stable-V816.0.13.0.UMNCNXM; DeviceId/alioth; NetworkType/WIFI; mix_version; WebViewVersion/118.0.0.0) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Mobile Safari/537.36 XiaoMi/MiuiBrowser/4.3', + 'User-Agent': user_agent, # 使用传入的UA } self.update_cookies(cookies) @@ -138,7 +152,7 @@ class RnlRequest: class RNL: - def __init__(self, c): + def __init__(self, c, user_agent: str): self.t_id = None self.options = { "task_list": True, @@ -148,7 +162,7 @@ class RNL: "UserJoin": True, } self.activity_code = '2211-videoWelfare' - self.rr = RnlRequest(c) + self.rr = RnlRequest(c, user_agent) # 传入UA self.current_display_name = None # 存储当前处理的账号显示名 self.total_days = "未知" self.today_records = [] @@ -310,11 +324,11 @@ class RNL: return True -def get_xiaomi_cookies(pass_token, user_id): +def get_xiaomi_cookies(pass_token, user_id, user_agent: str): session = requests.Session() login_url = 'https://account.xiaomi.com/pass/serviceLogin?callback=https%3A%2F%2Fapi.jr.airstarfinance.net%2Fsts%3Fsign%3D1dbHuyAmee0NAZ2xsRw5vhdVQQ8%253D%26followup%3Dhttps%253A%252F%252Fm.jr.airstarfinance.net%252Fmp%252Fapi%252Flogin%253Ffrom%253Dmipay_indexicon_TVcard%2526deepLinkEnable%253Dfalse%2526requestUrl%253Dhttps%25253A%25252F%25252Fm.jr.airstarfinance.net%25252Fmp%25252Factivity%25252FvideoActivity%25253Ffrom%25253Dmipay_indexicon_TVcard%252526_noDarkMode%25253Dtrue%252526_transparentNaviBar%25253Dtrue%252526cUserId%25253Dusyxgr5xjumiQLUoAKTOgvi858Q%252526_statusBarHeight%25253D137&sid=jrairstar&_group=DEFAULT&_snsNone=true&_loginType=ticket' headers = { - 'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/135.0.0.0 Safari/537.36 Edg/135.0.0.0', + 'user-agent': user_agent, # 使用传入的UA 'cookie': f'passToken={pass_token}; userId={user_id};' } @@ -343,6 +357,9 @@ def generate_notification(display_name, rnl_instance): if __name__ == "__main__": + # 获取自定义UA + CURRENT_UA = get_custom_ua() + ORIGINAL_COOKIES = get_ck_from_env() if not ORIGINAL_COOKIES: print("❌ 未获取到账号CK，请检查青龙环境变量 xmqb 是否配置正确！") @@ -358,7 +375,7 @@ if __name__ == "__main__": print(f"\n>>>>>>>>>> 正在处理账号 {display_name} <<<<<<<<<<") # 获取Cookie - cookie_result = get_xiaomi_cookies(account['passToken'], user_id) + cookie_result = get_xiaomi_cookies(account['passToken'], user_id, CURRENT_UA) # 处理返回结果 if isinstance(cookie_result, tuple): @@ -367,8 +384,8 @@ if __name__ == "__main__": new_cookie = cookie_result error = None - # 创建RNL实例并设置当前显示名 - rnl = RNL(new_cookie) + # 创建RNL实例并设置当前显示名，传入UA + rnl = RNL(new_cookie, CURRENT_UA) rnl.current_display_name = display_name if error: