From c75041a7d3e7b09b3f2d9a393bd3f2ab836c8bf1 Mon Sep 17 00:00:00 2001 From: litlighilit Date: Tue, 24 Jun 2025 17:14:18 +0800 Subject: [PATCH] refine: no use sed --- get_xha_user_pwd.py | 26 +++++++++++++++++--------- 1 file changed, 17 insertions(+), 9 deletions(-) diff --git a/get_xha_user_pwd.py b/get_xha_user_pwd.py index 0e1962e..66888b0 100644 --- a/get_xha_user_pwd.py +++ b/get_xha_user_pwd.py @@ -62,15 +62,16 @@ if check_connectivity(interface): with open(file, 'r') as f: user_pwd = [i.rstrip('\n\r').split() for i in f.readlines()] -user_pwd_error_or_not_unlimit_combo = [] +user_pwd_error_or_not_unlimit_combo_idx = set() # 检查是否付费 not_succ = True while not_succ: # 随机选择，防止前面有密码错误的用户卡死 - user, pwd = random.choice(user_pwd) - if user in user_pwd_error_or_not_unlimit_combo: + user_index = random.randrange(len(user_pwd)) + if user_index in user_pwd_error_or_not_unlimit_combo_idx: continue + (user, pwd) = user_pwd[user_index] url = f'http://192.168.101.201:801/eportal/portal/page/loadUserInfo?callback=dr1004&lang=zh-CN&program_index=ctshNw1713845951&page_index=V5fmKw1713845966&user_account={user}&wlan_user_ip=0.0.0.0&wlan_user_mac=000000000000&jsVersion=4.1&v=599&lang=zh' t = request_get_text(url, headers={ 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/105.0.0.0 Safari/537.36 Edg/105.0.1343.33', @@ -108,7 +109,7 @@ while not_succ: msg += f"接口 {interface}： " if '密码错误' in result['msg']: print(f"{user} {pwd} 密码错误") - user_pwd_error_or_not_unlimit_combo.append(user) + user_pwd_error_or_not_unlimit_combo_idx.add(user_index) elif '已经在线' in result['msg']: msg += "正常在线！" not_succ = False @@ -117,9 +118,16 @@ while not_succ: not_succ = False print(msg) else: - user_pwd_error_or_not_unlimit_combo.append(user) + user_pwd_error_or_not_unlimit_combo_idx.add(user_index) # 删除密码错误或不满足要求的用户 -if user_pwd_error_or_not_unlimit_combo: - sed_i_arg = '/^' + '\\|^'.join(user_pwd_error_or_not_unlimit_combo) + '/d' - cmd = ['sed','-i', sed_i_arg, file] - subprocess.run(cmd) \ No newline at end of file + +with open(file, 'w') as f: + for i in range(len(user_pwd)): + if i in user_pwd_error_or_not_unlimit_combo_idx: + user_pwd_error_or_not_unlimit_combo_idx.remove(i) + else: + t = user_pwd[i] + f.write(t[0]) + f.write(' ') + f.write(t[1]) + f.write('\n')