leadguru-jobs 0.606.0__tar.gz → 0.608.0__tar.gz

{leadguru_jobs-0.606.0 → leadguru_jobs-0.608.0}/PKG-INFO

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

{leadguru_jobs-0.606.0 → leadguru_jobs-0.608.0}/leadguru_jobs.egg-info/PKG-INFO

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

{leadguru_jobs-0.606.0 → leadguru_jobs-0.608.0}/lgt_jobs/jobs/update_slack_profile.py

@@ -2,7 +2,7 @@ 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 UserModel
+from lgt_jobs.lgt_data.model import Profile, SlackUser
 from lgt_jobs.lgt_data.mongo_repository import UserMongoRepository, DedicatedBotRepository
 from pydantic import BaseModel
 from lgt_jobs.basejobs import BaseBackgroundJobData, BaseBackgroundJob
@@ -21,31 +21,37 @@ class UpdateExternalUserProfileJob(BaseBackgroundJob, ABC):
 
     def exec(self, data: UpdateExternalUserProfileJobData):
         user = UserMongoRepository().get(data.user_id)
+        main_profile = None
+        active_users = [slack_user for slack_user in user.slack_users if not slack_user.deleted]
+        if active_users and active_users[0].profile.main:
+            main_profile = active_users[0].profile
         bots = DedicatedBotRepository().get_all(user_id=data.user_id, only_valid=True, include_deleted=False)
         for bot in bots:
+            current_user: SlackUser = next(filter(lambda x: x.email == bot.user_name, user.slack_users), None)
+            profile = main_profile if main_profile else current_user.profile
             if bot.source.source_type == SourceType.SLACK:
                 slack = SlackClient(bot.token, bot.cookies)
                 log.info(f'Updating profile in {bot.source.source_name}:{bot.source.source_id}')
-                UpdateExternalUserProfileJob.__update_slack_profile(user, slack)
+                UpdateExternalUserProfileJob.__update_slack_profile(slack, profile)
 
     @staticmethod
-    def __update_slack_profile(user: UserModel, slack: SlackClient):
-        slack.update_profile(user.slack_profile.to_dic())
-        if user.photo_url:
-            photo_url = SlackFilesClient().get_file_url(user.photo_url)
+    def __update_slack_profile(slack: SlackClient, profile: Profile = None):
+        slack.update_profile(profile.to_dic())
+        if profile.photo_url:
+            photo_url = SlackFilesClient().get_file_url(profile.photo_url)
             slack.update_profile_photo(photo_url)
 
-        profile = slack.get_team_profile()
+        team_profile = slack.get_team_profile()
         title_section_id = None
         title_field_id = None
         skype_section_id = None
-        for field_data in profile.get('profile', {}).get('fields', []):
+        for field_data in team_profile.get('profile', {}).get('fields', []):
             if field_data.get('field_name') == 'title':
                 title_section_id = field_data.get('section_id')
                 title_field_id = field_data.get('id')
                 break
         if title_field_id and title_section_id:
-            for section_data in profile.get('profile', {}).get('sections', []):
+            for section_data in team_profile.get('profile', {}).get('sections', []):
                 if section_data.get('section_type') == 'additional_info':
                     skype_section_id = section_data.get('id')
                     break
@@ -53,7 +59,7 @@ class UpdateExternalUserProfileJob(BaseBackgroundJob, ABC):
             auth = slack.test_auth().json()
             user_id = auth.get('user_id')
             title_element_id = title_field_id.replace(title_field_id[:2], 'Pe')
-            response = slack.update_section(user_id, title_section_id, title_element_id, user.slack_profile.title)
+            response = slack.update_section(user_id, title_section_id, title_element_id, profile.title)
             sections = response['result']['data']['setProfileSection']['profileSections']
             elements = []
             for section in sections:
@@ -65,4 +71,4 @@ class UpdateExternalUserProfileJob(BaseBackgroundJob, ABC):
                 if element['label'] == 'Skype':
                     skype_element_id = element['elementId']
                     break
-            slack.update_section(user_id, skype_section_id, skype_element_id, user.slack_profile.skype)
+            slack.update_section(user_id, skype_section_id, skype_element_id, profile.skype)

{leadguru_jobs-0.606.0 → leadguru_jobs-0.608.0}/lgt_jobs/lgt_common/discord_client/discord_client.py

@@ -43,6 +43,37 @@ class DiscordClient:
             self.__log_error(response)
         return response.json()
 
