leadguru-jobs 0.474.0__tar.gz → 0.475.0__tar.gz

{leadguru_jobs-0.474.0 → leadguru_jobs-0.475.0}/PKG-INFO

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

{leadguru_jobs-0.474.0 → leadguru_jobs-0.475.0}/leadguru_jobs.egg-info/PKG-INFO

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

{leadguru_jobs-0.474.0 → leadguru_jobs-0.475.0}/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'

{leadguru_jobs-0.474.0 → leadguru_jobs-0.475.0}/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

{leadguru_jobs-0.474.0 → leadguru_jobs-0.475.0}/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,18 @@ 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')
+        if subscription_id:
+            pipeline['_id'] = to_object_id(subscription_id)
+
+        return Subscription.from_dic(self.collection().find_one(pipeline))