leadguru-jobs 0.579.0__py3-none-any.whl → 0.581.0__py3-none-any.whl

{leadguru_jobs-0.579.0.dist-info → leadguru_jobs-0.581.0.dist-info}/METADATA

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

{leadguru_jobs-0.579.0.dist-info → leadguru_jobs-0.581.0.dist-info}/RECORD

@@ -12,7 +12,7 @@ lgt_jobs/assets/images/logo.png,sha256=nt377FGkBalBF9Z9TYQRv9loHolTXwU8_VRYvCXXz
 lgt_jobs/assets/images/mail.png,sha256=eORzQcAMkFr7DjgtABVhJ_zFuXgO7OXv78lLF4b39B0,7168
 lgt_jobs/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lgt_jobs/jobs/analytics.py,sha256=IIsWt4A1qUw3w-S-8W16uKY1FRVWfXXA41_mu4uCNiM,979
-lgt_jobs/jobs/bot_stats_update.py,sha256=R6mf5FxhWbyx92AYXlj0FZVjz2Tgz86mBvLHrsQ9AFE,7502
+lgt_jobs/jobs/bot_stats_update.py,sha256=RmPA-7Nel2kkZmfGuDSTiSQjipr_DcsL13QgVRjrx0w,7531
 lgt_jobs/jobs/chat_history.py,sha256=tZ4XyOGdy1k6r4uotxSSKbXjTR_oRK6Ym0YthFYBtl8,6200
 lgt_jobs/jobs/connect_sources.py,sha256=_eA86KnS3AC6YCI1xk7VCV7lFmPRxta-wUStfdmajQU,4790
 lgt_jobs/jobs/inbox_leads.py,sha256=OSX-FNx27gWEKNBBc-hyq2odCxXytz7WHtQJajtz274,5670
@@ -22,12 +22,12 @@ lgt_jobs/jobs/send_code.py,sha256=dIlLPkG3GgGKIEqGiElyzrtdrnJNTL1Ak2V0xnE-WIQ,82
 lgt_jobs/jobs/send_slack_message.py,sha256=IuRqDGcDrz2EAZzF6nrLXO0aSA4mETB_tGOEbcMhjHE,2562
 lgt_jobs/jobs/update_slack_profile.py,sha256=0fUHBd2gEBne52sTfxwji2Wu-oYDM5dLGsL-rHq5kwM,2765
 lgt_jobs/jobs/user_balance_update.py,sha256=HxeEmDdloK9EcoEj1ybYTgxIkB0IQNLnss_kk3rLI1Q,2404
-lgt_jobs/jobs/workspace_connect.py,sha256=qgjIRzAQRbJBx--eYsTtYgOUh2-p3N_-LIltfbJrYYo,7102
+lgt_jobs/jobs/workspace_connect.py,sha256=LSZkkuPuHTqxvaTgqz_Y_kGIkF3Mmn_xwKf2h0uDWVg,7115
 lgt_jobs/lgt_common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lgt_jobs/lgt_common/helpers.py,sha256=tJNrKJYPa_MBmfRVrpHryoOWe4gnv7xOlmHuTh-bMBw,428
 lgt_jobs/lgt_common/lgt_logging.py,sha256=Zxr2sPpY-qU_YbL_47OM6dE_JA1ejG_Vr1PNnjF_jwo,576
 lgt_jobs/lgt_common/discord_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lgt_jobs/lgt_common/discord_client/discord_client.py,sha256=tvRn0P039Dp5d22-DivQhP2uQMC7ixZc7oe7IA4WrPE,2442
+lgt_jobs/lgt_common/discord_client/discord_client.py,sha256=dJr1tw9h-lTp5PUiZUHO1lrMGpeG-78NXCzLo5FdKAU,2459
 lgt_jobs/lgt_common/discord_client/methods.py,sha256=C46Vq9_-dlvs0nEW6ck8QvAmE_nPUfU89MJUhHaPMLk,391
 lgt_jobs/lgt_common/enums/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lgt_jobs/lgt_common/enums/slack_errors.py,sha256=nK7uoguq0F8bIHROp73oeszehT7CgS_VU45n9_Gq_WY,172
