leadguru-jobs 0.638.0__tar.gz → 0.640.0__tar.gz

{leadguru_jobs-0.638.0 → leadguru_jobs-0.640.0}/PKG-INFO

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

{leadguru_jobs-0.638.0 → leadguru_jobs-0.640.0}/leadguru_jobs.egg-info/PKG-INFO

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

leadguru_jobs-0.640.0/leadguru_jobs.egg-info/SOURCES.txt

@@ -0,0 +1,115 @@
+MANIFEST.in
+README.md
+setup.cfg
+setup.py
+leadguru_jobs.egg-info/PKG-INFO
+leadguru_jobs.egg-info/SOURCES.txt
+leadguru_jobs.egg-info/dependency_links.txt
+leadguru_jobs.egg-info/not-zip-safe
+leadguru_jobs.egg-info/requires.txt
+leadguru_jobs.egg-info/top_level.txt
+lgt_jobs/__init__.py
+lgt_jobs/basejobs.py
+lgt_jobs/env.py
+lgt_jobs/main.py
+lgt_jobs/runner.py
+lgt_jobs/simple_job.py
+lgt_jobs/smtp.py
+lgt_jobs/assets/images/arrow.png
+lgt_jobs/assets/images/firework.png
+lgt_jobs/assets/images/lock.png
+lgt_jobs/assets/images/logo.png
+lgt_jobs/assets/images/mail.png
+lgt_jobs/jobs/__init__.py
+lgt_jobs/jobs/analytics.py
+lgt_jobs/jobs/bot_stats_update.py
+lgt_jobs/jobs/chat_history.py
+lgt_jobs/jobs/connect_sources.py
+lgt_jobs/jobs/inbox_leads.py
+lgt_jobs/jobs/load_slack_people.py
+lgt_jobs/jobs/mass_message.py
+lgt_jobs/jobs/send_code.py
+lgt_jobs/jobs/send_slack_message.py
+lgt_jobs/jobs/update_slack_profile.py
+lgt_jobs/jobs/workspace_connect.py
+lgt_jobs/lgt_common/__init__.py
+lgt_jobs/lgt_common/helpers.py
+lgt_jobs/lgt_common/discord_client/__init__.py
+lgt_jobs/lgt_common/discord_client/discord_client.py
+lgt_jobs/lgt_common/discord_client/methods.py
+lgt_jobs/lgt_common/enums/__init__.py
+lgt_jobs/lgt_common/enums/slack_errors.py
+lgt_jobs/lgt_common/pubsub/__init__.py
+lgt_jobs/lgt_common/pubsub/messages.py
+lgt_jobs/lgt_common/pubsub/pubsubfactory.py
+lgt_jobs/lgt_common/slack_client/__init__.py
+lgt_jobs/lgt_common/slack_client/methods.py
+lgt_jobs/lgt_common/slack_client/slack_client.py
+lgt_jobs/lgt_common/slack_client/web_client.py
+lgt_jobs/lgt_data/__init__.py
+lgt_jobs/lgt_data/analytics.py
+lgt_jobs/lgt_data/engine.py
+lgt_jobs/lgt_data/enums.py
+lgt_jobs/lgt_data/helpers.py
+lgt_jobs/lgt_data/model.py
+lgt_jobs/lgt_data/mongo_repository.py
+lgt_jobs/lgt_data/models/__init__.py
+lgt_jobs/lgt_data/models/base.py
+lgt_jobs/lgt_data/models/boards/__init__.py
+lgt_jobs/lgt_data/models/boards/board.py
+lgt_jobs/lgt_data/models/boards/status.py
+lgt_jobs/lgt_data/models/bots/__init__.py
+lgt_jobs/lgt_data/models/bots/base_bot.py
+lgt_jobs/lgt_data/models/bots/bot_info.py
+lgt_jobs/lgt_data/models/bots/dedicated_bot.py
+lgt_jobs/lgt_data/models/chat/__init__.py
+lgt_jobs/lgt_data/models/chat/file.py
+lgt_jobs/lgt_data/models/chat/grouped_messages.py
+lgt_jobs/lgt_data/models/chat/message.py
+lgt_jobs/lgt_data/models/chat/request.py
+lgt_jobs/lgt_data/models/chat/scheduled_message.py
+lgt_jobs/lgt_data/models/contacts/__init__.py
+lgt_jobs/lgt_data/models/contacts/contact.py
+lgt_jobs/lgt_data/models/external/__init__.py
+lgt_jobs/lgt_data/models/external/user_workspace.py
+lgt_jobs/lgt_data/models/external/cloud/__init__.py
+lgt_jobs/lgt_data/models/external/cloud/file.py
+lgt_jobs/lgt_data/models/external/discord/__init__.py
+lgt_jobs/lgt_data/models/external/discord/user.py
+lgt_jobs/lgt_data/models/external/slack/__init__.py
+lgt_jobs/lgt_data/models/external/slack/timezone.py
+lgt_jobs/lgt_data/models/external/slack/user.py
+lgt_jobs/lgt_data/models/leads/__init__.py
+lgt_jobs/lgt_data/models/leads/config.py
+lgt_jobs/lgt_data/models/leads/extended_lead.py
+lgt_jobs/lgt_data/models/leads/lead.py
+lgt_jobs/lgt_data/models/leads/message.py
+lgt_jobs/lgt_data/models/notifications/__init__.py
+lgt_jobs/lgt_data/models/notifications/notification.py
+lgt_jobs/lgt_data/models/notifications/notification_settings.py
+lgt_jobs/lgt_data/models/people/__init__.py
+lgt_jobs/lgt_data/models/people/people.py
+lgt_jobs/lgt_data/models/people/profile.py
+lgt_jobs/lgt_data/models/post/__init__.py
+lgt_jobs/lgt_data/models/post/message.py
+lgt_jobs/lgt_data/models/post/post.py
+lgt_jobs/lgt_data/models/templates/__init__.py
+lgt_jobs/lgt_data/models/templates/template.py
+lgt_jobs/lgt_data/models/user/__init__.py
+lgt_jobs/lgt_data/models/user/feature.py
+lgt_jobs/lgt_data/models/user/general_settings.py
+lgt_jobs/lgt_data/models/user/reset_password.py
+lgt_jobs/lgt_data/models/user/subscription.py
+lgt_jobs/lgt_data/models/user/user.py
+lgt_jobs/lgt_data/models/user/user_follow_ups.py
+lgt_jobs/lgt_data/models/user/user_page.py
+lgt_jobs/lgt_data/models/user/verification.py
+lgt_jobs/lgt_data/models/user_leads/__init__.py
+lgt_jobs/lgt_data/models/user_leads/extended_user_lead.py
+lgt_jobs/lgt_data/models/user_leads/user_lead.py
+lgt_jobs/lgt_data/repositories/__init__.py
+lgt_jobs/lgt_data/repositories/post/__init__.py
+lgt_jobs/lgt_data/repositories/post/posts.py
+lgt_jobs/services/__init__.py
+lgt_jobs/services/web_client.py
+lgt_jobs/templates/new_message.html

