telegram_libs 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

telegram_libs/__init__.py

@@ -2,4 +2,4 @@
 Telegram Libs - Common libraries for Telegram bots
 """
 
-__version__ = "0.1.4" 
+__version__ = "0.1.5" 

telegram_libs/constants.py

@@ -3,7 +3,12 @@ import os
 required_constants = []
 
 BOTS_AMOUNT = os.getenv("BOTS_AMOUNT")
+MONGO_URI = os.getenv("MONGO_URI")
+SUBSCRIPTION_DB_NAME = os.getenv("SUBSCRIPTION_DB_NAME")
+
 required_constants.append(("BOTS_AMOUNT", BOTS_AMOUNT))
+required_constants.append(("MONGO_URI", MONGO_URI))
+required_constants.append(("SUBSCRIPTION_DB_NAME", SUBSCRIPTION_DB_NAME))
 
 missing_constants = [name for name, value in required_constants if not value]
 if missing_constants:

telegram_libs/subscription.py

@@ -0,0 +1,55 @@
+from datetime import datetime
+from pymongo.mongo_client import MongoClient
+from pymongo.server_api import ServerApi
+from telegram_libs.constants import MONGO_URI, SUBSCRIPTION_DB_NAME
+
+# Create a new client and connect to the server
+client = MongoClient(MONGO_URI, server_api=ServerApi("1"))
+
+# Define the subscription database and collection
+subscription_db = client[SUBSCRIPTION_DB_NAME]
+subscription_collection = subscription_db["subscriptions"]
+
+
+def get_subscription(user_id: int) -> dict:
+    """Get user's subscription data from the shared subscription database."""
+    subscription = subscription_collection.find_one({"user_id": user_id})
+    if not subscription:
+        return {"user_id": user_id, "is_premium": False}
+    return subscription
+
+
+def update_subscription(user_id: int, updates: dict) -> None:
+    """Update user's subscription data in the shared subscription database."""
+    subscription_collection.update_one(
+        {"user_id": user_id},
+        {"$set": updates},
+        upsert=True
+    )
+
+
+def add_subscription_payment(user_id: int, payment_data: dict) -> None:
+    """Add a subscription payment record."""
+    subscription_collection.update_one(
+        {"user_id": user_id},
+        {
+            "$push": {"payments": payment_data},
+            "$set": {
+                "is_premium": True,
+                "premium_expiration": payment_data["expiration_date"],
+                "last_payment": payment_data["date"]
+            }
+        },
+        upsert=True
+    )
+
+
+def check_subscription_status(user_id: int) -> bool:
+    """Check if user has an active subscription."""
+    subscription = get_subscription(user_id)
+    
+    if not subscription.get("is_premium"):
+        return False
+        
+    expiration = datetime.fromisoformat(subscription["premium_expiration"])
+    return expiration > datetime.now() 

telegram_libs/utils.py

@@ -41,7 +41,7 @@ async def get_subscription_keyboard(update: Update, lang: str) -> InlineKeyboard
 async def more_bots_list_command(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
     message = """Here is the list of all bots: \n\n
     - <a href="https://t.me/MagMediaBot">Remove Background</a>
-    - <a href="https://t.me/upscale_image_bot">Upscale Image</a>
+    - <a href="https://t.me/UpscaleImageGBot">Upscale Image</a>
     - <a href="https://t.me/kudapoyti_go_bot">Recommend a place to visit</a>
     """
     await update.message.reply_text(message, disable_web_page_preview=True, parse_mode='HTML')

{telegram_libs-0.1.4.dist-info → telegram_libs-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: telegram_libs
-Version: 0.1.4
+Version: 0.1.5
 Summary: Common libraries for Telegram bots
 Author: Andrey Gritsaenko gricaenko.95a@gmail.com
 Requires-Python: >=3.11,<4.0
@@ -8,6 +8,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: pymongo (>=4.13.0,<5.0.0)
 Requires-Dist: pytest (>=8.3.5,<9.0.0)
 Requires-Dist: pytest-asyncio (>=1.0.0,<2.0.0)
 Requires-Dist: python-telegram-bot (>=22.1,<23.0)

telegram_libs-0.1.5.dist-info/RECORD

@@ -0,0 +1,9 @@
+telegram_libs/__init__.py,sha256=fo5Zc0080sDtseiBrQUIJw7X1l5eLxPMiHz2TTSVxVg,82
+telegram_libs/constants.py,sha256=F6pgwAuv2FgGZxxFwYYJomM4_hosexLoYsvHXQiuVNg,538
+telegram_libs/subscription.py,sha256=3NenqG9JL7FJEu56TqnYBF5piJg-r4j8mT4-TjPtoBM,1845
+telegram_libs/translation.py,sha256=OB89EyvwA89hxDfOJQcod96xoR5FSM4hsexl3_muqq0,1434
+telegram_libs/utils.py,sha256=S7_eRyyAAWUd1Q1SFreLxKAK0rqeDNDSddCLNWsD6CE,1594
+telegram_libs-0.1.5.dist-info/LICENSE,sha256=ZXkWPZbCc61L29Gz6ZHPwn1c4Pm0TnfIqtx8jGWi9F4,1069
+telegram_libs-0.1.5.dist-info/METADATA,sha256=97vMNd0jGrDk4vjJsrpA8iup0h1oLB9mPhdlN2Cf22E,803
+telegram_libs-0.1.5.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+telegram_libs-0.1.5.dist-info/RECORD,,

telegram_libs-0.1.4.dist-info/RECORD

@@ -1,8 +0,0 @@
-telegram_libs/__init__.py,sha256=UhMFbEOlA6Z8p2gBV8NxotAQTsstl7Ad9gxb-o66qPY,82
-telegram_libs/constants.py,sha256=mZC3_2_Favn01sI10VMz9RiQpNpShqdbFfv3aMA3DKg,319
-telegram_libs/translation.py,sha256=OB89EyvwA89hxDfOJQcod96xoR5FSM4hsexl3_muqq0,1434
-telegram_libs/utils.py,sha256=S2T6uPyhHJKq9zbcvTGXGUlvW3GetQ9jtA484h_Xqcc,1595
-telegram_libs-0.1.4.dist-info/LICENSE,sha256=ZXkWPZbCc61L29Gz6ZHPwn1c4Pm0TnfIqtx8jGWi9F4,1069
-telegram_libs-0.1.4.dist-info/METADATA,sha256=Qg3fWn6RaIamqUkuEj0fnvdYWhzyVv3ap-NrULCX1K4,762
-telegram_libs-0.1.4.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-telegram_libs-0.1.4.dist-info/RECORD,,