diff --git a/src/check_results.py b/src/check_results.py index f64c4a1..c23423d 100644 --- a/src/check_results.py +++ b/src/check_results.py @@ -38,11 +38,9 @@ class ResultChecker: def __init__(self): self.logger = logging.getLogger("Worker") - def load_page(self, playwright, proxy, url, headless) -> Union[str, None]: + def load_page(self, playwright, proxy, url, device, headless) -> Union[str, None]: try: self.browser = playwright.webkit.launch(headless=headless, timeout=90000, proxy=proxy) - device = random.choice(params.DEVICES) - self.logger.info("模拟设备: " + device) pixel_2 = self.tls.playwright.devices[device] context = self.browser.new_context(**pixel_2, locale='fr-FR') self.page = context.new_page() @@ -62,16 +60,16 @@ class ResultChecker: return None def run(self, reserve_pojo: ReserveResultPojo, firestore_collection, headless=False, need_send_email=False): - print("Launched worker in ", threading.current_thread().name) url = reserve_pojo.url print("url is " + url) content = None proxy = params.get_proxy(ProxyType.BRIGHT_DATA) + device = random.choice(params.DEVICES) + self.logger.info("模拟设备: " + device) while content is None: - content = self.load_page(self.tls.playwright, proxy, url, headless) + content = self.load_page(self.tls.playwright, proxy, url, device, headless) proxy = params.get_proxy(ProxyType.BRIGHT_DATA) print(content) - self.browser.close() print("Stopped worker in ", threading.current_thread().name) if SORRY_SENTENCE_FR in content: print("status is REFUSED") @@ -99,6 +97,7 @@ class ResultChecker: reserve_pojo.accepted = status firestore_collection.document(reserve_pojo.id).update({u'accepted': status.name}) MONGO_STORE_MANAGER.update_reserve_result(reserve_pojo.id, status) + self.browser.close() def check_results(headless=False):