diff --git a/src/cluster/rpa.py b/src/cluster/rpa.py index 6e29968..db77cae 100644 --- a/src/cluster/rpa.py +++ b/src/cluster/rpa.py @@ -163,6 +163,8 @@ async def get_promotion_list_text(page: Page, result: LiveStreamResult, start_polling = time.time() last_status_check = start_polling + result.start_time = int(time.time() * 1000) + while time.time() - start_polling < max_duration: # Check live status every 60 seconds @@ -450,7 +452,7 @@ with rpa_image.imports(): return result.dict() # Return as dict for JSON compatibility - async def goto_live(browser, data, result, webcast_id): + async def goto_live(browser, data: LiveStreamProductWatchRequest, result: LiveStreamResult, webcast_id): context = await browser.new_context() await context.add_cookies(cookies) logger.info("Loaded cookies") @@ -462,6 +464,7 @@ with rpa_image.imports(): result.title = await page.title() logger.info(f"Retrieved title: {result.title}") live_status = await get_live_status_by_web_page(page, webcast_id) + if live_status == LIVE_END: result.msg = LIVE_END else: