leadguru-jobs 0.256.0__tar.gz → 0.257.0__tar.gz

{leadguru_jobs-0.256.0 → leadguru_jobs-0.257.0}/PKG-INFO

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

{leadguru_jobs-0.256.0 → leadguru_jobs-0.257.0}/leadguru_jobs.egg-info/PKG-INFO

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

{leadguru_jobs-0.256.0 → leadguru_jobs-0.257.0}/lgt_jobs/jobs/bot_stats_update.py

@@ -1,6 +1,7 @@
 from abc import ABC
 from typing import Optional
 from lgt.common.python.slack_client.web_client import SlackWebClient
+from lgt_data.model import DedicatedBotModel
 from lgt_data.mongo_repository import BotMongoRepository, DedicatedBotRepository
 from pydantic import BaseModel
 from lgt_data.analytics import get_bots_aggregated_analytics
@@ -31,21 +32,23 @@ class BotStatsUpdateJob(BaseBackgroundJob, ABC):
             bots_rep = BotMongoRepository()
             bot = bots_rep.get_by_id(data.bot_name)
 
-        if not bot.invalid_creds:
-            return
+        client = SlackWebClient(bot.token, bot.cookies)
+        if bot.invalid_creds:
+            test_auth_response = client.test_auth()
+            if ((test_auth_response.status_code == 200) and
+                    (test_auth_response.json().get("error") != SlackErrors.INVALID_AUTH)):
+                BotStatsUpdateJob.__updated_invalid_creds_flag(bot, False)
 
         received_messages, filtered_messages = get_bots_aggregated_analytics(bot_ids=[bot.id])
-        client = SlackWebClient(bot.token, bot.cookies)
         if not bot.token or not bot.cookies:
             log.warning(f"[BotStatsUpdateJob]: Bot {bot.id} has no credentials.")
-            bot.invalid_creds = True
-            bots_rep.add_or_update(bot)
+            if not bot.invalid_creds:
+                BotStatsUpdateJob.__updated_invalid_creds_flag(bot, True)
             return
         channels_response = client.channels_list()
         if not channels_response['ok']:
             if channels_response.get("error") == SlackErrors.INVALID_AUTH:
-                bot.invalid_creds = True
-                bots_rep.add_or_update(bot)
+                BotStatsUpdateJob.__updated_invalid_creds_flag(bot, True)
             else:
                 log.warning(f"[BotStatsUpdateJob]: Error during update bot {bot.id} stats. Error: {channels_response}")
             return
@@ -75,3 +78,8 @@ class BotStatsUpdateJob(BaseBackgroundJob, ABC):
         # save only last 50 messages
         bot.recent_messages = bot.recent_messages[-50:]
         bots_rep.add_or_update(bot)
+
+    @staticmethod
+    def __updated_invalid_creds_flag(bot: DedicatedBotModel, invalid_creds: bool):
+        bot.invalid_creds = invalid_creds
+        DedicatedBotRepository().add_or_update(bot)