leadguru-jobs 0.473.0__py3-none-any.whl → 0.475.0__py3-none-any.whl

{leadguru_jobs-0.473.0.dist-info → leadguru_jobs-0.475.0.dist-info}/METADATA

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

{leadguru_jobs-0.473.0.dist-info → leadguru_jobs-0.475.0.dist-info}/RECORD

@@ -17,7 +17,7 @@ lgt_jobs/jobs/chat_history.py,sha256=ZSZyXZr8GNmilqtqhtZ_ZKn26IL_wI4p7QOcEYu5g8M
 lgt_jobs/jobs/inbox_leads.py,sha256=49Xc83MIC4paZjOOgIXQeOHIpq5oCfFZiI9S9UZ7kjI,4660
 lgt_jobs/jobs/mass_message.py,sha256=C67GGxe8fDEHPvKjLKvaD-sbz3-nLOo0jC717ZqHAWM,2046
 lgt_jobs/jobs/send_code.py,sha256=HCfDxdBsWfEhZBvRdmUcpR2juFFvU4RBMHYoZWD575k,835
-lgt_jobs/jobs/send_slack_message.py,sha256=_aQOU3S3I6zIUDzG4wo3Jj4CzI7j6_keorMWT9o3fD4,1907
+lgt_jobs/jobs/send_slack_message.py,sha256=nCicew3WjjxE2-MeBj_Qd678btRdcZDoGZ4hYn8et-I,2808
 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=5UPDgbAFLwz_sClIwQpSHQKbf3_34kzV0Rhay6SuwHk,7113
@@ -40,14 +40,14 @@ lgt_jobs/lgt_common/slack_client/web_client.py,sha256=oZV7oUcSXXngfMI6oXDedlQWeP
 lgt_jobs/lgt_data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lgt_jobs/lgt_data/analytics.py,sha256=qo5nNFQxOTTk8i3tmCq0d3F9iaV9MTIR84SSi6uNrJ8,23427
 lgt_jobs/lgt_data/engine.py,sha256=OkTIIsVhb4o0dcDOzolb-NNTftzhJvsCuldxg1PvLUM,8113
-lgt_jobs/lgt_data/enums.py,sha256=8cTLz8e6Jz4ikjNzqMiorwMMdVjU2SyXtl_tD7nHKxw,2326
+lgt_jobs/lgt_data/enums.py,sha256=P-Jo_kQfFanbN5ztjQZIaWkZ6qOmAjHdypZ85ReOgdA,2407
 lgt_jobs/lgt_data/helpers.py,sha256=58ycqKF1z5nAw0vs6yVCg-SQet107YHyKYXaQUc7M68,157
-lgt_jobs/lgt_data/model.py,sha256=wsa7_MzCU9EGcdclpzLaQx_ovybBARyfpnzJ-P-G2pM,28583
-lgt_jobs/lgt_data/mongo_repository.py,sha256=dEdpMkBMhdzSWzj_HJCZYEo89zkVBqGXgyjyFLNOTVA,45463
+lgt_jobs/lgt_data/model.py,sha256=6eds2J7gPfIVAs7E1bTJEvH_M8rWdJ091cQmIw5Sznk,29090
+lgt_jobs/lgt_data/mongo_repository.py,sha256=j_kvz8tp6YMhPYzjSPVX7AjtR1VybIQipNQ6qOKqukI,45867
 lgt_jobs/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lgt_jobs/services/web_client.py,sha256=eCgzyyXco6iPwAJziyRek_3ns1bxJZKknnLIJApKeoo,2070
 lgt_jobs/templates/new_message.html,sha256=I3J6YPz-uhGbGnuGc5d7Hxj0X4VzOtT8PqrQQGPJfc4,7002
-leadguru_jobs-0.473.0.dist-info/METADATA,sha256=vEJhaFlrgx7zJEWzGNTtN45CMh00uH_wrSU8QbLtI58,1319
-leadguru_jobs-0.473.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-leadguru_jobs-0.473.0.dist-info/top_level.txt,sha256=rIuw1DqwbnZyeoarBSC-bYeGOhv9mZBs7_afl9q4_JI,9
-leadguru_jobs-0.473.0.dist-info/RECORD,,
+leadguru_jobs-0.475.0.dist-info/METADATA,sha256=1VDKkIy3R5w3VggSNrxOeoJlVUJYsr3RIOaQOfF-AkM,1319
+leadguru_jobs-0.475.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+leadguru_jobs-0.475.0.dist-info/top_level.txt,sha256=rIuw1DqwbnZyeoarBSC-bYeGOhv9mZBs7_afl9q4_JI,9
+leadguru_jobs-0.475.0.dist-info/RECORD,,

lgt_jobs/jobs/send_slack_message.py

@@ -2,8 +2,10 @@ from datetime import datetime, UTC
 from abc import ABC
 from typing import Optional
 from lgt_jobs.lgt_common.lgt_logging import log
-from lgt_jobs.lgt_common.slack_client.web_client import SlackWebClient
-from lgt_jobs.lgt_data.mongo_repository import DedicatedBotRepository, UserContactsRepository
+from lgt_jobs.lgt_common.slack_client.web_client import SlackWebClient, SlackMessageConvertService
+from lgt_jobs.lgt_data.model import ChatMessage
+from lgt_jobs.lgt_data.mongo_repository import DedicatedBotRepository, UserContactsRepository, UserMongoRepository, \
+    ChatRepository
 from pydantic import BaseModel
 from lgt_jobs.basejobs import BaseBackgroundJobData, BaseBackgroundJob
 
@@ -29,6 +31,7 @@ class SendSlackMessageJob(BaseBackgroundJob, ABC):
         if not bot:
             return
 
+        user = UserMongoRepository().get(bot.user_id)
         user_contacts_repository = UserContactsRepository()
         uc = user_contacts_repository.find_one(bot.user_id, sender_id=data.sender_id,
                                                source_id=bot.source.source_id)
@@ -37,13 +40,23 @@ class SendSlackMessageJob(BaseBackgroundJob, ABC):
         user_contacts_repository.update(bot.user_id, uc.sender_id, bot.source.source_id,
                                         scheduled_messages=uc.scheduled_messages)
         slack_client = SlackWebClient(bot.token, bot.cookies)
-        resp = slack_client.im_open(data.sender_id)
-        if not resp['ok']:
-            log.warning(f"Unable to open im with user: {resp}")
+        im = slack_client.im_open(data.sender_id)
+        if not im['ok']:
+            log.warning(f"Unable to open im with user: {data.sender_id}")
             return
 
-        channel_id = resp['channel']['id']
+        channel_id = im['channel']['id']
         if data.files_ids:
-            slack_client.share_files(data.files_ids, channel_id, data.text)
+            resp = slack_client.share_files(data.files_ids, channel_id, data.text)
         else:
-            slack_client.post_message(channel_id, data.text)
+            resp = slack_client.post_message(channel_id, data.text)
+
+        message = resp.get('message') if 'message' in resp \
+            else slack_client.conversation_replies(channel_id, resp['file_msg_ts'])['messages'][0]
+
+        message_model: ChatMessage = SlackMessageConvertService.from_slack_response(user.email, "slack_files",
+                                                                                    bot.token, message, data.sender_id,
+                                                                                    bot.id, user.id,
+                                                                                    slack_client.client.cookies)
+        message_model.viewed = True
+        ChatRepository().upsert_messages([message_model.to_dic()])

lgt_jobs/lgt_data/enums.py

@@ -101,3 +101,8 @@ class FeaturesEnum(str, Enum):
     SCHEDULED_MESSAGES = 'scheduled_messages'
     SOURCES = 'sources'
     ALGORITHMS = 'algorithms'
+
+
+class FeatureOptions(str, Enum):
+    BASIC = 'basic'
+    ADVANCED = 'advanced'

lgt_jobs/lgt_data/model.py

@@ -4,7 +4,7 @@ import json
 from abc import ABC
 from datetime import datetime, UTC
 from typing import Optional, List, Dict
-from .enums import UserRole, SourceType
+from .enums import UserRole, SourceType, FeaturesEnum, FeatureOptions
 from .helpers import get_linkedin_search_contact
 from bson import ObjectId
 
@@ -287,6 +287,8 @@ class UserModel(BaseModel):
         self.slack_users: List[SlackUser] = []
         self.discord_users: List[DiscordUser] = []
         self.verified: bool = False
+        self.subscription_id: ObjectId | None = None
+        self.subscription_expired_at: datetime | None = None
 
     @classmethod
     def from_dic(cls, dic: dict):
@@ -990,4 +992,22 @@ class UsersPage:
     def from_dic(dic: dict):
         users = [UserModel.from_dic(doc) for doc in dic.get('page', [])]
         count = dic.get('count', 0)
-        return UsersPage(users=users, count=count)
+        return UsersPage(users=users, count=count)
+
+
+class Feature(DictionaryModel):
+    display_name: str
+    name: FeaturesEnum
+    description: str | None = None
+    limit: int | None = None
+    options: FeatureOptions | None = None
+
+
+class Subscription(BaseModel):
+    features: list[Feature]
+    duration_days: int
+    name: str
+    price: int
+    limits: int
+    trial: bool
+    updated_at: datetime = None

lgt_jobs/lgt_data/mongo_repository.py

@@ -9,7 +9,7 @@ from lgt_jobs.lgt_data.enums import SourceType
 from lgt_jobs.lgt_data.model import (LeadModel, BaseModel, UserModel, UserResetPasswordModel, BoardModel, BoardedStatus,
                                      DedicatedBotModel, SlackMemberInformation, UserTemplateModel, LinkedinContact,
                                      ExtendedUserLeadModel, UserLeadModel, ExtendedLeadModel, UserContact, ChatMessage,
-                                     GroupedMessagesModel, UserVerificationModel, UsersPage)
+                                     GroupedMessagesModel, UserVerificationModel, UsersPage, Subscription)
 from datetime import datetime, UTC, timedelta
 from collections import OrderedDict
 
@@ -1227,3 +1227,18 @@ class UserVerificationMongoRepository(BaseMongoRepository):
         model.email = email
         model.created_at = datetime.now(UTC)
         return self.collection().insert_one(model.to_dic()).inserted_id
+
+
+class SubscriptionsRepository(BaseMongoRepository):
+    pass
+
+    collection_name = 'subscriptions'
+    model = Subscription
+
+    def find_one(self, **kwargs):
+        pipeline = {}
+        subscription_id = kwargs.get('id')
+        if subscription_id:
+            pipeline['_id'] = to_object_id(subscription_id)
+
+        return Subscription.from_dic(self.collection().find_one(pipeline))