From d33af063da404e097ba9b53b180a1ae9372778d0 Mon Sep 17 00:00:00 2001 From: ThePhaseless Date: Wed, 26 Feb 2025 11:02:42 +0100 Subject: [PATCH] faster title check --- src/endpoints.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) mode change 100644 => 100755 src/endpoints.py diff --git a/src/endpoints.py b/src/endpoints.py old mode 100644 new mode 100755 index d800239..f848c0b --- a/src/endpoints.py +++ b/src/endpoints.py @@ -57,10 +57,7 @@ def read_item(request: LinkRequest, sb: SeleniumDep) -> LinkResponse: sb.uc_gui_click_captcha() logger.info("Clicked captcha") - source_bs = sb.get_beautiful_soup() - title_tag = source_bs.title - - if title_tag and title_tag.string in CHALLENGE_TITLES: + if sb.get_title() in CHALLENGE_TITLES: save_screenshot(sb) raise HTTPException(status_code=500, detail="Could not bypass challenge") @@ -72,7 +69,7 @@ def read_item(request: LinkRequest, sb: SeleniumDep) -> LinkResponse: status=200, cookies=sb.get_cookies(), headers={}, - response=str(source_bs), + response=str(sb.get_beautiful_soup()), ), start_timestamp=start_time, )