+    def patch_current_user(self, username: str, avatar: str = None) -> dict:
+        payload = {'username': username, 'avatar': avatar}
+        response = requests.patch(f"{self.base_url}{DiscordMethods.USER.value}", headers=self.headers, json=payload)
+        if response.status_code != 200:
+            self.__log_error(response)
+        return response.json()
+
+    def patch_current_user_in_guild(self, guild_id: str, username: str, avatar: str = None) -> dict:
+        payload = {'username': username, 'avatar': avatar}
+        response = requests.patch(f"{self.base_url}{DiscordMethods.guild_user(guild_id)}",
+                                  headers=self.headers, json=payload)
+        if response.status_code != 200:
+            self.__log_error(response)
+        return response.json()
+
+    def patch_current_user_bio(self, bio: str) -> dict:
+        payload = {'bio': bio}
+        response = requests.patch(f"{self.base_url}{DiscordMethods.USER_PROFILE.value}", headers=self.headers,
+                                  json=payload)
+        if response.status_code != 200:
+            self.__log_error(response)
+        return response.json()
+
+    def patch_user_bio_in_guild(self, guild_id: str, bio: str) -> dict:
+        payload = {'bio': bio}
+        response = requests.patch(f"{self.base_url}{DiscordMethods.guild_bio(guild_id)}",
+                                  headers=self.headers, json=payload)
+        if response.status_code != 200:
+            self.__log_error(response)
+        return response.json()
+
     def get_channels(self, guild_id: str) -> list | dict:
         response = requests.get(f"{self.base_url}{DiscordMethods.guild_channels(guild_id)}?with_counts=true",
                                 headers=self.headers)

{leadguru_jobs-0.606.0 → leadguru_jobs-0.608.0}/lgt_jobs/lgt_common/discord_client/methods.py

@@ -5,12 +5,21 @@ class DiscordMethods(str, Enum):
     USER_GUILDS = 'users/@me/guilds'
     USER_DMS = 'users/@me/channels'
     USER = 'users/@me'
+    USER_PROFILE = 'users/@me/profile'
     LOGIN = 'auth/login'
 
     @staticmethod
     def guild_channels(guild_id: str):
         return f'guilds/{guild_id}/channels'
 
+    @staticmethod
+    def guild_user(guild_id: str):
+        return f'guilds/{guild_id}/members/@me'
+
+    @staticmethod
+    def guild_bio(guild_id: str):
+        return f'guilds/{guild_id}/profile/@me'
+
     @staticmethod
     def channels_invites(channel_id: str):
         return f'channels/{channel_id}/invites'

{leadguru_jobs-0.606.0 → leadguru_jobs-0.608.0}/lgt_jobs/lgt_data/model.py

@@ -272,8 +272,6 @@ class UserModel(BaseModel):
         self.company_description: str = ''
         self.position: str = ''
         self.new_message_notified_at: Optional[datetime] = None
-        self.photo_url: str = ''
-        self.slack_profile = SlackProfile()
         self.leads_limit: Optional[int] = None
         self.leads_proceeded: Optional[int] = None
         self.leads_filtered: Optional[int] = None
@@ -307,7 +305,7 @@ class UserModel(BaseModel):
         if '_id' in dic:
             setattr(model, 'id', dic['_id'])
 
-        model.slack_profile = SlackProfile.from_dic(dic.get('slack_profile'))
+        model.slack_profile = Profile.from_dic(dic.get('slack_profile'))
         model.slack_users = [SlackUser.from_dic(user) for user in dic.get('slack_users', [])]
         model.discord_users = [DiscordUser.from_dic(user) for user in dic.get('discord_users', [])]
         return model
@@ -346,13 +344,17 @@ class DiscordUser(DictionaryModel):
         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):
+        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
@@ -365,6 +367,7 @@ class DiscordUser(DictionaryModel):
             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
 
 
@@ -378,6 +381,7 @@ class SlackUser:
         self.status = None
         self.workspaces: List[UserWorkspace] = []
         self.deleted = False
+        self.profile: Profile | None = None
 
     def to_dic(self):
         result = copy.deepcopy(self.__dict__)
@@ -385,6 +389,9 @@ class SlackUser:
         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
@@ -397,6 +404,7 @@ class SlackUser:
             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
 
 
@@ -730,7 +738,7 @@ class BoardedStatus:
         return model
 
 
-class SlackProfile:
+class Profile:
     pass
 
     def __init__(self):
@@ -740,6 +748,8 @@ class SlackProfile:
         self.display_name = ''
         self.real_name = ''
         self.email = ''
+        self.photo_url = ''
+        self.main = False
 
     def to_dic(self):
         return copy.deepcopy(self.__dict__)
@@ -754,7 +764,7 @@ class SlackProfile:
         return model
 
 
-class SlackMemberInformation(BaseModel, SlackProfile):
+class SlackMemberInformation(BaseModel, Profile):
     workspace: str
     sender_id: str
     images: dict