leadguru-jobs 0.400.0__py3-none-any.whl → 0.402.0__py3-none-any.whl

{leadguru_jobs-0.400.0.dist-info → leadguru_jobs-0.402.0.dist-info}/METADATA

@@ -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.dist-info → leadguru_jobs-0.402.0.dist-info}/RECORD

@@ -38,12 +38,12 @@ lgt_jobs/lgt_data/analytics.py,sha256=qo5nNFQxOTTk8i3tmCq0d3F9iaV9MTIR84SSi6uNrJ
 lgt_jobs/lgt_data/engine.py,sha256=iTcLvU7QZvB14hcERQrxSql9cVEcMEzu_tUcWD2Z24A,8076
 lgt_jobs/lgt_data/enums.py,sha256=A5nWwBgTVW7XaqL30NHNmO0SG_yeDk1dI-U5xrgU2zg,1552
 lgt_jobs/lgt_data/helpers.py,sha256=58ycqKF1z5nAw0vs6yVCg-SQet107YHyKYXaQUc7M68,157
-lgt_jobs/lgt_data/model.py,sha256=cdQpVTIB3FmFg1rcS3dbkkdjj1WhhTBDB9lfI_sVSHM,27584
-lgt_jobs/lgt_data/mongo_repository.py,sha256=rcfXaXdUWdLo9n8R7JElP_RIF-6UUHkUPPAoNWa3FRw,35928
+lgt_jobs/lgt_data/model.py,sha256=jK3jBpp2sHkoV2XKoBChKjAjWhtp_6eQra3_RnD6Adg,28356
+lgt_jobs/lgt_data/mongo_repository.py,sha256=tn_G-ZVOiQXanQ_L7dYeh7ftg6KK1ngyMRhMOrZq-ho,36012
 lgt_jobs/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lgt_jobs/services/web_client.py,sha256=D3y5KVVBk7mesRnsiByx4tFmNbfTOKOUoIp3VCHuk0I,1983
 lgt_jobs/templates/new_message_mail_template.html,sha256=82XdDPtmkJVUWUcZtgd8bv_9-viAG2WWOnf3AhR6PeY,3635
-leadguru_jobs-0.400.0.dist-info/METADATA,sha256=LHdKdO1BBOT72QTK1MQAD5adFXsJQNcC5uC3KuSvkG0,1296
-leadguru_jobs-0.400.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-leadguru_jobs-0.400.0.dist-info/top_level.txt,sha256=rIuw1DqwbnZyeoarBSC-bYeGOhv9mZBs7_afl9q4_JI,9
-leadguru_jobs-0.400.0.dist-info/RECORD,,
+leadguru_jobs-0.402.0.dist-info/METADATA,sha256=nwV7-rTXNjJxZmtMmjck9b0PUlBdxwDZaUQOaKNOzgU,1296
+leadguru_jobs-0.402.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+leadguru_jobs-0.402.0.dist-info/top_level.txt,sha256=rIuw1DqwbnZyeoarBSC-bYeGOhv9mZBs7_afl9q4_JI,9
+leadguru_jobs-0.402.0.dist-info/RECORD,,

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__()

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