zayac/tools-pyqt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

引入重试机制

Browse Source
This commit is contained in:
zayac 2024-04-06 12:07:11 +08:00
parent 23ac228925
commit 907692c5e8
1 changed files with 20 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/core/login.py
View File

@ -39,11 +39,11 @@ def perform_login(playwright, account: Account) -> Account:
page = context.new_page() page = context.new_page()
page.goto(account.url) page.goto(account.url)
fill_login_form(page, account) fill_login_form(page, account)
if handle_captcha(page,account.type): if handle_captcha(page, account.type, retry_count=0):
account.headers = capture_request_headers(page, account) account.headers = capture_request_headers(page, account)
logger.info('登录成功') logger.info('登录成功')
else: else:
logger.error('登录失败或验证码处理失败') logger.error('登录失败了')
close_resources(page, context, browser) close_resources(page, context, browser)
return account return account
@ -59,14 +59,17 @@ def fill_login_form(page, account: Account):
logger.info(f'{account.name}登录ing...........') logger.info(f'{account.name}登录ing...........')
def handle_captcha(page, account_type) -> bool: def handle_captcha(page, account_type, retry_count) -> bool:
if retry_count < 3:
try: try:
validate_code = page.locator('.geetest_box') validate_code = page.locator('.geetest_box')
validate_code.wait_for(state='visible') validate_code.wait_for(state='visible')
time.sleep(1) time.sleep(1)
return process_validate_code(page, validate_code, account_type) return process_validate_code(page, validate_code, account_type)
except TimeoutError: except TimeoutError:
logger.error('超时了') retry_count += 1
logger.error(f'验证码识别失败,正在重试:{retry_count}次')
return handle_captcha(page, account_type, retry_count)
return False return False
@ -99,6 +102,9 @@ def process_validate_code(page,validate_code, account_type):
except ValueError: except ValueError:
logger.error(f"获取移动距离失败,实际返回内容为：{res}") logger.error(f"获取移动距离失败,实际返回内容为：{res}")
return False return False
except TimeoutError:
logger.debug("验证码滑动失败")
return process_validate_code(page, validate_code, account_type)
else: else:
res = base64_api(img=img) res = base64_api(img=img)
if '|' in res: if '|' in res:
@ -157,7 +163,6 @@ def persistence(account: Account, headers: dict):
session.commit() session.commit()
logger.info(f'Headers persisted for account {account.name}') logger.info(f'Headers persisted for account {account.name}')
# if __name__ == '__main__': # if __name__ == '__main__':
# with open('C:\\Users\\Administrator\\Desktop\\Snipaste_2024-03-26_21-46-35.png', 'rb') as f: # with open('C:\\Users\\Administrator\\Desktop\\Snipaste_2024-03-26_21-46-35.png', 'rb') as f:
# base64_data = base64.b64encode(f.read()) # base64_data = base64.b64encode(f.read())