{leadguru_jobs-0.638.0 → leadguru_jobs-0.640.0}/lgt_jobs/jobs/bot_stats_update.py

@@ -1,12 +1,12 @@
 import random
 import time
 from abc import ABC
-from datetime import datetime, UTC, timedelta
 from typing import Optional
 from lgt_jobs.lgt_common.discord_client.discord_client import DiscordClient
 from lgt_jobs.lgt_common.slack_client.web_client import SlackWebClient
 from lgt_jobs.lgt_data.enums import SourceType
-from lgt_jobs.lgt_data.model import DedicatedBotModel, Server, Channel, DiscordUser
+from lgt_jobs.lgt_data.models.bots.dedicated_bot import Server, Channel, DedicatedBotModel
+from lgt_jobs.lgt_data.models.external.discord.user import DiscordUser
 from lgt_jobs.lgt_data.mongo_repository import (DedicatedBotRepository, UserMongoRepository, SlackContactUserRepository,
                                                 LeadMongoRepository, UserLeadMongoRepository)
 from pydantic import BaseModel
@@ -33,15 +33,7 @@ class BotStatsUpdateJob(BaseBackgroundJob, ABC):
         bots_rep = DedicatedBotRepository()
         users_rep = UserMongoRepository()
         bot = bots_rep.get_one(id=data.bot_id, include_deleted=True)
