leadguru-jobs 0.638.0__py3-none-any.whl → 0.640.0__py3-none-any.whl

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)

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

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

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

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
+    

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

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

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
+    

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
+

lgt_jobs/lgt_data/models/{message_request.py → 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):

lgt_jobs/lgt_data/models/chat/scheduled_message.py

@@ -0,0 +1,14 @@
+from datetime import datetime
+
+from bson import ObjectId
+
+from lgt_jobs.lgt_data.models.base import BaseModel
+
+
+class ScheduledMessage(BaseModel):
+    post_at: datetime
+    bot_id: ObjectId
+    user_id: ObjectId
+    sender_id: str
+    text: str | None
+    files: list

lgt_jobs/lgt_data/models/contacts/contact.py

@@ -0,0 +1,22 @@
+from datetime import datetime
+from typing import Optional
+
+from bson import ObjectId
+
+from lgt_jobs.lgt_data.models.chat.message import ChatMessage
+from lgt_jobs.lgt_data.models.people.people import SlackMemberInformation
+
+
+class UserContact(SlackMemberInformation):
+    chat_id: str
+    user_id: ObjectId
+    source_id: str
+    last_message_at: Optional[datetime]
+
+    @classmethod
+    def from_dic(cls, dic: dict):
+        if not dic:
+            return None
+        model: UserContact | None = super().from_dic(dic)
+        model.chat_history = [ChatMessage.from_dic(message) for message in dic.get('chat_history', [])]
+        return model

lgt_jobs/lgt_data/models/external/cloud/file.py

@@ -0,0 +1,18 @@
+from bson import ObjectId
+
+from lgt_jobs.lgt_data.models.base import BaseModel
+
+
+class CloudFileModel(BaseModel):
+    blob_path: str
+    public_url: str
+    file_name: str
+
+    def __init__(self, blob_path: str, public_url: str, file_name: str):
+        super().__init__()
+        if not self.id:
+            self.id = str(ObjectId())
+        self.blob_path = blob_path
+        self.public_url = public_url
+        self.file_name = file_name
+        

lgt_jobs/lgt_data/models/external/discord/user.py

@@ -0,0 +1,45 @@
+import copy
+from datetime import datetime, UTC
+from typing import List
+
+from lgt_jobs.lgt_data.models.base import DictionaryModel
+from lgt_jobs.lgt_data.models.external.user_workspace import UserWorkspace
+from lgt_jobs.lgt_data.models.people.profile import Profile
+
+
+class DiscordUser(DictionaryModel):
+    pass
+
+    def __init__(self):
+        super().__init__()
+        self.created_at = datetime.now(UTC)
+        self.login = ''
+        self.captcha_key = ''
+        self.status = None
+        self.workspaces: List[UserWorkspace] = []
+        self.deleted = False
+        self.profile: Profile | None = None
+
+    def to_dic(self):
+        result = copy.deepcopy(self.__dict__)
+
+        if result.get('workspaces'):
+            result['workspaces'] = [ws.__dict__ for ws in result.get('workspaces')]
+
+        if result.get('profile'):
+            result['profile'] = result.get('profile').__dict__
+
+        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)
+
+        model.workspaces = [UserWorkspace.from_dic(ws) for ws in dic.get('workspaces', [])]
+        model.profile = Profile.from_dic(dic.get('profile'))
+        return model

lgt_jobs/lgt_data/models/external/slack/timezone.py

@@ -0,0 +1,8 @@
+from typing import Optional
+
+
+class SlackTimeZone:
+    tz: Optional[str]
+    tz_label: Optional[str]
+    tz_offset: Optional[int]
+    

lgt_jobs/lgt_data/models/external/slack/user.py

