leadguru-jobs 0.414.0__tar.gz → 0.416.0__tar.gz

{leadguru_jobs-0.414.0 → leadguru_jobs-0.416.0}/PKG-INFO

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

{leadguru_jobs-0.414.0 → leadguru_jobs-0.416.0}/leadguru_jobs.egg-info/PKG-INFO

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

{leadguru_jobs-0.414.0 → leadguru_jobs-0.416.0}/lgt_jobs/lgt_data/model.py

@@ -112,7 +112,6 @@ class BaseBotModel(Credentials):
         self.banned: bool = False
         self.associated_user = None
         self.type: SourceType | None = None
-        self.deleted = False
 
 
 class DedicatedBotModel(BaseBotModel):

{leadguru_jobs-0.414.0 → leadguru_jobs-0.416.0}/lgt_jobs/lgt_data/mongo_repository.py

@@ -767,6 +767,17 @@ class DedicatedBotRepository(BaseMongoRepository):
         docs = self.collection().find(pipeline)
         return [DedicatedBotModel.from_dic(doc) for doc in docs]
 
+    def get_source_ids_for_user(self, user_id, source_type):
+        docs = self.collection().find({
+            'user_id': user_id,
+            'source.source_type': source_type
+        }, {
+            '_id': 0,
+            'source_id': '$source.source_id'
+        })
+        source_ids = [item['source_id'] for item in docs]
+        return source_ids
+
     def get_one(self, **kwargs):
         pipeline = self.__create_bots_filter(**kwargs)
         return DedicatedBotModel.from_dic(self.collection().find_one(pipeline))
@@ -991,24 +1002,11 @@ class TeamRepository(BaseMongoRepository):
 class ChatRepository(BaseMongoRepository):
     collection_name = 'chat_messages'
 
-    def get_list(self, **kwargs):
-        pipeline = {}
-        text = kwargs.get('text')
-        user_id = kwargs.get('user_id')
-        bot_id = kwargs.get('bot_id')
-        sender_id = kwargs.get('sender_id')
-
-        if text:
-            pipeline['text'] = {"$regex": text, "$options": 'i'}
-        if bot_id:
-            pipeline['bot_id'] = to_object_id(bot_id)
-        if user_id:
-            pipeline['user_id'] = to_object_id(user_id)
-        if sender_id:
-            pipeline['sender_id'] = sender_id
-
+    def get_list(self, sender_id: str, bot_id: ObjectId):
         return [ChatMessage.from_dic(msg)
-                for msg in self.collection().find(pipeline).sort("ts", pymongo.ASCENDING)]
+                for msg in self.collection().find({'sender_id': sender_id,
+                                                   'bot_id': to_object_id(bot_id)}).sort("ts",
+                                                                                         pymongo.ASCENDING)]
 
     def upsert_messages(self, messages: list[dict]):
         operations = [UpdateOne({"ts": msg['ts']}, {'$set': msg}, upsert=True) for msg in messages]