leadguru-jobs 0.474.0__py3-none-any.whl → 0.476.0__py3-none-any.whl

{leadguru_jobs-0.474.0.dist-info → leadguru_jobs-0.476.0.dist-info}/METADATA

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

{leadguru_jobs-0.474.0.dist-info → leadguru_jobs-0.476.0.dist-info}/RECORD

@@ -40,14 +40,14 @@ lgt_jobs/lgt_common/slack_client/web_client.py,sha256=oZV7oUcSXXngfMI6oXDedlQWeP
 lgt_jobs/lgt_data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lgt_jobs/lgt_data/analytics.py,sha256=qo5nNFQxOTTk8i3tmCq0d3F9iaV9MTIR84SSi6uNrJ8,23427
 lgt_jobs/lgt_data/engine.py,sha256=OkTIIsVhb4o0dcDOzolb-NNTftzhJvsCuldxg1PvLUM,8113
-lgt_jobs/lgt_data/enums.py,sha256=8cTLz8e6Jz4ikjNzqMiorwMMdVjU2SyXtl_tD7nHKxw,2326
+lgt_jobs/lgt_data/enums.py,sha256=P-Jo_kQfFanbN5ztjQZIaWkZ6qOmAjHdypZ85ReOgdA,2407
 lgt_jobs/lgt_data/helpers.py,sha256=58ycqKF1z5nAw0vs6yVCg-SQet107YHyKYXaQUc7M68,157
-lgt_jobs/lgt_data/model.py,sha256=wsa7_MzCU9EGcdclpzLaQx_ovybBARyfpnzJ-P-G2pM,28583
-lgt_jobs/lgt_data/mongo_repository.py,sha256=dEdpMkBMhdzSWzj_HJCZYEo89zkVBqGXgyjyFLNOTVA,45463
+lgt_jobs/lgt_data/model.py,sha256=6eds2J7gPfIVAs7E1bTJEvH_M8rWdJ091cQmIw5Sznk,29090
+lgt_jobs/lgt_data/mongo_repository.py,sha256=NO6w1mkTfzFa3_qIP-2_SnPGaItxI33Z1EXbQMgbRng,45971
 lgt_jobs/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lgt_jobs/services/web_client.py,sha256=eCgzyyXco6iPwAJziyRek_3ns1bxJZKknnLIJApKeoo,2070
 lgt_jobs/templates/new_message.html,sha256=I3J6YPz-uhGbGnuGc5d7Hxj0X4VzOtT8PqrQQGPJfc4,7002
-leadguru_jobs-0.474.0.dist-info/METADATA,sha256=b-2UyZBdRHr6UglK1sU4Axa9SqtCTn01aEI2ga6kg_8,1319
-leadguru_jobs-0.474.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-leadguru_jobs-0.474.0.dist-info/top_level.txt,sha256=rIuw1DqwbnZyeoarBSC-bYeGOhv9mZBs7_afl9q4_JI,9
-leadguru_jobs-0.474.0.dist-info/RECORD,,
+leadguru_jobs-0.476.0.dist-info/METADATA,sha256=hNAP57CcPNJkE0MXhAf98UZmF3ex2ZqWxnu0cZRw6DE,1319
+leadguru_jobs-0.476.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+leadguru_jobs-0.476.0.dist-info/top_level.txt,sha256=rIuw1DqwbnZyeoarBSC-bYeGOhv9mZBs7_afl9q4_JI,9
+leadguru_jobs-0.476.0.dist-info/RECORD,,

lgt_jobs/lgt_data/enums.py

@@ -101,3 +101,8 @@ class FeaturesEnum(str, Enum):
     SCHEDULED_MESSAGES = 'scheduled_messages'
     SOURCES = 'sources'
     ALGORITHMS = 'algorithms'
+
+
+class FeatureOptions(str, Enum):
+    BASIC = 'basic'
+    ADVANCED = 'advanced'

lgt_jobs/lgt_data/model.py

@@ -4,7 +4,7 @@ import json
 from abc import ABC
 from datetime import datetime, UTC
 from typing import Optional, List, Dict
-from .enums import UserRole, SourceType
+from .enums import UserRole, SourceType, FeaturesEnum, FeatureOptions
 from .helpers import get_linkedin_search_contact
 from bson import ObjectId
 
@@ -287,6 +287,8 @@ class UserModel(BaseModel):
         self.slack_users: List[SlackUser] = []
         self.discord_users: List[DiscordUser] = []
         self.verified: bool = False
+        self.subscription_id: ObjectId | None = None
+        self.subscription_expired_at: datetime | None = None
 
     @classmethod
     def from_dic(cls, dic: dict):
@@ -990,4 +992,22 @@ class UsersPage:
     def from_dic(dic: dict):
         users = [UserModel.from_dic(doc) for doc in dic.get('page', [])]
         count = dic.get('count', 0)
-        return UsersPage(users=users, count=count)
+        return UsersPage(users=users, count=count)
+
+
+class Feature(DictionaryModel):
+    display_name: str
+    name: FeaturesEnum
+    description: str | None = None
+    limit: int | None = None
+    options: FeatureOptions | None = None
+
+
+class Subscription(BaseModel):
+    features: list[Feature]
+    duration_days: int
+    name: str
+    price: int
+    limits: int
+    trial: bool
+    updated_at: datetime = None

lgt_jobs/lgt_data/mongo_repository.py

@@ -9,7 +9,7 @@ from lgt_jobs.lgt_data.enums import SourceType
 from lgt_jobs.lgt_data.model import (LeadModel, BaseModel, UserModel, UserResetPasswordModel, BoardModel, BoardedStatus,
                                      DedicatedBotModel, SlackMemberInformation, UserTemplateModel, LinkedinContact,
                                      ExtendedUserLeadModel, UserLeadModel, ExtendedLeadModel, UserContact, ChatMessage,
-                                     GroupedMessagesModel, UserVerificationModel, UsersPage)
+                                     GroupedMessagesModel, UserVerificationModel, UsersPage, Subscription)
 from datetime import datetime, UTC, timedelta
 from collections import OrderedDict
 
@@ -1227,3 +1227,21 @@ class UserVerificationMongoRepository(BaseMongoRepository):
         model.email = email
         model.created_at = datetime.now(UTC)
         return self.collection().insert_one(model.to_dic()).inserted_id
+
+
+class SubscriptionsRepository(BaseMongoRepository):
+    pass
+
+    collection_name = 'subscriptions'
+    model = Subscription
+
+    def find_one(self, **kwargs):
+        pipeline = {}
+        subscription_id = kwargs.get('id')
+        trial = kwargs.get('trial')
+        if subscription_id:
+            pipeline['_id'] = to_object_id(subscription_id)
+        if trial is not None:
+            pipeline['trial'] = trial
+
+        return Subscription.from_dic(self.collection().find_one(pipeline))