leadguru-jobs 0.706.0__tar.gz → 0.708.0__tar.gz

{leadguru_jobs-0.706.0 → leadguru_jobs-0.708.0}/PKG-INFO

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

{leadguru_jobs-0.706.0 → leadguru_jobs-0.708.0}/leadguru_jobs.egg-info/PKG-INFO

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

{leadguru_jobs-0.706.0 → leadguru_jobs-0.708.0}/lgt_jobs/jobs/workspace_connect.py

@@ -42,78 +42,84 @@ class ConnectSlackAccountJob(BaseBackgroundJob, ABC):
             current_user.slack_users.append(slack_user)
         slack_user.status = StatusConnection.IN_PROGRESS
 
-        client = SlackWebClient('')
-        code_confirmed_response = client.confirm_code(data.slack_email, data.code, data.user_agent)
-        if code_confirmed_response.status_code != 200:
-            log.warning(f'Unable to confirm code due to error: {code_confirmed_response.content}')
-            slack_user.status = StatusConnection.FAILED
-            UserMongoRepository().set(current_user.id, slack_users=[user.to_dic() for user in current_user.slack_users])
-            return
-
-        code_confirmed = code_confirmed_response.json().get('ok', False)
-        if not code_confirmed:
-            slack_user.status = StatusConnection.FAILED
-            UserMongoRepository().set(current_user.id, slack_users=[user.to_dic() for user in current_user.slack_users])
-            log.warning(f'Invalid code')
-            return
-
-        slack_user.cookies = client.client.cookies = code_confirmed_response.cookies.get_dict()
-
-        workspaces_response = None
-        attempt = 1
-        while attempt <= 5:
-            workspaces_response = client.find_workspaces(data.user_agent)
-            if workspaces_response.status_code != 200:
+        try:
+            client = SlackWebClient('')
+            code_confirmed_response = client.confirm_code(data.slack_email, data.code, data.user_agent)
+            if code_confirmed_response.status_code != 200:
+                log.warning(f'Unable to confirm code due to error: {code_confirmed_response.content}')
                 slack_user.status = StatusConnection.FAILED
-                log.warning(f'Attempt: {attempt}. Unable to get workspaces due to error: {workspaces_response.content}')
-                attempt += 1
-                time.sleep(60)
-            if not workspaces_response.json().get('ok', False):
+                UserMongoRepository().set(current_user.id, slack_users=[user.to_dic() for user in current_user.slack_users])
+                return
+
+            code_confirmed = code_confirmed_response.json().get('ok', False)
+            if not code_confirmed:
                 slack_user.status = StatusConnection.FAILED
-                log.warning(f'Attempt: {attempt}. Unable to get workspaces due to error: {workspaces_response.json()}')
-                attempt += 1
-                time.sleep(60)
-            else:
-                slack_user.status = StatusConnection.IN_PROGRESS
-                break
+                UserMongoRepository().set(current_user.id, slack_users=[user.to_dic() for user in current_user.slack_users])
+                log.warning(f'Invalid code')
+                return
+
+            slack_user.cookies = client.client.cookies = code_confirmed_response.cookies.get_dict()
+
+            workspaces_response = None
+            attempt = 1
+            while attempt <= 5:
+                workspaces_response = client.find_workspaces(data.user_agent)
+                if workspaces_response.status_code != 200:
+                    slack_user.status = StatusConnection.FAILED
+                    log.warning(f'Attempt: {attempt}. Unable to get workspaces due to error: {workspaces_response.content}')
+                    attempt += 1
+                    time.sleep(60)
+                if not workspaces_response.json().get('ok', False):
+                    slack_user.status = StatusConnection.FAILED
+                    log.warning(f'Attempt: {attempt}. Unable to get workspaces due to error: {workspaces_response.json()}')
+                    attempt += 1
+                    time.sleep(60)
+                else:
+                    slack_user.status = StatusConnection.IN_PROGRESS
+                    break
 
-        UserMongoRepository().set(current_user.id, slack_users=[user.to_dic() for user in current_user.slack_users])
+            UserMongoRepository().set(current_user.id, slack_users=[user.to_dic() for user in current_user.slack_users])
 
