From d84ff711ad5652468f4e884acc5abb53eec5f58a Mon Sep 17 00:00:00 2001 From: litlighilit Date: Tue, 24 Jun 2025 22:17:28 +0800 Subject: [PATCH] refact: merge common logic --- xha_auto_login.py | 80 ++++++++++++++++++++++++++++------------------- 1 file changed, 47 insertions(+), 33 deletions(-) diff --git a/xha_auto_login.py b/xha_auto_login.py index 6e9e6d7..b588824 100644 --- a/xha_auto_login.py +++ b/xha_auto_login.py @@ -28,7 +28,7 @@ class Loginer: log = print, ): self.interface = interface - self.device = device + self.device_str = str(device) self.log = log self.interface_def = interface is None def check_connectivity(self): @@ -54,42 +54,56 @@ class Loginer: return True except: return False + referrer = "http://192.168.101.201/" + header = { + "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", + "Referer": referrer + } + @staticmethod + def url(last, dr, v): + return "/eportal/portal/page/" + last + "?callback=dr" + dr + "&v=" + v + \ + "&lang=zh-CN&program_index=ctshNw1713845951&page_index=V5fmKw1713845966&wlan_user_ip=0.0.0.0&wlan_user_mac=000000000000&jsVersion=4.1" + + ip0 = "&wlan_user_ip=0.0.0.0" + urlloadUserInfo = referrer + url("loadUserInfo", "1004", "599") + ip0 + urlloadOnlineRecord = referrer + url("loadOnlineRecord", "1006", "2399") + "&wlan_user_ip=10.169.0.241&start_time=2010-01-01&end_time=2100-01-01&start_rn=1&end_rn=5" + urllogin = referrer + url("loadOnlineRecord", "1006", "2399") + ip0 + "&login_method=1&wlan_user_ipv6=&wlan_user_mac=000000000000&wlan_ac_ip=&wlan_ac_name=&terminal_type=1" + + @staticmethod + def json_from_drnnnn(t): + "drNNNN(`result`); <- e.g. t.replace('dr1006(', '').replace(')', '').replace(';', '')" + return json.loads(t[7:-2]) def login(self, user, pwd): interface = self.interface interface_def = self.interface_def - 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' + u = "&user_account=" + user + + url = self.urlloadUserInfo + u try: - 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', - 'Referer': 'http://192.168.101.201/' - }) + t = request_get_text(url, headers=self.header) except URLError as e: if e.reason.errno == ENETUNREACH: return LoginStatus.no_wifi raise - text = t.replace('dr1004(', '').replace(')', '').replace(';', '') - j = json.loads(text) + j = self.json_from_drnnnn(t) # 检查是否付费 if j['user_info']['user_state'] == "正常" and j['user_info']['available_flow'] in ("0MB", "无限制"): - url = f"http://192.168.101.201:801/eportal/portal/page/loadOnlineRecord?callback=dr1006&lang=zh-CN&program_index=ctshNw1713845951&page_index=V5fmKw1713845966&user_account={user}&wlan_user_ip=10.169.0.241&wlan_user_mac=000000000000&start_time=2010-01-01&end_time=2100-01-01&start_rn=1&end_rn=5&jsVersion=4.1&v=2399&lang=zh" + url = self.urlloadOnlineRecord + u # 获取在线设备 - 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', - 'Referer': 'http://192.168.101.201/' - }) - text = t.replace('dr1006(', '').replace(')', '').replace(';', '') - j1 = json.loads(text) - if int(j1['count']) == self.device: + t = request_get_text(url, headers=self.header) + j1 = self.json_from_drnnnn(t) + result = LoginStatus.unknown + if j1['count'] == self.device_str: # 判断密码是否正确 - url = f"https://xha.ouc.edu.cn:802/eportal/portal/login?callback=dr1003&login_method=1&user_account={user}&user_password={pwd}&wlan_user_ip=0.0.0.0&wlan_user_ipv6=&wlan_user_mac=000000000000&wlan_ac_ip=&wlan_ac_name=&jsVersion=4.1&terminal_type=1&lang=zh-cn&v=2425&lang=zh" + url = self.urllogin + u + "&user_password=" + pwd if interface_def: res = request_get_text(url) else: command = ["curl", url, "--interface", interface] res = subprocess.check_output(command, text=True) - result = json.loads(res[7:-2]) + j_res = self.json_from_drnnnn(res) """ dr1003({"result":0,"msg":"账号不存在","ret_code":1}); dr1003({"result":0,"msg":"密码错误","ret_code":1}); @@ -100,26 +114,26 @@ class Loginer: if not interface_def: self.log(f"使用接口 {interface} 进行请求") msg += f"接口 {interface}： " - if '密码错误' in result['msg']: - self.log(f"{user} {pwd} 密码错误") - self.log(msg) - return LoginStatus.bad_pwd - elif '已经在线' in result['msg']: + + j_msg = j_res['msg'] + if '密码错误' in j_msg: + msg += f"{user} {pwd} 密码错误" + result = LoginStatus.bad_pwd + elif '已经在线' in j_msg: msg += "正常在线！" - self.log(msg) - return LoginStatus.used_online - elif '认证成功' in result['msg']: + result = LoginStatus.used_online + elif '认证成功' in j_msg: msg += f"使用账号{user}登录成功!" - self.log(msg) - return LoginStatus.succ - return LoginStatus.unknown + result = LoginStatus.succ + self.log(msg) + return result else: return LoginStatus.not_unlimit def login_till_succ(self, user_pwd_gen, bad_user_callback=lambda u: None): - ''' + """ user_pwd_gen will be called multiply times until login succeeds. - ''' + """ not_succ = True while not_succ: (user, pwd) = user_pwd_gen() @@ -139,10 +153,10 @@ class Loginer: # 检测网络联通性 if self.check_connectivity(): - exit() + return with open(file, 'r') as f: - user_pwd = [i.rstrip('\n\r').split() for i in f.readlines()] + user_pwd = [i.rstrip("\r\n").split(' ') for i in f.readlines()] # login user_pwd_error_or_not_unlimit_combo_idx = set()