From f932c10606db290ef8a33e8fb52e1ea6fde5430a Mon Sep 17 00:00:00 2001 From: Lei PAN Date: Fri, 26 Jan 2024 18:56:06 +0100 Subject: [PATCH] support ip_country field --- workers/cookie_generator.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/workers/cookie_generator.py b/workers/cookie_generator.py index 134ac71..65c3e71 100644 --- a/workers/cookie_generator.py +++ b/workers/cookie_generator.py @@ -8,6 +8,7 @@ import pika from queue_message.CookiesPublisher import CookiesPublisher, QUEUE_HOST, credentials, REQUEST_DATA_QUEUE, \ REQUEST_DATA_QUEUE_TEST from workers.captcha_result_getter import CaptchaResultGetter +from workers.proxies_constants import PROXY_LIST_FR DVM_LIST = [2, 3, 4, 6] @@ -36,8 +37,10 @@ class CookiesGenerator(threading.Thread): def on_message(self, ch, method, properties, body): print(f" [x] Received {body}") - _message_count = self.cookiesPublisher.message_count() - print("message count in queue is {}".format(_message_count)) + _cached_message_count = self.cookiesPublisher.message_count() + _valid_message_count = self.message_count() + print("message count in cache is {}".format(_cached_message_count)) + print("message count in current queue is {}".format(_valid_message_count)) # generate 10 cookies to new_queue captchaResultGetter = CaptchaResultGetter() @@ -45,7 +48,7 @@ class CookiesGenerator(threading.Thread): _received_cookies = body.decode("UTF-8") dvm = random.choice(DVM_LIST) hc = random.choice(DVM_LIST) - if _message_count < 10: + if _cached_message_count < 10 and _valid_message_count > 40: _proxy_to_use = random.choice(self.proxy_to_use_list) for i in range(1, _cookies_count): # new_cookie = captchaResultGetter.get_valid_ch_cookie(proxy_to_use=_proxy_to_use, dvm=dvm, hc=hc) @@ -92,5 +95,5 @@ def get_app_and_app_sig(cookies: str): if __name__ == '__main__': cookiesPublisher = CookiesPublisher(queue_name=REQUEST_DATA_QUEUE_TEST) cookiesPublisher.set_up_connection() - cookieGenerator = CookiesGenerator(cookiesPublisher) + cookieGenerator = CookiesGenerator(proxy_to_use_list=PROXY_LIST_FR, cookiesPublisher=cookiesPublisher) cookieGenerator.run()