-
         user = users_rep.get(bot.user_id)
-        if user.subscription_expired_at.replace(tzinfo=UTC) < (datetime.now(UTC) - timedelta(7)):
-            bot.deleted = True
-            for server in bot.servers:
-                server.deleted = True
-            bots_rep.add_or_update(bot)
-            return
-
         received_messages, filtered_messages = get_bots_aggregated_analytics(bot_ids=[str(bot.id)])
 
         if bot.source.source_type == SourceType.DISCORD:

{leadguru_jobs-0.638.0 → leadguru_jobs-0.640.0}/lgt_jobs/jobs/chat_history.py

@@ -3,7 +3,9 @@ from abc import ABC
 from typing import Optional, List
 import logging as log
 from lgt_jobs.lgt_common.slack_client.web_client import SlackWebClient
-from lgt_jobs.lgt_data.model import ChatMessage, UserContact, DedicatedBotModel
+from lgt_jobs.lgt_data.models.bots.dedicated_bot import DedicatedBotModel
+from lgt_jobs.lgt_data.models.chat.message import ChatMessage
+from lgt_jobs.lgt_data.models.contacts.contact import UserContact
 from lgt_jobs.lgt_data.mongo_repository import UserMongoRepository, DedicatedBotRepository, UserContactsRepository, \
     ChatRepository, UserLeadMongoRepository
 from pydantic import BaseModel
@@ -39,7 +41,6 @@ class LoadChatHistoryJob(BaseBackgroundJob, ABC):
             return
         last_message = None
         last_message_contact = None
-        last_message_bot = None
         contacts_groups = self.contacts_repo.find_grouped_actual_contacts(user.id, spam=False, with_chat_only=False)
         for bot in bots:
             contacts = contacts_groups.get(bot.source.source_id)
@@ -54,12 +55,10 @@ class LoadChatHistoryJob(BaseBackgroundJob, ABC):
                     continue
 
                 if not last_message:
-                    last_message_bot = bot
                     last_message = message
                     last_message_contact = contact
 
                 if message.created_at > last_message.created_at and message.user == contact.sender_id:
-                    last_message_bot = bot
                     last_message = message
                     last_message_contact = contact
 
@@ -67,8 +66,6 @@ class LoadChatHistoryJob(BaseBackgroundJob, ABC):
         has_to_be_notified = (not last_notification or (last_message and last_message.created_at > last_notification))
         if last_message and has_to_be_notified and last_message.user == last_message_contact.sender_id:
             user.notification_settings.incoming_messages.need_to_notify = True
-            user.notification_settings.incoming_messages.attributes = [last_message_contact.sender_id,
-                                                                       last_message_bot.source.source_id]
             UserMongoRepository().set(data.user_id, notification_settings=user.notification_settings.to_dic())
 
     def _get_new_messages(self, contact: UserContact, bot: DedicatedBotModel, slack_chat: List[ChatMessage]):

{leadguru_jobs-0.638.0 → leadguru_jobs-0.640.0}/lgt_jobs/jobs/connect_sources.py

@@ -5,10 +5,13 @@ import logging as log
 from lgt_jobs.jobs.load_slack_people import LoadSlackPeopleJob, LoadSlackPeopleJobData
 from lgt_jobs.basejobs import BaseBackgroundJob, BaseBackgroundJobData
 from lgt_jobs.jobs.bot_stats_update import BotStatsUpdateJob, BotStatsUpdateJobData
+from lgt_jobs.lgt_data.models.base import BaseModel
+from lgt_jobs.lgt_data.models.bots.base_bot import Source
+from lgt_jobs.lgt_data.models.bots.dedicated_bot import DedicatedBotModel, Server
+from lgt_jobs.lgt_data.models.external.slack.user import SlackUser
 from lgt_jobs.runner import BackgroundJobRunner
 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
 
 

{leadguru_jobs-0.638.0 → leadguru_jobs-0.640.0}/lgt_jobs/jobs/inbox_leads.py

@@ -1,8 +1,10 @@
 from abc import ABC
 import logging as log
 from lgt_jobs.lgt_common.slack_client.web_client import SlackWebClient
-from lgt_jobs.lgt_data.model import UserModel, DedicatedBotModel, SlackMemberInformation
-from lgt_jobs.lgt_data.models.message_request import MessageRequest
+from lgt_jobs.lgt_data.models.bots.dedicated_bot import DedicatedBotModel
+from lgt_jobs.lgt_data.models.chat.request import MessageRequest
+from lgt_jobs.lgt_data.models.people.people import SlackMemberInformation
+from lgt_jobs.lgt_data.models.user.user import UserModel
 from lgt_jobs.lgt_data.mongo_repository import UserMongoRepository, DedicatedBotRepository, \
     SlackContactUserRepository, UserContactsRepository, MessageRequestsRepository
 from pydantic import BaseModel
@@ -84,6 +86,8 @@ class InboxLeadsJob(BaseBackgroundJob, ABC):
                     if not sender_request:
                         log.info(f"[InboxLeadsJob]: New message request from {sender_id} for user: {user.email}")
                         requests_repo.upsert(user.id, sender_id, message_request)
+                        user.notification_settings.inbox.need_to_notify = True
+                        UserMongoRepository().set(user.id, notification_settings=user.notification_settings.to_dic())
 
     @staticmethod
     def create_people(slack_profile: dict, dedicated_bot: DedicatedBotModel):

{leadguru_jobs-0.638.0 → leadguru_jobs-0.640.0}/lgt_jobs/jobs/load_slack_people.py

@@ -1,7 +1,8 @@
 from abc import ABC
 from lgt_jobs.lgt_common.enums.slack_errors import SlackErrors
 from lgt_jobs.lgt_common.slack_client.slack_client import SlackClient
-from lgt_jobs.lgt_data.model import Source, SlackMemberInformation
+from lgt_jobs.lgt_data.models.bots.base_bot import Source
+from lgt_jobs.lgt_data.models.people.people import SlackMemberInformation
 from lgt_jobs.lgt_data.mongo_repository import DedicatedBotRepository, SlackContactUserRepository
 import logging as log
 from pydantic import BaseModel

{leadguru_jobs-0.638.0 → leadguru_jobs-0.640.0}/lgt_jobs/jobs/send_slack_message.py

@@ -3,7 +3,7 @@ from abc import ABC
 from typing import Optional
 import logging as log
 from lgt_jobs.lgt_common.slack_client.web_client import SlackWebClient
-from lgt_jobs.lgt_data.model import ChatMessage
+from lgt_jobs.lgt_data.models.chat.message import ChatMessage
 from lgt_jobs.lgt_data.mongo_repository import DedicatedBotRepository, UserContactsRepository, ChatRepository
 from pydantic import BaseModel
 from lgt_jobs.basejobs import BaseBackgroundJobData, BaseBackgroundJob

{leadguru_jobs-0.638.0 → leadguru_jobs-0.640.0}/lgt_jobs/jobs/update_slack_profile.py

@@ -2,7 +2,8 @@ from abc import ABC
 from lgt_jobs.lgt_common.slack_client.slack_client import SlackClient
 from lgt_jobs.lgt_common.slack_client.web_client import SlackFilesClient
 from lgt_jobs.lgt_data.enums import SourceType
-from lgt_jobs.lgt_data.model import Profile, SlackUser
+from lgt_jobs.lgt_data.models.external.slack.user import SlackUser
+from lgt_jobs.lgt_data.models.people.profile import Profile
 from lgt_jobs.lgt_data.mongo_repository import UserMongoRepository, DedicatedBotRepository
 from pydantic import BaseModel
 from lgt_jobs.basejobs import BaseBackgroundJobData, BaseBackgroundJob

{leadguru_jobs-0.638.0 → leadguru_jobs-0.640.0}/lgt_jobs/jobs/workspace_connect.py

