nite/Byparr
1
Fork 0
mirror of https://github.com/ThePhaseless/Byparr.git synced 2025-03-15 09:50:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

save cookies between requests

Browse Source
This commit is contained in:
Thephaseless 2024-12-12 23:45:24 +00:00
parent e2114d7eb0
commit a31aa1e6d8
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
main.py
View File

@ -19,6 +19,8 @@ from src.utils.consts import LOG_LEVEL
app = FastAPI(debug=LOG_LEVEL == logging.DEBUG, log_level=LOG_LEVEL) app = FastAPI(debug=LOG_LEVEL == logging.DEBUG, log_level=LOG_LEVEL)
cookies = []
@app.get("/") @app.get("/")
def read_root(): def read_root():
@ -56,6 +58,10 @@ def read_item(request: LinkRequest) -> LinkResponse:
with SB(uc=True, locale_code="en", test=False, xvfb=True, ad_block=True) as sb: with SB(uc=True, locale_code="en", test=False, xvfb=True, ad_block=True) as sb:
try: try:
sb: BaseCase sb: BaseCase
global cookies # noqa: PLW0603
if cookies:
sb.uc_open_with_reconnect(request.url)
sb.add_cookies(cookies)
sb.uc_open_with_reconnect(request.url) sb.uc_open_with_reconnect(request.url)
source = sb.get_page_source() source = sb.get_page_source()
source_bs = BeautifulSoup(source, "html.parser") source_bs = BeautifulSoup(source, "html.parser")
@ -86,6 +92,7 @@ def read_item(request: LinkRequest) -> LinkResponse:
), ),
startTimestamp=start_time, startTimestamp=start_time,
) )
cookies = sb.get_cookies()
except Exception as e: except Exception as e:
logger.error(f"Error: {e}") logger.error(f"Error: {e}")
if sb.driver: if sb.driver: