leadguru-jobs 0.473.0__tar.gz → 0.475.0__tar.gz

{leadguru_jobs-0.473.0 → leadguru_jobs-0.475.0}/PKG-INFO

@@ -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 → leadguru_jobs-0.475.0}/leadguru_jobs.egg-info/PKG-INFO

@@ -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 → leadguru_jobs-0.475.0}/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()])

{leadguru_jobs-0.473.0 → leadguru_jobs-0.475.0}/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'

{leadguru_jobs-0.473.0 → leadguru_jobs-0.475.0}/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

{leadguru_jobs-0.473.0 → leadguru_jobs-0.475.0}/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))