@@ -4,7 +4,9 @@ import requests
 from lgt_jobs.lgt_common.helpers import update_credentials
 from lgt_jobs.lgt_common.slack_client.web_client import SlackWebClient
 from lgt_jobs.lgt_data.enums import StatusConnection, SourceType
-from lgt_jobs.lgt_data.model import UserWorkspace, SlackUser, DedicatedBotModel
+from lgt_jobs.lgt_data.models.bots.dedicated_bot import DedicatedBotModel
+from lgt_jobs.lgt_data.models.external.slack.user import SlackUser
+from lgt_jobs.lgt_data.models.external.user_workspace import UserWorkspace
 from lgt_jobs.lgt_data.mongo_repository import UserMongoRepository, DedicatedBotRepository
 from lgt_jobs.basejobs import BaseBackgroundJobData, BaseBackgroundJob
 from lgt_jobs.jobs.bot_stats_update import BotStatsUpdateJob, BotStatsUpdateJobData

leadguru_jobs-0.640.0/lgt_jobs/lgt_data/models/base.py

@@ -0,0 +1,29 @@
+import copy
+from abc import ABC
+from datetime import datetime, UTC
+
+
+class DictionaryModel(ABC):
+    @classmethod
+    def from_dic(cls, dic: dict):
+        if not dic:
+            return None
+
+        model = cls()
+        for k, v in dic.items():
+            setattr(model, k, v)
+
+        if '_id' in dic:
+            setattr(model, 'id', dic['_id'])
+
+        return model
+
+    def to_dic(self):
+        result = copy.deepcopy(self.__dict__)
+        return result
+
+
+class BaseModel(DictionaryModel):
+    def __init__(self):
+        self.id = None
+        self.created_at = datetime.now(UTC)

leadguru_jobs-0.640.0/lgt_jobs/lgt_data/models/boards/board.py

@@ -0,0 +1,39 @@
+import copy
+from typing import List
+
+from lgt_jobs.lgt_data.models.base import BaseModel
+from lgt_jobs.lgt_data.models.boards.status import BoardedStatus
+
+
+class BoardModel(BaseModel):
+    pass
+
+    def __init__(self):
+        super().__init__()
+        self.name = None
+        self.user_id = None
+        self.statuses: List[BoardedStatus] = []
+        self.is_primary = None
+        self.default = False
+
+    @classmethod
+    def from_dic(cls, dic: dict):
+        if not dic:
+            return None
+
+        model = BoardModel()
+        for k, v in dic.items():
+            setattr(model, k, v)
+
+        model.id = dic.get('_id')
+        model.statuses = [BoardedStatus.from_dic(status) for status in dic.get('statuses', [])]
+        return model
+
+    def to_dic(self):
+        result = copy.deepcopy(self.__dict__)
+        result["statuses"] = [BoardedStatus.to_dic(status) for status in self.statuses]
+
+        for status in result['statuses']:
+            status['board_id'] = result['id']
+
+        return result

leadguru_jobs-0.640.0/lgt_jobs/lgt_data/models/boards/status.py

@@ -0,0 +1,14 @@
+from lgt_jobs.lgt_data.models.base import DictionaryModel
+
+
+class BoardedStatus(DictionaryModel):
+    pass
+
+    def __init__(self):
+        self.id = None
+        self.name = None
+        self.order = 0
+        self.is_primary = False
+        self.default = False
+        self.user_leads = 0
+        self.collapsed = False

leadguru_jobs-0.640.0/lgt_jobs/lgt_data/models/bots/base_bot.py

@@ -0,0 +1,44 @@
+from datetime import datetime
+from typing import List
+
+from lgt_jobs.lgt_data.enums import SourceType
+from lgt_jobs.lgt_data.models.base import BaseModel, DictionaryModel
+
+
+class Credentials(BaseModel):
+    def __init__(self):
+        super().__init__()
+        self.token = None
+        self.cookies = None
+        self.invalid_creds = False
+
+
+class Source(DictionaryModel):
+    def __init__(self):
+        self.source_type: SourceType | None = None
+        self.source_name: str | None = None
+        self.source_id = None
+
+
+class BaseBotModel(Credentials):
+    def __init__(self):
+        super().__init__()
+        self.created_by = None
+        self.user_name = None
+        self.slack_url = None
+        self.registration_link = None
+        self.channels = None
+        self.connected_channels = None
+        self.channels_users = None
+        self.users_count = None
+        self.recent_messages: List[str] = []
+        self.icon = None
+        self.active_channels = {}
+        self.paused_channels = []
+        self.source: Source | None = None
+        self.two_factor_required: bool = False
+        self.banned: bool = False
+        self.associated_user = None
+        self.type: SourceType | None = None
+        self.deleted = False
+        self.deactivated_at: datetime | None = None

