telegram_libs 0.1.1__tar.gz → 0.1.8__tar.gz

telegram_libs-0.1.8/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 AndreyGritsa
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

{telegram_libs-0.1.1 → telegram_libs-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: telegram_libs
-Version: 0.1.1
+Version: 0.1.8
 Summary: Common libraries for Telegram bots
 Author: Andrey Gritsaenko gricaenko.95a@gmail.com
 Requires-Python: >=3.11,<4.0
@@ -8,7 +8,11 @@ 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)
+Project-URL: Repository, https://github.com/AndreyGritsa/telegram_libs
 Description-Content-Type: text/markdown
 
 # Telegram Libs

{telegram_libs-0.1.1 → telegram_libs-0.1.8}/pyproject.toml

@@ -1,14 +1,18 @@
 [tool.poetry]
 name = "telegram_libs"
-version = "0.1.1"
+version = "0.1.8"
 description = "Common libraries for Telegram bots"
 authors = ["Andrey Gritsaenko gricaenko.95a@gmail.com"]
 readme = "README.md"
 packages = [{include = "telegram_libs", from = "src"}]
+repository = "https://github.com/AndreyGritsa/telegram_libs"
 
 [tool.poetry.dependencies]
 python = "^3.11"
 pytest = "^8.3.5"
+python-telegram-bot = "^22.1"
+pytest-asyncio = "^1.0.0"
+pymongo = "^4.13.0"
 
 [build-system]
 requires = ["poetry-core"]

{telegram_libs-0.1.1 → telegram_libs-0.1.8}/src/telegram_libs/__init__.py

@@ -2,4 +2,4 @@
 Telegram Libs - Common libraries for Telegram bots
 """
 
-__version__ = "0.1.0" 
+__version__ = "0.1.8"

telegram_libs-0.1.8/src/telegram_libs/constants.py

@@ -0,0 +1,15 @@
+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:
+    raise ValueError(f"Required constants are not set: {', '.join(missing_constants)}")

telegram_libs-0.1.8/src/telegram_libs/locales/en.json

@@ -0,0 +1,10 @@
+{
+  "subscription": {
+    "choose_plan": "Choose a subscription plan:",
+    "plans": {
+      "1month": "1 Month - 250 Stars",
+      "3months": "3 Months - 500 Stars",
+      "1year": "1 Year - 1600 Stars"
+    }
+  }
+}

telegram_libs-0.1.8/src/telegram_libs/locales/ru.json

@@ -0,0 +1,10 @@
+{
+  "subscription": {
+    "choose_plan": "Выберите план подписки:",
+    "plans": {
+      "1month": "1 месяц - 250 Stars",
+      "3months": "3 месяца - 500 Stars",
+      "1year": "1 год - 1600 Stars"
+    }
+  }
+}

telegram_libs-0.1.8/src/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-0.1.8/src/telegram_libs/translation.py

@@ -0,0 +1,63 @@
+import json
+import os
+from typing import Any
+
+
+def _load_translations_from_dir(locales_dir: str) -> dict:
+    """Helper to load translations from a given directory"""
+    translations = {}
+    if not os.path.exists(locales_dir):
+        print(f"Warning: No 'locales' directory found in {locales_dir}")
+        return translations
+    for filename in os.listdir(locales_dir):
+        if filename.endswith('.json'):
+            lang = filename.split('.')[0]
+            with open(os.path.join(locales_dir, filename), 'r', encoding='utf-8') as f:
+                translations[lang] = json.load(f)
+    return translations
+
+
+def load_translations() -> dict:
+    """Load translations from locales directory
+
+    Returns:
+        dict: Translations dictionary
+    """
+    # Get the project's root directory (where the script is being run from)
+    project_root = os.path.abspath(os.getcwd())
+    locales_dir = os.path.join(project_root, 'locales')
+    return _load_translations_from_dir(locales_dir)
+
+
+def load_common_translations() -> dict:
+    """Load translations from locales directory in the project
+
+    Returns:
+        dict: Translations dictionary
+    """
+    locales_dir = os.path.join(os.path.dirname(__file__), 'locales')
+    return _load_translations_from_dir(locales_dir)
+
+
+TRANSLATIONS = load_translations()
+COMMON_TRANSLATIONS = load_common_translations()
+
+
+def t(key: str, lang: str = 'ru', common: bool = False, **kwargs: Any) -> str:
+    """Get translation for a key with optional formatting"""
+    try:
+        # Support nested keys like "buttons.start"
+        keys = key.split('.')
+        if common:
+            value = COMMON_TRANSLATIONS[lang]
+        else:
+            value = TRANSLATIONS[lang]
+        for k in keys:
+            value = value[k]
+        
+        return value.format(**kwargs) if kwargs else value
+    except KeyError:
+        # Fallback to English if translation missing
+        if lang != 'en':
+            return t(key, 'en', common=common, **kwargs)
+        return key  # Return the key itself as last resort

telegram_libs-0.1.8/src/telegram_libs/utils.py

@@ -0,0 +1,49 @@
+from telegram import (
+    InlineKeyboardButton,
+    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
+
+    Args:
+        update (Update): Update object
+        lang (str): Language code
+
+    Returns:
+        InlineKeyboardMarkup: Inline keyboard markup
+    """
+    await update.message.reply_text(
+        f"Buying a subscription you will get unlimited access to other {int(BOTS_AMOUNT) - 1} bots, to see all bots click /more"
+    )
+    return [
+        [
+            InlineKeyboardButton(
+                t("subscription.plans.1month", lang, common=True), callback_data="sub_1month"
+            ),
+            InlineKeyboardButton(
+                t("subscription.plans.3months", lang, common=True), callback_data="sub_3months"
+            ),
+        ],
+        [
+            InlineKeyboardButton(
+                t("subscription.plans.1year", lang, common=True), 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/GenerateBackgroundGBot">Generate a Background</a>
+    - <a href="https://t.me/kudapoyti_go_bot">Recommend a place to visit</a>
+    - <a href="https://t.me/TryOnOutfitGBot">Try On Outfit</a>
+    """
+    await update.message.reply_text(message, disable_web_page_preview=True, parse_mode='HTML')

telegram_libs-0.1.1/src/telegram_libs/translation.py

@@ -1,34 +0,0 @@
-import json
-import os
-
-def load_translations():
-    translations = {}
-    # Assume 'locales' is at the project root
-    project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))
-    locales_dir = os.path.join(project_root, 'locales')
-    
-    for filename in os.listdir(locales_dir):
-        if filename.endswith('.json'):
-            lang = filename.split('.')[0]
-            with open(os.path.join(locales_dir, filename), 'r', encoding='utf-8') as f:
-                translations[lang] = json.load(f)
-    
-    return translations
-
-TRANSLATIONS = load_translations()
-
-def t(key, lang='ru', **kwargs):
-    """Get translation for a key with optional formatting"""
-    try:
-        # Support nested keys like "buttons.start"
-        keys = key.split('.')
-        value = TRANSLATIONS[lang]
-        for k in keys:
-            value = value[k]
-        
-        return value.format(**kwargs) if kwargs else value
-    except KeyError:
-        # Fallback to English if translation missing
-        if lang != 'en':
-            return t(key, 'en', **kwargs)
-        return key  # Return the key itself as last resort