telegram_libs 0.1.3__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.3" 
+__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/translation.py

@@ -1,7 +1,14 @@
 import json
 import os
+from typing import Any
 
-def load_translations():
+
+def load_translations() -> dict:
+    """Load translations from locales directory
+
+    Returns:
+        dict: Translations dictionary
+    """
     translations = {}
     # Get the project's root directory (where the script is being run from)
     project_root = os.path.abspath(os.getcwd())
@@ -21,7 +28,7 @@ def load_translations():
 
 TRANSLATIONS = load_translations()
 
-def t(key, lang='ru', **kwargs):
+def t(key: str, lang: str = 'ru', **kwargs: Any) -> str:
     """Get translation for a key with optional formatting"""
     try:
         # Support nested keys like "buttons.start"

telegram_libs/utils.py

@@ -3,11 +3,11 @@ from telegram import (
     InlineKeyboardMarkup,
 )
 from telegram import Update
+from telegram.ext import ContextTypes
 from telegram_libs.constants import BOTS_AMOUNT
 from telegram_libs.translation import t
 
 
-
 async def get_subscription_keyboard(update: Update, lang: str) -> InlineKeyboardMarkup:
     """Get subscription keyboard
 
@@ -19,7 +19,7 @@ async def get_subscription_keyboard(update: Update, lang: str) -> InlineKeyboard
         InlineKeyboardMarkup: Inline keyboard markup
     """
     await update.message.reply_text(
-        f"Buying a subscription you will get unlimited access to other {BOTS_AMOUNT} bots, to see all bots click /more"
+        f"Buying a subscription you will get unlimited access to other {int(BOTS_AMOUNT) - 1} bots, to see all bots click /more"
     )
     return [
         [
@@ -35,4 +35,13 @@ async def get_subscription_keyboard(update: Update, lang: str) -> InlineKeyboard
                 t("subscription.plans.1year", lang), callback_data="sub_1year"
             ),
         ],
-    ]
+    ]
+    
+
+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/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.3.dist-info → telegram_libs-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: telegram_libs
-Version: 0.1.3
+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.3.dist-info/RECORD

@@ -1,8 +0,0 @@
-telegram_libs/__init__.py,sha256=70j0wY1ZGtZQWN4EOWmM3recKj_K6pXSGUernohm4uQ,82
-telegram_libs/constants.py,sha256=mZC3_2_Favn01sI10VMz9RiQpNpShqdbFfv3aMA3DKg,319
-telegram_libs/translation.py,sha256=LCJLzKzAfd-65pAvSpW7MFr8o34nsdfWuVWonHI7SQU,1270
-telegram_libs/utils.py,sha256=DAhnilUE7-WlXMoK1LnwcIJ4kgHDdmM3HV06ScyarMY,1089
-telegram_libs-0.1.3.dist-info/LICENSE,sha256=ZXkWPZbCc61L29Gz6ZHPwn1c4Pm0TnfIqtx8jGWi9F4,1069
-telegram_libs-0.1.3.dist-info/METADATA,sha256=NHNMoASTZNfeF1IVfP6_I_jRcpAtkxaH44QkNjexyaw,762
-telegram_libs-0.1.3.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-telegram_libs-0.1.3.dist-info/RECORD,,