-        if attempt > 5:
+            if attempt > 5:
+                return
+
+            log.info(f'{slack_user.email}: got workspaces data {workspaces_response.json()}')
+            user_workspaces = next((user for user in workspaces_response.json()['current_teams']
+                                    if user['email'] == data.slack_email), {}).get('teams', [])
+            user_workspaces = [UserWorkspace.from_dic(ws) for ws in user_workspaces]
+            user_workspaces = sorted(user_workspaces, key=lambda ws: ws.domain)
+
+            session = requests.Session()
+            session.cookies = code_confirmed_response.cookies
+            session.headers.update({'User-Agent': data.user_agent})
+            log.info(f'{slack_user.email}: started getting of tokens for {len(user_workspaces)} workspaces')
+            for workspace in user_workspaces:
+                if not workspace.magic_login_code:
+                    continue
+                login_url = f"https://app.slack.com/t/{workspace.domain}/login/{workspace.magic_login_code}"
+                magic_response = session.post(login_url, cookies=session.cookies, headers=session.headers)
+                content = magic_response.content.decode('utf-8')
+                start_token_index = content.find("xox")
+                sliced_content = content[start_token_index:]
+                end_token_index = sliced_content.find('"')
+                token = sliced_content[:end_token_index]
+                workspace.magic_login_url = login_url
+                workspace.token = token
+                workspace.domain = workspace.domain
+                time.sleep(random.uniform(0.2, 0.5))
+
+            slack_user.cookies = session.cookies.get_dict()
+            slack_user.workspaces = user_workspaces
+            slack_user.status = StatusConnection.COMPLETE
+
+            slack_users_dict = [user.to_dic() for user in current_user.slack_users]
+            UserMongoRepository().set(current_user.id, slack_users=slack_users_dict)
+
+        except:
+            slack_user.status = StatusConnection.FAILED
+            UserMongoRepository().set(current_user.id, slack_users=[user.to_dic() for user in current_user.slack_users])
             return
 
-        log.info(f'{slack_user.email}: got workspaces data {workspaces_response.json()}')
-        user_workspaces = next((user for user in workspaces_response.json()['current_teams']
-                                if user['email'] == data.slack_email), {}).get('teams', [])
-        user_workspaces = [UserWorkspace.from_dic(ws) for ws in user_workspaces]
-        user_workspaces = sorted(user_workspaces, key=lambda ws: ws.domain)
-
-        session = requests.Session()
-        session.cookies = code_confirmed_response.cookies
-        session.headers.update({'User-Agent': data.user_agent})
-        log.info(f'{slack_user.email}: started getting of tokens for {len(user_workspaces)} workspaces')
-        for workspace in user_workspaces:
-            if not workspace.magic_login_code:
-                continue
-            login_url = f"https://app.slack.com/t/{workspace.domain}/login/{workspace.magic_login_code}"
-            magic_response = session.post(login_url, cookies=session.cookies, headers=session.headers)
-            content = magic_response.content.decode('utf-8')
-            start_token_index = content.find("xox")
-            sliced_content = content[start_token_index:]
-            end_token_index = sliced_content.find('"')
-            token = sliced_content[:end_token_index]
-            workspace.magic_login_url = login_url
-            workspace.token = token
-            workspace.domain = workspace.domain
-            time.sleep(random.uniform(0.2, 0.5))
-
-        slack_user.cookies = session.cookies.get_dict()
-        slack_user.workspaces = user_workspaces
-        slack_user.status = StatusConnection.COMPLETE
-
-        slack_users_dict = [user.to_dic() for user in current_user.slack_users]
-        UserMongoRepository().set(current_user.id, slack_users=slack_users_dict)
-
         dedicated_bots_repository = DedicatedBotRepository()
         dedicated_bots = dedicated_bots_repository.get_all(user_id=current_user.id, user_name=data.slack_email,
                                                            source_type=SourceType.SLACK, include_deleted=True)
@@ -141,4 +147,4 @@ class ConnectSlackAccountJob(BaseBackgroundJob, ABC):
                 dedicated_bot.invalid_creds = True
                 for server in dedicated_bot.servers:
                     server.deleted = True
-                dedicated_bots_repository.add_or_update(dedicated_bot)
+                dedicated_bots_repository.add_or_update(dedicated_bot)

{leadguru_jobs-0.706.0 → leadguru_jobs-0.708.0}/lgt_jobs/lgt_data/mongo_repository.py

@@ -31,7 +31,7 @@ from lgt_jobs.lgt_data.models.user.verification import UserVerificationModel
 from lgt_jobs.lgt_data.models.user_leads.extended_user_lead import ExtendedUserLeadModel
 from lgt_jobs.lgt_data.models.user_leads.user_lead import UserLeadModel
 
-client = MongoClient(os.environ.get('MONGO_CONNECTION_STRING', 'mongodb://127.0.0.1:27018/'))
+client = MongoClient(os.environ.get('MONGO_CONNECTION_STRING', 'mongodb://127.0.0.1:27017/'))
 
 
 def to_object_id(oid):
@@ -1418,7 +1418,7 @@ class MessageRequestsRepository(BaseMongoRepository):
         self.collection().update_one({'user_id': to_object_id(user_id), 'sender_id': sender_id},
                                      {'$set': message_request.to_dic()}, upsert=True)
 
-    def get_many(self, user_id: str, viewed: bool = False, **kwargs):
+    def get_many(self, user_id: str, viewed: bool = False, sort_direction: int = pymongo.ASCENDING, **kwargs):
         match_pipeline = {'user_id': to_object_id(user_id), 'viewed': viewed}
         from_date = kwargs.get('from_date')
 
@@ -1450,7 +1450,7 @@ class MessageRequestsRepository(BaseMongoRepository):
                 '$addFields': {'user': {'$first': '$user'}, 'bot': {'$first': '$bot'}}
             },
             {
-                '$sort': {'created_at': 1}
+                '$sort': {'created_at': sort_direction}
             }
         ]
         return self.collection().aggregate(pipeline)