@@ -0,0 +1,43 @@
+import copy
+from datetime import datetime, UTC
+from typing import List
+
+from lgt_jobs.lgt_data.models.external.user_workspace import UserWorkspace
+from lgt_jobs.lgt_data.models.people.profile import Profile
+
+
+class SlackUser:
+    pass
+
+    def __init__(self):
+        self.created_at = datetime.now(UTC)
+        self.cookies = {}
+        self.email = ''
+        self.status = None
+        self.workspaces: List[UserWorkspace] = []
+        self.deleted = False
+        self.profile: Profile | None = None
+
+    def to_dic(self):
+        result = copy.deepcopy(self.__dict__)
+
+        if result.get('workspaces', None):
+            result['workspaces'] = [ws.__dict__ for ws in result.get('workspaces')]
+
+        if result.get('profile'):
+            result['profile'] = result.get('profile').__dict__
+
+        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)
+
+        model.workspaces = [UserWorkspace.from_dic(ws) for ws in dic.get('workspaces', [])]
+        model.profile = Profile.from_dic(dic.get('profile'))
+        return model

lgt_jobs/lgt_data/models/external/user_workspace.py

@@ -0,0 +1,32 @@
+class UserWorkspace:
+    pass
+
+    def __init__(self):
+        super().__init__()
+        self.id = ''
+        self.name = ''
+        self.url = ''
+        self.domain = ''
+        self.active_users = ''
+        self.profile_photos = []
+        self.associated_user = ''
+        self.magic_login_url = ''
+        self.magic_login_code = ''
+        self.user_email = ''
+        self.user_type = ''
+        self.variant = ''
+        self.token = ''
+        self.icon = ''
+        self.two_factor_required = False
+
+    @classmethod
+    def from_dic(cls, dic: dict):
+        if not dic:
+            return None
+
+        model: UserWorkspace = cls()
+        for k, v in dic.items():
+            setattr(model, k, v)
+
+        model.icon = dic.get('icon_88', "")
+        return model

lgt_jobs/lgt_data/models/leads/config.py

@@ -0,0 +1,13 @@
+from lgt_jobs.lgt_data.models.base import DictionaryModel
+
+
+class BaseConfig(DictionaryModel):
+    def __init__(self):
+        self.owner = None
+        self.id = None
+
+
+class Config(BaseConfig):
+    def __init__(self):
+        super().__init__()
+        self.name = None

lgt_jobs/lgt_data/models/leads/extended_lead.py

@@ -0,0 +1,46 @@
+from typing import List
+
+from lgt_jobs.lgt_data.models.bots.bot_info import BotInfo
+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.models.leads.lead import LeadModel
+from lgt_jobs.lgt_data.models.people.people import SlackMemberInformation
+from lgt_jobs.lgt_data.models.user_leads.user_lead import UserLeadModel
+
+
+class ExtendedLeadModel(LeadModel):
+    def __init__(self):
+        super().__init__()
+        self.previous_publications = []
+        self.last_conversation: List[ChatMessage] = []
+        self.contact: SlackMemberInformation | None = None
+        self.deleted = False
+        self.user_lead: UserLeadModel | None = None
+        self.dedicated: bool = False
+        self.bots: List[BotInfo] = []
+        self.user_contact: UserContact | None = None
+        self.paid: bool = False
+        self.hidden_by_user: bool = False
+
+    @classmethod
+    def from_dic(cls, dic: dict):
+        if not dic:
+            return None
+
+        result: ExtendedLeadModel | None = LeadModel.from_dic(dic)
+        if not result:
+            return None
+
+        result.contact = SlackMemberInformation.from_dic(dic.get('contact'))
+        result.user_contact = UserContact.from_dic(dic.get('user_contact'))
+        result.previous_publications = [LeadModel.from_dic(lead) for lead in dic.get('previous_publications', [])]
+        result.user_lead = UserLeadModel.from_dic(dic.get('user_lead'))
+        result.last_conversation = [ChatMessage.from_dic(message) for message in dic.get('last_conversation', [])]
+        result.bots = [BotInfo.from_dic(bot) for bot in dic.get('bots', [])]
+        return result
+
+    def to_csv(self, board_name: str) -> List[str]:
+        return [self.message.source, self.contact.real_name, self.contact.title, self.contact.email,
+                self.notes, board_name, self.status,
+                self.followup_date.strftime("%d.%m.%Y %H:%M") if self.followup_date else "",
+                self.message.message.replace('\n', ' ').strip()]