leadguru-jobs 0.553.0__py3-none-any.whl → 0.554.0__py3-none-any.whl

{leadguru_jobs-0.553.0.dist-info → leadguru_jobs-0.554.0.dist-info}/METADATA

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

{leadguru_jobs-0.553.0.dist-info → leadguru_jobs-0.554.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-lgt_jobs/__init__.py,sha256=lrhMUJ7a5YSkYswQkQe981b_w8h8NHyncwfEduXEq6Q,1992
+lgt_jobs/__init__.py,sha256=kmM5J3qBJd-fOZa2I4gkh36IytrgN6_18gWwwpR3n6Y,2156
 lgt_jobs/basejobs.py,sha256=LPEclvY6fIG17YMGy9pRK0Q58n62gkI5W1BTK6R2p-c,1234
 lgt_jobs/env.py,sha256=cRO03GGvstUjBsv3uYO-iakrOvAk_ZWUP_fnmf21iZQ,789
 lgt_jobs/main.py,sha256=cK_nkBtJHnUNDbba1WZotqPtI_6OWxiYQkAgco9OAmE,1539
@@ -14,6 +14,7 @@ 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/chat_history.py,sha256=QZdSWlk_DrDJt5yrNApYHiUEWQMYcKyuXaP_PlzkjYw,6303
+lgt_jobs/jobs/connect_sources.py,sha256=2tl1MGdm6l0yFwCXlD0J2tFSQpj7i1eT9sYd1nMOwWA,3596
 lgt_jobs/jobs/inbox_leads.py,sha256=EYxdQPseNBY5ON7BK3LpnXDCgQpzHTPLRJCZmZiAAHA,5582
 lgt_jobs/jobs/mass_message.py,sha256=1mFcBlL2MhzLbj5yrd_NyJc7TXDWCcROAzGDnr0miMU,2035
 lgt_jobs/jobs/send_code.py,sha256=dIlLPkG3GgGKIEqGiElyzrtdrnJNTL1Ak2V0xnE-WIQ,824
@@ -41,12 +42,12 @@ lgt_jobs/lgt_data/analytics.py,sha256=TFUffL8Dn-vIJ3ZaCVN-cj1Qnjb8F9GJyuk6Fo7mR1
 lgt_jobs/lgt_data/engine.py,sha256=Rsbz-CApAo_TVDssdjBkv8v_fVOZm_Uh1S6W4fnaEWo,7728
 lgt_jobs/lgt_data/enums.py,sha256=GZn6wm-kOd3veEOo1qlOR8QYje4vKaGV0na9jGt9388,2160
 lgt_jobs/lgt_data/helpers.py,sha256=NDa-V5EYaJfkGoWsmQSwSe6N_jxNxs8tHRQzW1iST6k,480
-lgt_jobs/lgt_data/model.py,sha256=DDfoWeH6rLWKoZsFJ2tI5Yagj1WlOa0PIJp-GqBIVSA,28112
+lgt_jobs/lgt_data/model.py,sha256=HZenmwr_fmxVw0XhUlgHZJOaSK_Ouv9mBcY6DfC61oU,28098
 lgt_jobs/lgt_data/mongo_repository.py,sha256=qqbt-oJ-RHmICLARgWvHuyPwVeKFkUqJZ2MA1bfIlXs,44762
 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=BD_E90akmQ1Wf07wtZAjeK_7DUKRmja5HFHVo_AKI24,6994
-leadguru_jobs-0.553.0.dist-info/METADATA,sha256=aW7QHrxDxKqTKUytouVMGYCf2UD7ZYHULpKOthzi_p0,1319
-leadguru_jobs-0.553.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-leadguru_jobs-0.553.0.dist-info/top_level.txt,sha256=rIuw1DqwbnZyeoarBSC-bYeGOhv9mZBs7_afl9q4_JI,9
-leadguru_jobs-0.553.0.dist-info/RECORD,,
+leadguru_jobs-0.554.0.dist-info/METADATA,sha256=Rney8pzxyO6HwI3AvxAeekzMpaaXotwznfx8IgZ3fmM,1319
+leadguru_jobs-0.554.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+leadguru_jobs-0.554.0.dist-info/top_level.txt,sha256=rIuw1DqwbnZyeoarBSC-bYeGOhv9mZBs7_afl9q4_JI,9
+leadguru_jobs-0.554.0.dist-info/RECORD,,

lgt_jobs/__init__.py

@@ -3,6 +3,7 @@ name = "lgt_jobs"
 from .jobs.user_balance_update import UpdateUserBalanceJob, UpdateUserBalanceJobData
 from .jobs.send_slack_message import SendSlackMessageJob, SendSlackMessageJobData
 from .jobs.analytics import (TrackAnalyticsJob, TrackAnalyticsJobData)
+from .jobs.connect_sources import (ConnectSourceJobData, ConnectSourceJob)
 from .jobs.bot_stats_update import (BotStatsUpdateJob, BotStatsUpdateJobData)
 from .jobs.chat_history import (LoadChatHistoryJob, LoadChatHistoryJobData)
 from .jobs.update_slack_profile import (UpdateExternalUserProfileJob, UpdateExternalUserProfileJobData)
@@ -22,6 +23,7 @@ jobs_map = {
     "SendSlackMessageJob": SendSlackMessageJob,
     "UpdateUserBalanceJob": UpdateUserBalanceJob,
     "SendMassMessageSlackChannelJob": SendMassMessageSlackChannelJob,
+    "ConnectSourceJob": ConnectSourceJob
 }
 __all__ = [
     # Jobs
@@ -35,6 +37,7 @@ __all__ = [
     SendSlackMessageJob,
     UpdateUserBalanceJob,
     SendMassMessageSlackChannelJob,
+    ConnectSourceJob,
 
     # module classes
     BackgroundJobRunner,
@@ -50,6 +53,7 @@ __all__ = [
     SendSlackMessageJobData,
     UpdateUserBalanceJobData,
     SendMassMessageSlackChannelJobData,
+    ConnectSourceJobData,
     # mapping
     jobs_map
 ]

lgt_jobs/jobs/connect_sources.py

@@ -0,0 +1,81 @@
+from abc import ABC
+from typing import List
+import logging as log
+from lgt_jobs import BaseBackgroundJob, BaseBackgroundJobData, BackgroundJobRunner, BotStatsUpdateJob, \
+    BotStatsUpdateJobData
+from lgt_jobs.lgt_data.engine import UserTrackAction
+from lgt_jobs.lgt_data.enums import SourceType, UserAction, StatusConnection
+from lgt_jobs.lgt_data.model import BaseModel, DedicatedBotModel, Server, Source, SlackUser
+from lgt_jobs.lgt_data.mongo_repository import DedicatedBotRepository, UserMongoRepository
+
+
+class ConnectSourceJobData(BaseBackgroundJobData, BaseModel):
+    slack_email: str
+    sources_ids: List[str]
+    action_user_email: str
+    user_email: str
+
+
+class ConnectSourceJob(BaseBackgroundJob, ABC):
+    @property
+    def job_data_type(self) -> type:
+        return ConnectSourceJobData
+
+    def exec(self, data: ConnectSourceJobData):
+        users_repository = UserMongoRepository()
+        bots_repository = DedicatedBotRepository()
+        action_user = users_repository.get_by_email(data.action_user_email)
+        user = users_repository.get_by_email(data.user_email)
+        if not action_user:
+            log.info(f'Unable to find user with email: {data.action_user_email}')
+            return
+
+        slack_user: SlackUser = action_user.get_slack_user(data.slack_email)
+        if not slack_user:
+            log.info(f"You have no credentials for {data.slack_email}. Try to login to Slack.")
+        bots = []
+        for ws in slack_user.workspaces:
+            if ws.id in data.sources_ids:
+                active_bot = bots_repository.get_one(user_id=action_user.id, active_server_id=ws.id)
+                bot = bots_repository.get_one(user_id=action_user.id, server_id=ws.id, include_deleted=True,
+                                              user_name=data.slack_email)
+                if active_bot:
+                    continue
+                if not bot:
+                    bot = DedicatedBotModel()
+                    bot.servers = [Server()]
+
+                bot.user_name = slack_user.email
+                bot.slack_url = ws.url
+                bot.user_id = action_user.id
+                bot.created_by = user.email
+                bot.registration_link = ws.url
+                bot.token = ws.token
+                bot.cookies = slack_user.cookies
+                bot.deleted = False
+                bot.paused = False
+                bot.invalid_creds = False
+                bot.banned = False
+                bot.two_factor_required = ws.two_factor_required
+                source = Source()
+
+                for server in bot.servers:
+                    server.id = ws.id
+                    server.name = ws.domain
+                    server.deleted = False
+                    server.active = True
+                    server.icon = ws.icon
+
+                source.source_id = ws.id
+                source.source_name = ws.domain
+                bot.type = source.source_type = SourceType.SLACK
+                bot.icon = ws.icon
+                bot.source = source
+                bots_repository.add_or_update(bot)
+                UserTrackAction.track(action_user.id, UserAction.START_SOURCE, ws.id)
+                bot = bots_repository.get_one(user_id=action_user.id, server_id=ws.id, user_name=slack_user.email)
+                bots.append(bot)
+                BackgroundJobRunner.submit(BotStatsUpdateJob, BotStatsUpdateJobData(bot_id=str(bot.id)))
+        slack_user.status = StatusConnection.COMPLETE
+        users_repository.set(action_user.id, slack_users=[user.to_dic() for user in action_user.slack_users])
+        return bots

lgt_jobs/lgt_data/model.py

@@ -290,8 +290,8 @@ class UserModel(BaseModel):
         self.subscription_expired_at: datetime | None = None
         self.balance: str | None = None
         self.subscription_name: str | None = None
-        self.subscription_expiration_notified = False
-        self.subscription_expiration_warning_notified = False
+        self.trial_expiration_notified = False
+        self.trial_expiration_warning_notified = False
 
     @classmethod
     def from_dic(cls, dic: dict):