leadguru_jobs-0.640.0/lgt_jobs/lgt_data/models/bots/bot_info.py

@@ -0,0 +1,33 @@
+import copy
+
+
+class BotInfo:
+    def __init__(self):
+        self.id = None
+        self.invalid_creds: bool | None = False
+        self.source = None
+        self.banned: bool | None = False
+        self.user_name: str = ''
+        self.associated_user: str | None = ''
+        self.deleted: bool = False
+        self.two_factor_required = False
+
+    @classmethod
+    def from_dic(cls, dic: dict):
+        if not dic:
+            return None
+
+        model = BotInfo()
+        for k, v in dic.items():
+            if hasattr(model, k):
+                setattr(model, k, v)
+
+        if '_id' in dic:
+            setattr(model, 'id', dic['_id'])
+
+        return model
+
+    def to_dic(self):
+        result = copy.deepcopy(self.__dict__)
+        return result
+    

leadguru_jobs-0.640.0/lgt_jobs/lgt_data/models/bots/dedicated_bot.py

@@ -0,0 +1,90 @@
+import copy
+from datetime import datetime, UTC
+from typing import Optional, List
+
+from bson import ObjectId
+
+from lgt_jobs.lgt_data.models.base import DictionaryModel
+from lgt_jobs.lgt_data.models.bots.base_bot import BaseBotModel, Source
+
+
+class DedicatedBotModel(BaseBotModel):
+    def __init__(self):
+        super().__init__()
+        self.user_id: ObjectId | None = None
+        self.updated_at: Optional[datetime] = datetime.now(UTC)
+        self.servers: List[Server] = []
+        self.state = 0
+        self.source: Source | None = None
+
+    def to_dic(self):
+        result = copy.deepcopy(self.__dict__)
+
+        if result.get('source'):
+            result['source'] = Source.to_dic(result.get('source'))
+
+        result['servers'] = [Server.to_dic(server) for server in self.servers]
+        return result
+
+    @classmethod
+    def from_dic(cls, dic: dict):
+        if not dic:
+            return None
+
+        model: DedicatedBotModel = cls()
+        for k, v in dic.items():
+            setattr(model, k, v)
+
+        if '_id' in dic:
+            setattr(model, 'id', dic['_id'])
+
+        model.source = Source.from_dic(dic.get("source"))
+        model.servers = [Server.from_dic(server) for server in dic.get("servers", [])]
+        return model
+
+
+class Server:
+    pass
+
+    def __init__(self):
+        self.id = None
+        self.name = None
+        self.channels: List[Channel] = []
+        self.icon = None
+        self.active = False
+        self.deleted = False
+        self.subscribers = 0
+        self.approximate_member_count = 0
+        self.messages_received: int = 0
+        self.messages_filtered: int = 0
+
+    @classmethod
+    def from_dic(cls, dic: dict):
+        if not dic:
+            return None
+
+        model = cls()
+        for k, v in dic.items():
+            if hasattr(model, k):
+                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):
+        result = copy.deepcopy(self.__dict__)
+        result['channels'] = [Channel.to_dic(channel) for channel in self.channels]
+        return result
+
+
+class Channel(DictionaryModel):
+    pass
+
+    def __init__(self):
+        self.id = None
+        self.name = None
+        self.type = None
+        self.is_member = True
+        self.active = True
+        self.subscribers = 0

leadguru_jobs-0.640.0/lgt_jobs/lgt_data/models/chat/file.py

