leadguru-jobs 0.543.0__tar.gz → 0.545.0__tar.gz

{leadguru_jobs-0.543.0 → leadguru_jobs-0.545.0}/PKG-INFO

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

{leadguru_jobs-0.543.0 → leadguru_jobs-0.545.0}/leadguru_jobs.egg-info/PKG-INFO

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

{leadguru_jobs-0.543.0 → leadguru_jobs-0.545.0}/lgt_jobs/lgt_common/slack_client/slack_client.py

@@ -1,3 +1,5 @@
+import asyncio
+import aiohttp
 import requests
 import json
 import io
@@ -22,6 +24,34 @@ class SlackClient:
         else:
             self.cookies = cookies
 
+    def join_channels(self, channels):
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+        tasks = asyncio.gather(*[self.join_channel_async(channel) for channel in channels])
+        results = loop.run_until_complete(tasks)
+        loop.close()
+        return results
+
+    def leave_channels(self, channels):
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+        tasks = asyncio.gather(*[self.leave_channel_async(channel) for channel in channels])
+        results = loop.run_until_complete(tasks)
+        loop.close()
+        return results
+
+    async def join_channel_async(self, channel):
+        async with aiohttp.ClientSession() as session:
+            url = f'{self.base_url}{SlackMethods.conversations_join}?{self.__channel_payload(channel)}'
+            async with session.post(url=url, cookies=self.cookies) as response:
+                return await response.json()
+
+    async def leave_channel_async(self, channel):
+        async with aiohttp.ClientSession() as session:
+            url = f'{self.base_url}{SlackMethods.conversations_leave}?{self.__channel_payload(channel)}'
+            async with session.post(url=url, cookies=self.cookies) as response:
+                return await response.json()
+
     def upload_file(self, file, file_name):
         payload = {"content": file, "filename": file_name}
         return requests.post(f"{self.base_url}{SlackMethods.upload_file}", data=payload,

{leadguru_jobs-0.543.0 → leadguru_jobs-0.545.0}/lgt_jobs/lgt_common/slack_client/web_client.py

@@ -158,3 +158,9 @@ class SlackWebClient:
 
     def get_file_info(self, file_id: str):
         return self.client.get_file_info(file_id)
+
+    def channel_join(self, channels):
+        return self.client.join_channels(channels)
+
+    def channel_leave(self, channels):
+        return self.client.leave_channels(channels)

{leadguru_jobs-0.543.0 → leadguru_jobs-0.545.0}/lgt_jobs/lgt_data/analytics.py

@@ -81,6 +81,51 @@ def _prepare_date_analytics_doc(doc, ordered_result_dict: Dict[str, int]):
         ordered_result_dict[str_date] = item["count"]
     return ordered_result_dict
 
+def get_channel_aggregated_analytics(from_date: datetime = None, to_date: datetime = None,
+                                     bot_id: Optional[str | ObjectId] = None):
+    pipeline = [
+        {
+            '$match': {
+                'extra_ids': {"$in": [str(bot_id)]}
+            }
+        },
+        {
+            "$group": {
+                '_id': {
+                    '$arrayElemAt': [
+                        '$attributes', 0
+                    ]
+                },
+                'count': {"$sum": 1}
+            }
+        },
+        {
+            '$limit': 1000
+        }
+    ]
+
+    if from_date:
+        beginning_of_the_day = datetime(from_date.year, from_date.month, from_date.day, 0, 0, 0, 0)
+        pipeline.insert(0, {"$match": {"created_at": {"$gte": beginning_of_the_day}}})
+
+    if to_date:
+        end_of_the_day = datetime(to_date.year, to_date.month, to_date.day, 23, 59, 59, 999)
+        pipeline.insert(0, {"$match": {"created_at": {"$lte": end_of_the_day}}})
+
+    received_messages = list(db[f'received_messages'].aggregate(pipeline))
+    filtered_messages = list(db[f'filtered_messages'].aggregate(pipeline))
+
+    received_messages_dic = OrderedDict()
+    filtered_messages_dic = OrderedDict()
+
+    for item in received_messages:
+        received_messages_dic[item["_id"]] = item["count"]
+
+    for item in filtered_messages:
+        filtered_messages_dic[item["_id"]] = item["count"]
+
+    return received_messages_dic, filtered_messages_dic
+
 
 def get_aggregated_user_leads(user_id: Union[ObjectId, str],
                               from_date: datetime,

{leadguru_jobs-0.543.0 → leadguru_jobs-0.545.0}/lgt_jobs/runner.py

@@ -32,13 +32,15 @@ class BackgroundJobRunner:
         BackgroundJobRunner.submit_data(job_data)
 
     @staticmethod
-    def submit_data(data: dict):
+    def submit_data(data: dict, topic: str = None):
         json_str = json.dumps(data, ensure_ascii=False, default=datetime_converter)
-        BackgroundJobRunner.submit_json(json_str)
+        BackgroundJobRunner.submit_json(json_str, topic)
 
     @staticmethod
-    def submit_json(json_str: str):
-        publish_message2_pubsub(background_jobs_topic, message_json=json_str)
+    def submit_json(json_str: str, topic: str = None):
+        if not topic:
+            topic = background_jobs_topic
+        publish_message2_pubsub(topic, message_json=json_str)
 
     @staticmethod
     def schedule(jib: str, job: type, data: BaseBackgroundJobData, scheduled_at: datetime):