leadguru-jobs 0.724.0__py3-none-any.whl → 0.725.0__py3-none-any.whl

{leadguru_jobs-0.724.0.dist-info → leadguru_jobs-0.725.0.dist-info}/METADATA

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

{leadguru_jobs-0.724.0.dist-info → leadguru_jobs-0.725.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-lgt_jobs/__init__.py,sha256=fQ97ZFeVtbOgODPdB26x7ku45AQPQ8ixeziiCAjNYZM,2451
+lgt_jobs/__init__.py,sha256=NQozhvhEGeXUvyj8kkT-Q5FduKAkQt1bDxlczkjbmQY,2635
 lgt_jobs/basejobs.py,sha256=LPEclvY6fIG17YMGy9pRK0Q58n62gkI5W1BTK6R2p-c,1234
 lgt_jobs/env.py,sha256=HHlpQgZrRQMMuPZYhP94A8b29T_10sKRo98mh0_hh3Y,907
 lgt_jobs/main.py,sha256=zd0TdwFBUXUmFfUw552rPt6Rd_uG-LWEE6wTs8TCxIk,2081
@@ -22,6 +22,7 @@ lgt_jobs/jobs/load_slack_people.py,sha256=RnuHRPvZkAlM9o7i6hQtBBmnN_nu5X6LLZY5eC
 lgt_jobs/jobs/load_slack_users.py,sha256=d_Hy-MM86OqWdsaLX06n-ovNL6whY1xQz6sv_Ls3C4I,1404
 lgt_jobs/jobs/mass_message.py,sha256=1mFcBlL2MhzLbj5yrd_NyJc7TXDWCcROAzGDnr0miMU,2035
 lgt_jobs/jobs/send_code.py,sha256=dIlLPkG3GgGKIEqGiElyzrtdrnJNTL1Ak2V0xnE-WIQ,824
+lgt_jobs/jobs/send_slack_invites.py,sha256=tO4XZ_g-YhLZAhf871qHs4SVd9z9ILRVKVC6cBWlhC0,1850
 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=3uGxI0gTXVloZWxo32uhRmHjVwCGzET2CLUNjN4CcS8,7829
@@ -45,7 +46,7 @@ lgt_jobs/lgt_data/engine.py,sha256=GY8FOl3dl7euKjQRML_H3ophaSTgfQ8ICMRv1NpE2oE,7
 lgt_jobs/lgt_data/enums.py,sha256=xNE5uIot8yT2Yl6HufTYlPiR0uSJVP2fNbnObL5tsNE,2669
 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=02ot-tPDDqjZYjuRwkrH3biKUYBSxmKQaltxhK0IeS8,57939
+lgt_jobs/lgt_data/mongo_repository.py,sha256=KRfnAHzBbPWPHe3gc2L3cNcSv4Qq4yvG_by1Oq41rsE,54763
 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 +109,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.724.0.dist-info/METADATA,sha256=HL0s6RKqssyN7LZdE1t2lkFZ70wcyIBPpA-qWNck3xg,1399
-leadguru_jobs-0.724.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-leadguru_jobs-0.724.0.dist-info/top_level.txt,sha256=rIuw1DqwbnZyeoarBSC-bYeGOhv9mZBs7_afl9q4_JI,9
-leadguru_jobs-0.724.0.dist-info/RECORD,,
+leadguru_jobs-0.725.0.dist-info/METADATA,sha256=c6qwlQOg87kjWjPc87vZigeNg_M6yPBDholW0VKR9Mk,1399
+leadguru_jobs-0.725.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+leadguru_jobs-0.725.0.dist-info/top_level.txt,sha256=rIuw1DqwbnZyeoarBSC-bYeGOhv9mZBs7_afl9q4_JI,9
+leadguru_jobs-0.725.0.dist-info/RECORD,,

lgt_jobs/__init__.py

@@ -16,6 +16,7 @@ from .simple_job import (SimpleTestJob, SimpleTestJobData)
 from .jobs.load_slack_people import (LoadSlackPeopleJob, LoadSlackPeopleJobData)
 from .jobs.load_slack_users import LoadSlackUsersJob, LoadSlackUsersJobData
 from .jobs.bots_killer import BotsKillerJob, BotsKillerData
+from .jobs.send_slack_invites import SendSlackInvitesJob, SendSlackInvitesJobData
 
 jobs_map = {
     "SimpleTestJob": SimpleTestJob,
@@ -29,7 +30,8 @@ jobs_map = {
     "ConnectSourceJob": ConnectSourceJob,
     "LoadSlackPeopleJob": LoadSlackPeopleJob,
     "BotsKillerJob": BotsKillerJob,
-    "LoadSlackUsersJob": LoadSlackUsersJob
+    "LoadSlackUsersJob": LoadSlackUsersJob,
+    "SendSlackInvitesJob": SendSlackInvitesJob
 }
 __all__ = [
     # Jobs
@@ -46,6 +48,7 @@ __all__ = [
     LoadSlackPeopleJob,
     LoadSlackUsersJob,
     BotsKillerJob,
+    SendSlackInvitesJob,
 
     # module classes
     BackgroundJobRunner,
@@ -64,6 +67,7 @@ __all__ = [
     LoadSlackPeopleJobData,
     LoadSlackUsersJobData,
     BotsKillerData,
+    SendSlackInvitesJobData,
     # mapping
     jobs_map
 ]

lgt_jobs/jobs/send_slack_invites.py

@@ -0,0 +1,45 @@
+from abc import ABC
+from lgt_jobs.lgt_common.slack_client.web_client import SlackWebClient
+from pydantic import BaseModel
+from lgt_jobs.basejobs import BaseBackgroundJobData, BaseBackgroundJob
+from lgt_jobs.lgt_data.enums import SourceType
+from lgt_jobs.lgt_data.mongo_repository import DedicatedBotRepository, UserMongoRepository
+
+"""
+Send Slack Code
+"""
+
+
+class SendSlackInvitesJobData(BaseBackgroundJobData, BaseModel):
+    emails: list[str]
+    email_to_invite: str
+
+
+class SendSlackInvitesJob(BaseBackgroundJob, ABC):
+    @property
+    def job_data_type(self) -> type:
+        return SendSlackInvitesJobData
+
+    def exec(self, data: SendSlackInvitesJobData):
+        print('[SendSlackInvitesJob]: Start...')
+        users = UserMongoRepository().get_users(emails=data.emails)
+        bots = DedicatedBotRepository().get_all(user_ids=[user.id for user in users],
+                                                source_type=SourceType.SLACK,
+                                                only_valid=True,
+                                                include_paused=True,
+                                                include_deleted=True)
+        used_sources = {}
+        count = 0
+        for bot in bots:
+            if not used_sources.get(bot.source.source_id):
+                used_sources[bot.source.source_id] = None
+                try:
+                    client = SlackWebClient(bot.token, bot.cookies)
+                    resp = client.send_slack_invite_to_workspace(data.email_to_invite)
+                    print(f'Sent invite to {bot.source.source_name}. Count is {count}. Response: {resp.content}')
+                    count += 1
+                except:
+                    print(f'Error to sent invite to {bot.source.source_name}')
+
+
+        print(f'[SendSlackInvitesJob]: STOP...User were invited to {count} workspaces.')

lgt_jobs/lgt_data/mongo_repository.py

@@ -1,5 +1,4 @@
 import os
-import re
 from typing import List, Optional
 import pymongo
 from dateutil import tz
@@ -70,6 +69,23 @@ class UserMongoRepository(BaseMongoRepository):
     collection_name = 'users'
     model = UserModel
 
+    def get_inactive_users(self):
+        pipeline = [
+            {'$lookup': {'from': 'user_track_action', 'as': 'last_action', 'let': {'id': '$_id' }, 'pipeline':
+                [
+                    {'$match': {'$expr': {'$eq': ['$user_id', '$$id']}}},
+                    {'$sort': {'created_at': -1}},
+                    {'$project': {'created_at': 1}},
+                    {'$limit': 1}
+                ]}},
+            {'$addFields': {'last_action_date': {'$first': '$last_action.created_at'}}},
+            {'$lookup': {'as': 'subscription', 'from': 'subscriptions', 'foreignField': '_id', 'localField': 'subscription_id'}},
+            {'$addFields': {'period': {'$first': '$subscription.disable_bots_period'}}},
+            {'$match': {'$expr': {'$lte': ['$last_action_date', {'$dateSubtract': {'startDate': '$$NOW', 'unit': 'day', 'amount': '$period'}}]}}},
+            {'$project': {'_id': 1}}]
+
+        return {doc['_id']: None for doc in self.collection().aggregate(pipeline)}
+
     def get(self, _id):
         return UserModel.from_dic(self.collection().find_one({'_id': to_object_id(_id)}))
 
@@ -94,6 +110,7 @@ class UserMongoRepository(BaseMongoRepository):
         has_new_replies = kwargs.get('has_new_replies')
         has_new_reactions = kwargs.get('has_new_reactions')
         min_days = kwargs.get('min_days_soon_subscription_expiration', 3)
+        emails = kwargs.get('emails')
 
         if subscription_expired:
             pipeline['subscription_expired_at'] = {'$lte': datetime.now(UTC)}
@@ -129,6 +146,9 @@ class UserMongoRepository(BaseMongoRepository):
         if users_ids is not None:
             pipeline['_id'] = {'$in': [to_object_id(_id) for _id in users_ids]}
 
+        if emails is not None:
+            pipeline['email'] = {'$in': emails}
+
         if name is not None:
             pipeline['user_name'] = {"$regex": name, "$options": 'i'}
 
@@ -682,95 +702,6 @@ class LeadMongoRepository(BaseMongoRepository):
         update_dict = {k: v for k, v in kwargs.items() if v is not None}
         self.collection().update_one({'id': _id}, {'$set': update_dict}, upsert=False)
 
-    def get_list(self, skip, limit, **kwargs):
-        pipeline = self.__create_leads_filter(**kwargs)
-
-        sort_field = kwargs.get('sort_field', 'created_at')
-        sort_direction = kwargs.get('sort_direction', 'ASCENDING')
-        sort_direction = pymongo.ASCENDING if sort_direction == 'ASCENDING' else pymongo.DESCENDING
-
-        leads = self.collection().find(pipeline).sort([(sort_field, sort_direction)]).skip(skip).limit(limit)
-        return [LeadModel.from_dic(lead) for lead in leads]
-
-    def __create_leads_filter(self, **kwargs):
-        pipeline: dict = {"hidden": False}
-
-        from_date: datetime = kwargs.get('from_date')
-        to_date: datetime = kwargs.get('to_date')
-
-        country = kwargs.get('country')
-        user_id = kwargs.get('user_id')
-        tags = kwargs.get('tags')
-        text = kwargs.get('text')
-        stop_words = kwargs.get('stop_words')
-        exclude_leads = kwargs.get('exclude_leads')
-        exclude_senders = kwargs.get('exclude_senders')
-        sender_ids = kwargs.get('sender_ids')
-        configs = kwargs.get('config')
-        bots_names = kwargs.get('bots_names')
-        locations = kwargs.get('locations')
-        publication_text = kwargs.get('publication_text')
-
-        pipeline['message.profile.display_name'] = {
-            "$ne": "Slackbot"
-        }
-
-        if from_date or to_date:
-            pipeline['created_at'] = {}
-
-        if from_date:
-            start = from_date.astimezone(tz.tzutc())
-            pipeline['created_at']['$gte'] = start
-
-        if to_date:
-            end = to_date.astimezone(tz.tzutc())
-            pipeline['created_at']['$lte'] = end
-
-        if locations and len(locations) > 0:
-            pipeline['message.locations'] = {"$in": locations}
-
-        if country:
-            pipeline["message.slack_options.country"] = re.compile(country, re.IGNORECASE)
-
-        if user_id:
-            pipeline["$or"] = [
-                {"message.dedicated_slack_options": {"$exists": False}},
-                {"message.dedicated_slack_options": None},
-                {"message.dedicated_slack_options.user_id": f"{user_id}"},
-            ]
-        else:
-            pipeline["user_id"] = {'$exists': False}
-
-        if stop_words:
-            pipeline['full_message_text'] = {'$regex': f'^(?!.*({stop_words})).*$', '$options': 'i'}
-        elif text:
-            pipeline['$text'] = {'$search': text}
-
-        if publication_text:
-            pipeline['message.message'] = {'$regex': publication_text, '$options': 'i'}
-
-        if tags:
-            pipeline["tags"] = {"$elemMatch": {"$in": tags}}
-
-        if exclude_leads:
-            pipeline['id'] = {'$nin': exclude_leads}
-
-        if exclude_senders:
-            pipeline['message.sender_id'] = {'$nin': exclude_senders}
-
-        if sender_ids:
-            pipeline['message.sender_id'] = {'$in': sender_ids}
-
-        if configs:
-            pipeline["message.configs.id"] = {"$not": {"$elemMatch": {"$nin": configs}}}
-
-        if bots_names is not None:
-            pipeline['message.name'] = {'$in': bots_names}
-
-        pipeline['message.profile.real_name'] = {'$ne': 'Slackbot'}
-
-        return pipeline
-
     def get_per_day(self, date: datetime):
         start_day = datetime(date.year, date.month, date.day, 0, 0, 0, tzinfo=tz.tzutc())
         end_day = datetime(date.year, date.month, date.day, 23, 59, 59, tzinfo=tz.tzutc())
@@ -778,35 +709,6 @@ class LeadMongoRepository(BaseMongoRepository):
         return [LeadModel.from_dic(x) for x in docs]
 
 
-class SpamLeadsMongoRepository(LeadMongoRepository):
-    pass
-
-    def __init__(self):
-        self.collection_name = 'spam_leads'
-
-    def get_list(self, skip, limit, sort_field: str = 'created_at', sort_direction: str = 'ASCENDING', **kwargs):
-        pipeline = self.__create_leads_filter(**kwargs)
-        sort_direction = pymongo.ASCENDING if sort_direction == 'ASCENDING' else pymongo.DESCENDING
-        docs = self.collection().find(pipeline).sort([(sort_field, sort_direction)]).skip(skip).limit(limit)
-        leads = [ExtendedLeadModel.from_dic(doc) for doc in docs]
-        senders = [lead.message.sender_id for lead in leads]
-        contacts = SlackContactUserRepository().find(users=senders)
-        for lead in leads:
-            lead.contact = next(filter(lambda x: x.sender_id == lead.message.sender_id, contacts), None)
-        return leads
-
-    def __create_leads_filter(self, **kwargs):
-        pipeline = {"user_id": {'$exists': False}}
-        text = kwargs.get('text', None)
-        stop_words = kwargs.get('stop_words', None)
-        if stop_words:
-            pipeline['full_message_text'] = {'$regex': f'^(?!.*({stop_words})).*$', '$options': 'i'}
-        elif text:
-            pipeline['$text'] = {'$search': text}
-
-        return pipeline
-
-
 class SpamUserLeadsMongoRepository(UserLeadMongoRepository):
     pass
 
@@ -1023,6 +925,7 @@ class DedicatedBotRepository(BaseMongoRepository):
         invalid_creds = kwargs.get('invalid_creds')
         sort_by = kwargs.get('sort_by')
         deactivated_from = kwargs.get('deactivated_from')
+        user_ids = kwargs.get('user_ids')
 
         if deactivated_from:
             pipeline['deactivated_at'] = {'$gte': deactivated_from}
@@ -1066,6 +969,9 @@ class DedicatedBotRepository(BaseMongoRepository):
         if sort_by:
             pipeline[sort_by] = {'$exists': True}
 
+        if user_ids is not None:
+            pipeline["user_id"] = {'$in': [to_object_id(_id) for _id in user_ids]}
+
         return pipeline