leadguru-jobs 0.623.0__tar.gz → 0.624.0__tar.gz

{leadguru_jobs-0.623.0 → leadguru_jobs-0.624.0}/PKG-INFO

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

{leadguru_jobs-0.623.0 → leadguru_jobs-0.624.0}/leadguru_jobs.egg-info/PKG-INFO

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

{leadguru_jobs-0.623.0 → leadguru_jobs-0.624.0}/leadguru_jobs.egg-info/SOURCES.txt

@@ -31,7 +31,6 @@ lgt_jobs/jobs/mass_message.py
 lgt_jobs/jobs/send_code.py
 lgt_jobs/jobs/send_slack_message.py
 lgt_jobs/jobs/update_slack_profile.py
-lgt_jobs/jobs/user_balance_update.py
 lgt_jobs/jobs/workspace_connect.py
 lgt_jobs/lgt_common/__init__.py
 lgt_jobs/lgt_common/helpers.py

{leadguru_jobs-0.623.0 → leadguru_jobs-0.624.0}/lgt_jobs/__init__.py

@@ -1,6 +1,5 @@
 name = "lgt_jobs"
 
-from .jobs.user_balance_update import UpdateUserBalanceJob, UpdateUserBalanceJobData
 from .jobs.send_slack_message import SendSlackMessageJob, SendSlackMessageJobData
 from .jobs.analytics import (TrackAnalyticsJob, TrackAnalyticsJobData)
 from .jobs.connect_sources import (ConnectSourceJobData, ConnectSourceJob)
@@ -22,7 +21,6 @@ jobs_map = {
     "LoadChatHistoryJob": LoadChatHistoryJob,
     "UpdateExternalUserProfileJob": UpdateExternalUserProfileJob,
     "SendSlackMessageJob": SendSlackMessageJob,
-    "UpdateUserBalanceJob": UpdateUserBalanceJob,
     "SendMassMessageSlackChannelJob": SendMassMessageSlackChannelJob,
     "ConnectSourceJob": ConnectSourceJob,
     "LoadSlackPeopleJob": LoadSlackPeopleJob
@@ -37,7 +35,6 @@ __all__ = [
     UpdateExternalUserProfileJob,
     TrackAnalyticsJob,
     SendSlackMessageJob,
-    UpdateUserBalanceJob,
     SendMassMessageSlackChannelJob,
     ConnectSourceJob,
     LoadSlackPeopleJob,
@@ -54,7 +51,6 @@ __all__ = [
     UpdateExternalUserProfileJobData,
     TrackAnalyticsJobData,
     SendSlackMessageJobData,
-    UpdateUserBalanceJobData,
     SendMassMessageSlackChannelJobData,
     ConnectSourceJobData,
     LoadSlackPeopleJobData,

leadguru_jobs-0.623.0/lgt_jobs/jobs/user_balance_update.py

@@ -1,71 +0,0 @@
-import datetime
-from abc import ABC
-from typing import Optional
-from lgt_jobs.lgt_data.engine import UserCreditStatementDocument
-from lgt_jobs.lgt_data.enums import UserAccountState, UserAction
-from lgt_jobs.lgt_data.model import UserModel
-from lgt_jobs.lgt_data.mongo_repository import UserMongoRepository
-from lgt_jobs.lgt_data.helpers import to_object_id
-from pydantic import BaseModel
-from lgt_jobs.basejobs import BaseBackgroundJobData, BaseBackgroundJob
-
-"""
-User balance handling
-"""
-
-
-class UpdateUserBalanceJobData(BaseBackgroundJobData, BaseModel):
-    user_id: Optional[str]
-
-
-class UpdateUserBalanceJob(BaseBackgroundJob, ABC):
-    @property
-    def job_data_type(self) -> type:
-        return UpdateUserBalanceJobData
-
-    @staticmethod
-    def __update_processed(user: UserModel):
-        pipeline = [
-            {
-                "$match": {
-                    "user_id": to_object_id(user.id),
-                    "balance": {"$lte": 0},
-                    "action": {"$nin": [UserAction.ADMIN_CREDITS_ADDED, UserAction.ADMIN_CREDITS_SET]}
-                }
-            },
-            {
-                '$group': {
-                    '_id': "$user_id",
-                    'count': {'$sum': {"$multiply": [-1, "$balance"]}}
-                }
-            }
-        ]
-
-        results = list(UserCreditStatementDocument.objects.aggregate(*pipeline))
-        if not results:
-            return
-
-        count = results[0]["count"]
-        if count >= user.leads_limit:
-            # suspend account
-            if user.state != UserAccountState.Suspended.value:
-                UserMongoRepository().set(user.id, leads_proceeded=count,
-                                          credits_exceeded_at=datetime.datetime.now(datetime.UTC),
-                                          state=UserAccountState.Suspended.value)
-                return
-
-        if user.state == UserAccountState.Suspended.value:
-            # unsuspend account
-            UserMongoRepository().set(user.id, leads_proceeded=count,
-                                      credits_exceeded_at=None,
-                                      state=UserAccountState.Operational.value)
-            return
-
-        UserMongoRepository().set(user.id, leads_proceeded=count)
-
-    def exec(self, data: UpdateUserBalanceJobData):
-        user = UserMongoRepository().get(data.user_id)
-        if not user:
-            return
-
-        self.__update_processed(user=user)