leadguru-jobs 0.706.0__py3-none-any.whl → 0.708.0__py3-none-any.whl

{leadguru_jobs-0.706.0.dist-info → leadguru_jobs-0.708.0.dist-info}/METADATA

@@ -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.dist-info → leadguru_jobs-0.708.0.dist-info}/RECORD

@@ -24,7 +24,7 @@ lgt_jobs/jobs/mass_message.py,sha256=1mFcBlL2MhzLbj5yrd_NyJc7TXDWCcROAzGDnr0miMU
 lgt_jobs/jobs/send_code.py,sha256=dIlLPkG3GgGKIEqGiElyzrtdrnJNTL1Ak2V0xnE-WIQ,824
 lgt_jobs/jobs/send_slack_message.py,sha256=eiSN-yZ7XJ2WLUTQOkViSYOCYwaEwLry0ldGAcONNsM,1708
 lgt_jobs/jobs/update_slack_profile.py,sha256=GU28azaNz1zYsCAAF_BShJxLUecQb7AbO5XPgiv7yHg,3654
-lgt_jobs/jobs/workspace_connect.py,sha256=rR02czRNFG1vbWfooT3mafQ_D-jIDTQ6YvjWJqcTzcQ,7356
+lgt_jobs/jobs/workspace_connect.py,sha256=3uGxI0gTXVloZWxo32uhRmHjVwCGzET2CLUNjN4CcS8,7829
 lgt_jobs/lgt_common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lgt_jobs/lgt_common/helpers.py,sha256=3T47_Abw7datSIxVqIHTCB-sh-rnahWrmrFG8sSY9ZE,178
 lgt_jobs/lgt_common/discord_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -45,7 +45,7 @@ lgt_jobs/lgt_data/engine.py,sha256=GY8FOl3dl7euKjQRML_H3ophaSTgfQ8ICMRv1NpE2oE,7
 lgt_jobs/lgt_data/enums.py,sha256=SN2eLoB2m-bVswgLVuY46v0DhPj8IurgHY6R-KRsd6s,2621
 lgt_jobs/lgt_data/helpers.py,sha256=S_1P7pnx14dllmen-TzjA9pkKWQr39x8-v6iv7FUYnQ,423
 lgt_jobs/lgt_data/model.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-lgt_jobs/lgt_data/mongo_repository.py,sha256=XrRxW7zh-TC33oYrpC6Cfn25MTY6RNcrNiA-4SjvSXE,57885
+lgt_jobs/lgt_data/mongo_repository.py,sha256=02ot-tPDDqjZYjuRwkrH3biKUYBSxmKQaltxhK0IeS8,57939
 lgt_jobs/lgt_data/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lgt_jobs/lgt_data/models/base.py,sha256=YaiceCaaF45snOLx-bQoYJ1aTWrfgMQeHWjAaRnzGKE,592
 lgt_jobs/lgt_data/models/boards/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -108,7 +108,7 @@ lgt_jobs/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 lgt_jobs/services/k8_manager.py,sha256=bXpka9psIQ5UJ6QG_e4xolmE_3gtmNrzzZeL03bJ62I,995
 lgt_jobs/services/web_client.py,sha256=oMyWJxwGeIe3f40fPT7xcisjDg3BhA3Ipf8dr1jVT-Y,1549
 lgt_jobs/templates/new_message.html,sha256=dZl8UmdAOOMq4nidvAgMFroSrTV7Pw0RWt2yLp_2idg,6989
-leadguru_jobs-0.706.0.dist-info/METADATA,sha256=gQ-10oYW0PBsPeVygvf2ChsJNhl8blZ8LsIwALf-9Qc,1399
-leadguru_jobs-0.706.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-leadguru_jobs-0.706.0.dist-info/top_level.txt,sha256=rIuw1DqwbnZyeoarBSC-bYeGOhv9mZBs7_afl9q4_JI,9
-leadguru_jobs-0.706.0.dist-info/RECORD,,
+leadguru_jobs-0.708.0.dist-info/METADATA,sha256=Wn8REnDUdVrmJXSp7zl5xK0wS4wwKcfbTseARAODWoo,1399
+leadguru_jobs-0.708.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+leadguru_jobs-0.708.0.dist-info/top_level.txt,sha256=rIuw1DqwbnZyeoarBSC-bYeGOhv9mZBs7_afl9q4_JI,9
+leadguru_jobs-0.708.0.dist-info/RECORD,,

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)

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)