From 9417c9dfeb277f8662d38fc0b2692e7ffada4a8c Mon Sep 17 00:00:00 2001
From: PAN Lei <panleicim@gmail.com>
Date: Thu, 13 Jun 2024 22:25:09 +0200
Subject: [PATCH] add ua to db

---
 models/ReserveResultPojo.py | 2 ++
 models/contact_pojo.py      | 5 ++++-
 workers/sender.py           | 2 ++
 3 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/models/ReserveResultPojo.py b/models/ReserveResultPojo.py
index 2cf6892..e4eb7b7 100755
--- a/models/ReserveResultPojo.py
+++ b/models/ReserveResultPojo.py
@@ -35,6 +35,7 @@ class ReserveResultPojo:
     created_at = None
     validated_at = None
     proxy: str = None
+    ua: str = None
 
     @staticmethod
     def from_firestore_dict(source):
@@ -129,6 +130,7 @@ class ReserveResultPojo:
             u'accepted': self.accepted,
             u'url_validated': self.url_validated,
             u'proxy': self.proxy,
+            u'ua': self.ua,
         }
 
         return dest
diff --git a/models/contact_pojo.py b/models/contact_pojo.py
index 54102f3..494135e 100755
--- a/models/contact_pojo.py
+++ b/models/contact_pojo.py
@@ -12,6 +12,7 @@ class ContactPojo:
     position: int
     store: str
     note: str
+    ua: str
 
     def __init__(self, phone_number: str, passport_number: str, last_name: str, first_name: str, mail: str,
                  ccid: str = "",
@@ -24,6 +25,7 @@ class ContactPojo:
         self.mail = mail
         self.position = position
         self.note = ""
+        self.ua = ""
 
     def to_firestore_dict(self):
         dest = {
@@ -34,7 +36,8 @@ class ContactPojo:
             u'mail': self.mail,
             u'store': self.store,
             u'ccid': self.ccid,
-            u'position': self.position
+            u'position': self.position,
+            u'ua': self.ua
         }
 
         return dest
diff --git a/workers/sender.py b/workers/sender.py
index 894d07b..cfa3efc 100644
--- a/workers/sender.py
+++ b/workers/sender.py
@@ -36,6 +36,7 @@ class Sender:
         result = ReserveResultPojo(type=status, phone=contact.phone, message=status.value, url=url,
                                    firstName=contact.first_name, lastName=contact.last_name, email=contact.mail,
                                    passport=contact.passport, ccid=contact.ccid)
+        result.ua = contact.ua
         if "oxylabs" in self.proxy_to_use["http"]:
             if "mob" in self.proxy_to_use["http"]:
                 result.proxy = "oxylabs_mob"
@@ -93,6 +94,7 @@ class Sender:
                 # add to mongodb
                 self.logger.info(response.text)
                 self.logger.info("{}:{}".format(contact.mail, response.url))
+                contact.ua = js_data.ua
                 self.publish_message_to_queue(contact, status=PublishType.SUCCESS, url=response.url,
                                               store_type=_selected_store)
                 cookies_to_set = response.headers['set-cookie']