import json import threading from datetime import datetime import pytz import config import api_client accounts_data = {} data_lock = threading.Lock() shanghai_tz = pytz.timezone(config.SHANGHAI_TZ_STR) def parse_accounts(): global accounts_data if not config.ACCOUNTS_JSON: print("错误: ACCOUNTS 环境变量未设置。") return False try: accounts_list = json.loads(config.ACCOUNTS_JSON) if not isinstance(accounts_list, list): print("错误: ACCOUNTS 环境变量必须是一个 JSON 数组。") return False temp_accounts_data = {} for acc in accounts_list: if isinstance(acc, dict) and "email" in acc and "password" in acc: temp_accounts_data[acc["email"]] = { "password": acc["password"], "token": None, "profile": None, "last_login_success": None, "last_profile_success": None, "last_login_attempt": None, "last_profile_attempt": None, "login_error": None, "profile_error": None, "cards_info": None, "last_card_info_success": None, "last_card_info_attempt": None, "card_info_error": None, } else: print(f"警告: ACCOUNTS 中的条目格式不正确: {acc}") with data_lock: accounts_data = temp_accounts_data print(f"成功加载 {len(accounts_data)} 个账户。") return True except json.JSONDecodeError: print("错误: ACCOUNTS 环境变量不是有效的 JSON 格式。") return False def login_and_store_token(email): global accounts_data with data_lock: account_info = accounts_data.get(email) if not account_info: print(f"错误: 账户 {email} 未找到。") return password = account_info["password"] print(f"[{datetime.now(shanghai_tz)}] 尝试为账户 {email} 登录...") token, error = api_client.api_login(email, password) with data_lock: accounts_data[email]["last_login_attempt"] = datetime.now(shanghai_tz) if token: accounts_data[email]["token"] = token accounts_data[email]["last_login_success"] = True accounts_data[email]["login_error"] = None print(f"[{datetime.now(shanghai_tz)}] 账户 {email} 登录成功。") else: accounts_data[email]["token"] = None accounts_data[email]["last_login_success"] = False accounts_data[email]["login_error"] = error print(f"[{datetime.now(shanghai_tz)}] 账户 {email} 登录失败: {error}") def fetch_and_store_profile(email): global accounts_data with data_lock: account_info = accounts_data.get(email) if not account_info: print(f"错误: 账户 {email} 未找到 (fetch_and_store_profile)。") return token = account_info.get("token") if not token: print(f"[{datetime.now(shanghai_tz)}] 账户 {email} 没有有效的 token，跳过获取 Profile。") with data_lock: accounts_data[email]["last_profile_attempt"] = datetime.now(shanghai_tz) accounts_data[email]["last_profile_success"] = False accounts_data[email]["profile_error"] = "无有效 Token" accounts_data[email]["profile"] = None accounts_data[email]["last_card_info_attempt"] = datetime.now(shanghai_tz) accounts_data[email]["cards_info"] = None accounts_data[email]["last_card_info_success"] = False accounts_data[email]["card_info_error"] = "因 Token 为空未尝试" return print(f"[{datetime.now(shanghai_tz)}] 尝试为账户 {email} 获取 Profile...") profile, error = api_client.get_api_profile(email, token) profile_fetch_successful_this_attempt = False with data_lock: accounts_data[email]["last_profile_attempt"] = datetime.now(shanghai_tz) if profile: accounts_data[email]["profile"] = profile accounts_data[email]["last_profile_success"] = True accounts_data[email]["profile_error"] = None profile_fetch_successful_this_attempt = True print(f"[{datetime.now(shanghai_tz)}] 账户 {email} 获取 Profile 成功。") else: accounts_data[email]["profile"] = None accounts_data[email]["last_profile_success"] = False accounts_data[email]["profile_error"] = error print(f"[{datetime.now(shanghai_tz)}] 账户 {email} 获取 Profile 失败: {error}") if error and ("token" in error.lower() or "auth" in error.lower() or "登录" in error.lower()): print(f"[{datetime.now(shanghai_tz)}] 账户 {email} 获取 Profile 失败，疑似 Token 失效，将尝试重新登录。") accounts_data[email]["token"] = None accounts_data[email]["last_card_info_attempt"] = datetime.now(shanghai_tz) accounts_data[email]["cards_info"] = None accounts_data[email]["last_card_info_success"] = False accounts_data[email]["card_info_error"] = "因 Profile 获取失败未尝试" return if profile_fetch_successful_this_attempt: current_token_for_cards = None with data_lock: if email in accounts_data: current_token_for_cards = accounts_data[email].get("token") else: print(f"[{datetime.now(shanghai_tz)}] 账户 {email} 在获取卡片信息前数据结构异常，跳过。") return if not current_token_for_cards: print(f"[{datetime.now(shanghai_tz)}] 账户 {email} 在获取卡片信息前发现 Token 已失效或被清除，跳过。") with data_lock: if email in accounts_data: accounts_data[email]["last_card_info_attempt"] = datetime.now(shanghai_tz) accounts_data[email]["cards_info"] = None accounts_data[email]["last_card_info_success"] = False accounts_data[email]["card_info_error"] = "因 Token 失效未尝试" return print(f"[{datetime.now(shanghai_tz)}] Profile 获取成功，继续为账户 {email} 获取卡片信息...") cards_info, card_error = api_client.get_api_card_info(email, current_token_for_cards) with data_lock: if email in accounts_data: accounts_data[email]["last_card_info_attempt"] = datetime.now(shanghai_tz) if cards_info: accounts_data[email]["cards_info"] = cards_info accounts_data[email]["last_card_info_success"] = True accounts_data[email]["card_info_error"] = None print(f"[{datetime.now(shanghai_tz)}] 账户 {email} 获取卡片信息成功。") elif card_error: accounts_data[email]["cards_info"] = None accounts_data[email]["last_card_info_success"] = False accounts_data[email]["card_info_error"] = card_error print(f"[{datetime.now(shanghai_tz)}] 账户 {email} 获取卡片信息失败: {card_error}") else: # No error, but no card_info (e.g. no cards for the account) accounts_data[email]["cards_info"] = None accounts_data[email]["last_card_info_success"] = True # Success in fetching, just no data accounts_data[email]["card_info_error"] = None print(f"[{datetime.now(shanghai_tz)}] 账户 {email} 获取卡片信息成功，但无卡片数据。") def get_accounts_data_for_display(): with data_lock: display_data = {} for email, data in accounts_data.items(): display_data[email] = { "profile": data.get("profile"), "last_login_success": data.get("last_login_success"), "last_profile_success": data.get("last_profile_success"), "last_login_attempt": data.get("last_login_attempt").isoformat() if data.get("last_login_attempt") else None, "last_profile_attempt": data.get("last_profile_attempt").isoformat() if data.get("last_profile_attempt") else None, "login_error": data.get("login_error"), "profile_error": data.get("profile_error"), "token_present": bool(data.get("token")), "cards_info": data.get("cards_info"), "last_card_info_success": data.get("last_card_info_success"), "last_card_info_attempt": data.get("last_card_info_attempt").isoformat() if data.get("last_card_info_attempt") else None, "card_info_error": data.get("card_info_error") } return display_data def get_all_account_emails(): with data_lock: return list(accounts_data.keys()) def get_accounts_from_config(): if not config.ACCOUNTS_JSON: print("错误: ACCOUNTS 环境变量未设置 (get_accounts_from_config)。") return [] try: accounts_list = json.loads(config.ACCOUNTS_JSON) if not isinstance(accounts_list, list): print("错误: ACCOUNTS 环境变量必须是一个 JSON 数组 (get_accounts_from_config)。") return [] valid_accounts = [] for acc in accounts_list: if isinstance(acc, dict) and "email" in acc and "password" in acc: valid_accounts.append({"email": acc["email"], "password": acc["password"]}) else: print(f"警告: ACCOUNTS 中的条目格式不正确，已跳过: {acc} (get_accounts_from_config)") return valid_accounts except json.JSONDecodeError: print("错误: ACCOUNTS 环境变量不是有效的 JSON 格式 (get_accounts_from_config)。") return []