@@ -0,0 +1,8 @@
+from lgt_jobs.lgt_data.models.base import DictionaryModel
+
+
+class LeadGuruFile(DictionaryModel):
+    id: str = None
+    blob_path: str
+    content_type: str
+    file_name: str = None

leadguru_jobs-0.640.0/lgt_jobs/lgt_data/models/chat/grouped_messages.py

@@ -0,0 +1,17 @@
+from typing import List
+
+from lgt_jobs.lgt_data.models.chat.message import ChatMessage
+
+
+class GroupedMessagesModel:
+    messages: List[ChatMessage] = []
+
+    @classmethod
+    def from_dic(cls, dic: dict):
+        if not dic:
+            return None
+
+        model = cls()
+        model.messages = [ChatMessage.from_dic(message) for message in dic.get('messages', [])]
+        return model
+    

leadguru_jobs-0.640.0/lgt_jobs/lgt_data/models/chat/message.py

@@ -0,0 +1,85 @@
+import copy
+from datetime import datetime
+
+from bson import ObjectId
+
+from lgt_jobs.lgt_data.models.bots.dedicated_bot import DedicatedBotModel
+from lgt_jobs.lgt_data.models.chat.file import LeadGuruFile
+
+
+class ChatMessage:
+    bot_id: ObjectId
+    user_id: ObjectId
+    sender_id: str
+    text: str
+    user: str
+    id: str
+    viewed: bool
+    files: list
+    attachments: list[dict] | None
+    created_at: datetime | None
+    source_id: str | None
+
+    class SlackFileModel:
+        def __init__(self):
+            self.id = None
+            self.name = None
+            self.title = None
+            self.filetype = None
+            self.size = 0
+            self.mimetype = None
+            self.download_url = None
+
+        def to_dic(self):
+            result = copy.deepcopy(self.__dict__)
+            return result
+
+    def __init__(self):
+        self.viewed = False
+        self.text: str = ''
+        self.created_at: datetime
+        self.user = ''
+        self.id = ''
+        self.files = []
+        self.attachments = []
+
+    def to_dic(self):
+        result = copy.deepcopy(self.__dict__)
+        result['files'] = [x.to_dic() for x in result.get('files', [])]
+        return result
+
+    @classmethod
+    def from_dic(cls, dic: dict):
+        if not dic:
+            return None
+        model = cls()
+        for k, v in dic.items():
+            setattr(model, k, v)
+        return model
+
+    @classmethod
+    def from_slack_response(cls, bot: DedicatedBotModel, message_data: dict, sender_id: str):
+        model = cls()
+        model.sender_id = sender_id
+        model.bot_id = bot.id
+        model.text = message_data.get('text', '')
+        model.user = message_data.get('user', '')
+        model.id = message_data.get('ts', '')
+        model.attachments = message_data.get('attachments', [])
+        model.files = []
+        model.user_id = bot.user_id
+        model.source_id = bot.source.source_id
+        if 'files' in message_data:
+            for file in message_data.get('files'):
+                if file.get('mode') != "tombstone" and file.get('url_private_download'):
+                    leadguru_file = LeadGuruFile()
+                    leadguru_file.id = file['id']
+                    leadguru_file.content_type = file['mimetype']
+                    leadguru_file.file_name = file['name']
+                    leadguru_file.blob_path = f'slack_files/{bot.user_name}/slack_files/{file["id"]}'
+                    model.files.append(leadguru_file)
+
+        js_ticks = int(model.id.split('.')[0] + model.id.split('.')[1][3:])
+        model.created_at = datetime.fromtimestamp(js_ticks / 1000.0)
+        return model
+

leadguru_jobs-0.638.0/lgt_jobs/lgt_data/models/message_request.py → leadguru_jobs-0.640.0/lgt_jobs/lgt_data/models/chat/request.py

@@ -1,4 +1,5 @@
-from lgt_jobs.lgt_data.model import ChatMessage, DedicatedBotModel
+from lgt_jobs.lgt_data.models.bots.dedicated_bot import DedicatedBotModel
+from lgt_jobs.lgt_data.models.chat.message import ChatMessage
 
 
 class MessageRequest(ChatMessage):