leadguru-jobs 0.400.0__tar.gz → 0.402.0__tar.gz

{leadguru_jobs-0.400.0 → leadguru_jobs-0.402.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: leadguru_jobs
-Version: 0.400.0
+Version: 0.402.0
 Summary: LGT jobs builds
 Author-email: developer@leadguru.co
 Classifier: Development Status :: 5 - Production/Stable

{leadguru_jobs-0.400.0 → leadguru_jobs-0.402.0}/leadguru_jobs.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: leadguru_jobs
-Version: 0.400.0
+Version: 0.402.0
 Summary: LGT jobs builds
 Author-email: developer@leadguru.co
 Classifier: Development Status :: 5 - Production/Stable

{leadguru_jobs-0.400.0 → leadguru_jobs-0.402.0}/lgt_jobs/lgt_data/model.py

@@ -492,6 +492,7 @@ class ExtendedLeadModel(LeadModel):
         self.deleted = False
         self.user_lead: UserLeadModel | None = None
         self.dedicated: bool = False
+        self.bots: List[BotInfo] = []
 
     @classmethod
     def from_dic(cls, dic: dict):
@@ -507,6 +508,7 @@ class ExtendedLeadModel(LeadModel):
         result.user_lead = UserLeadModel.from_dic(dic.get('user_lead'))
         result.last_conversation = [SlackHistoryMessageModel.from_dic(message)
                                     for message in dic.get('last_conversation', [])]
+        result.bots = [BotInfo.from_dic(bot) for bot in dic.get('bots', [])]
         return result
 
     def to_csv(self, board_name: str) -> List[str]:
@@ -516,6 +518,35 @@ class ExtendedLeadModel(LeadModel):
                 self.message.message.replace('\n', ' ').strip()]
 
 
+class BotInfo:
+    def __init__(self):
+        self.id = None
+        self.invalid_creds: bool | None = False
+        self.source = None
+        self.banned: bool | None
+        self.user_name: str = ''
+        self.associated_user: str | None
+
+    @classmethod
+    def from_dic(cls, dic: dict):
+        if not dic:
+            return None
+
+        model = BotInfo()
+        for k, v in dic.items():
+            if hasattr(model, k):
+                setattr(model, k, v)
+
+        if '_id' in dic:
+            setattr(model, 'id', dic['_id'])
+
+        return model
+
+    def to_dic(self):
+        result = copy.deepcopy(self.__dict__)
+        return result
+
+
 class SlackReplyModel(BaseModel):
     def __init__(self):
         super().__init__()

{leadguru_jobs-0.400.0 → leadguru_jobs-0.402.0}/lgt_jobs/lgt_data/mongo_repository.py

@@ -755,6 +755,7 @@ class DedicatedBotRepository(BaseMongoRepository):
         if '_id' in bot_dict:
             bot_dict.pop('_id')
         update_response = self.collection().update_one({"source.source_id": bot.source.source_id,
+                                                        "user_name": bot.user_name,
                                                         "user_id": to_object_id(bot.user_id)},
                                                        {'$set': bot_dict}, upsert=True)
         bot.id = update_response.upserted_id if update_response.upserted_id else bot.id