@@ -39,16 +39,16 @@ lgt_jobs/lgt_common/slack_client/methods.py,sha256=ctKF_1UHEmSWaRsoGEgMZicVabV7e
 lgt_jobs/lgt_common/slack_client/slack_client.py,sha256=icHapSOszFYnSofJ-eWl4lUWSOqng6S9XSRFfMx87Rg,13836
 lgt_jobs/lgt_common/slack_client/web_client.py,sha256=JrkEepL6EhEtX3Psv_sd25bcRB2P23LoGnFHcczQpnc,4778
 lgt_jobs/lgt_data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lgt_jobs/lgt_data/analytics.py,sha256=TFUffL8Dn-vIJ3ZaCVN-cj1Qnjb8F9GJyuk6Fo7mR1s,21645
+lgt_jobs/lgt_data/analytics.py,sha256=fiN88zcIxs_bRMmXL7ftp4FvBeJ5I7QBPE4tbwKJ39E,21689
 lgt_jobs/lgt_data/engine.py,sha256=Rsbz-CApAo_TVDssdjBkv8v_fVOZm_Uh1S6W4fnaEWo,7728
 lgt_jobs/lgt_data/enums.py,sha256=jBH5WGBtDAvFrh4iiPIzlQ-XrImMpuwqstuasG55mJ0,2209
 lgt_jobs/lgt_data/helpers.py,sha256=NDa-V5EYaJfkGoWsmQSwSe6N_jxNxs8tHRQzW1iST6k,480
-lgt_jobs/lgt_data/model.py,sha256=QAWnCPHEoNdXaPPCgMi_tHLx11zj0xcGUUUg2CDS3DQ,28554
+lgt_jobs/lgt_data/model.py,sha256=Nkl28_3u3auFAylDBIpiwQfNEqx3hpQaFz2SyigoSxM,28660
 lgt_jobs/lgt_data/mongo_repository.py,sha256=kLE906lfqhHF9gp1qvH_3XKCZGCZnxjLMrBLWBy7doo,46099
 lgt_jobs/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lgt_jobs/services/web_client.py,sha256=GLWsJkIC8rv6xLFaLwcMm4EwBlVDu0njORwkZqBJaE4,2086
 lgt_jobs/templates/new_message.html,sha256=dZl8UmdAOOMq4nidvAgMFroSrTV7Pw0RWt2yLp_2idg,6989
-leadguru_jobs-0.579.0.dist-info/METADATA,sha256=gN-uDnFZEp0vfTweIZrh38230PA_Gasm5gBdaF56btE,1319
-leadguru_jobs-0.579.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-leadguru_jobs-0.579.0.dist-info/top_level.txt,sha256=rIuw1DqwbnZyeoarBSC-bYeGOhv9mZBs7_afl9q4_JI,9
-leadguru_jobs-0.579.0.dist-info/RECORD,,
+leadguru_jobs-0.581.0.dist-info/METADATA,sha256=TOqrRqqYVy5rp1Old-oRd7OL0FhkNtR30j-OlA9cmvU,1319
+leadguru_jobs-0.581.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+leadguru_jobs-0.581.0.dist-info/top_level.txt,sha256=rIuw1DqwbnZyeoarBSC-bYeGOhv9mZBs7_afl9q4_JI,9
+leadguru_jobs-0.581.0.dist-info/RECORD,,

lgt_jobs/jobs/bot_stats_update.py

@@ -32,6 +32,9 @@ class BotStatsUpdateJob(BaseBackgroundJob, ABC):
         bots_rep = DedicatedBotRepository()
         users_rep = UserMongoRepository()
         bot = bots_rep.get_one(id=data.bot_id, include_deleted=True)
+        received_messages, filtered_messages = get_bots_aggregated_analytics(bot_ids=[str(bot.id)])
+        bot.messages_received = sum([received_messages[doc] for doc in received_messages])
+        bot.messages_filtered = sum([filtered_messages[doc] for doc in filtered_messages])
 
         if bot.source.source_type == SourceType.DISCORD:
             client = DiscordClient(bot.token)
@@ -107,7 +110,6 @@ class BotStatsUpdateJob(BaseBackgroundJob, ABC):
             bot.source.source_name = bot_name
             bot.slack_url = bot.registration_link = team_info['team']['url']
             bots_rep.add_or_update(bot)
-        received_messages, filtered_messages = get_bots_aggregated_analytics(bot_ids=[bot.id])
         try:
             channels_response = client.channels_list()
         except:
@@ -134,8 +136,6 @@ class BotStatsUpdateJob(BaseBackgroundJob, ABC):
             users_count += num_members
 
         bot.active_channels = active_channels
-        bot.messages_received = received_messages.get(bot.source.source_id, 0)
-        bot.messages_filtered = filtered_messages.get(bot.source.source_id, 0)
         bot.connected_channels = connected_channels
         bot.channels = len(channels)
         bot.channels_users = channels_users

lgt_jobs/jobs/workspace_connect.py

@@ -7,7 +7,7 @@ from lgt_jobs.lgt_data.enums import StatusConnection, SourceType
 from lgt_jobs.lgt_data.model import UserWorkspace, SlackUser, DedicatedBotModel
 from lgt_jobs.lgt_data.mongo_repository import UserMongoRepository, DedicatedBotRepository
 from lgt_jobs.basejobs import BaseBackgroundJobData, BaseBackgroundJob
-from .bot_stats_update import BotStatsUpdateJob, BotStatsUpdateJobData
+from lgt_jobs.jobs.bot_stats_update import BotStatsUpdateJob, BotStatsUpdateJobData
 from lgt_jobs.runner import BackgroundJobRunner
 import logging as log
 from pydantic import BaseModel

lgt_jobs/lgt_common/discord_client/discord_client.py

@@ -29,7 +29,7 @@ class DiscordClient:
         return {}
 
     def get_servers(self) -> list | dict:
-        response = requests.get(f"{self.base_url}{DiscordMethods.USER_GUILDS.value}", headers=self.headers)
+        response = requests.get(f"{self.base_url}{DiscordMethods.USER_GUILDS.value}?with_counts=true", headers=self.headers)
         return self.__response(response).json()
 
     def get_dms(self) -> list | dict:

lgt_jobs/lgt_data/analytics.py

@@ -8,7 +8,7 @@ from lgt_jobs.lgt_data.mongo_repository import to_object_id
 from pymongo import MongoClient
 
 client = MongoClient(os.environ.get('MONGO_CONNECTION_STRING', 'mongodb://127.0.0.1:27017/'))
-db = client.lgt_admin
+db = client.get_database('lgt_admin')
 
 
 def _build_date_aggregated_analytics_pipeline(source_id=None, email=None, started_at: datetime = None,
@@ -540,12 +540,12 @@ def get_bots_aggregated_analytics(from_date: datetime = None,
     if bot_ids is not None:
         pipeline.insert(0, {"$match": {"extra_ids": {"$in": bot_ids}}})
 
-    received_messages = list(db[f'received_messages'].aggregate(pipeline))
+    received_messages = list(db.get_collection('received_messages').aggregate(pipeline))
 
     if configs is not None:
         pipeline.insert(0, {"$match": {"attributes": {"$in": configs}}})
 
-    filtered_messages = list(db[f'filtered_messages'].aggregate(pipeline))
+    filtered_messages = list(db.get_collection('filtered_messages').aggregate(pipeline))
     received_messages_dic = OrderedDict()
     filtered_messages_dic = OrderedDict()
 

lgt_jobs/lgt_data/model.py

@@ -162,6 +162,7 @@ class Server:
         self.active = False
         self.deleted = False
         self.subscribers = 0
+        self.approximate_member_count = 0
 
     @classmethod
     def from_dic(cls, dic: dict):
@@ -174,6 +175,7 @@ class Server:
                 setattr(model, k, v)
 
         model.channels = [Channel.from_dic(channel) for channel in dic.get("channels", [])]
+        model.subscribers = dic.get('approximate_member_count')
         return model
 
     def to_dic(self):