telegram_libs 0.1.14__tar.gz → 0.1.16__tar.gz

{telegram_libs-0.1.14 → telegram_libs-0.1.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: telegram_libs
-Version: 0.1.14
+Version: 0.1.16
 Summary: Common libraries for Telegram bots
 Author: Andrey Gritsaenko gricaenko.95a@gmail.com
 Requires-Python: >=3.11,<4.0

{telegram_libs-0.1.14 → telegram_libs-0.1.16}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "telegram_libs"
-version = "0.1.14"
+version = "0.1.16"
 description = "Common libraries for Telegram bots"
 authors = ["Andrey Gritsaenko gricaenko.95a@gmail.com"]
 readme = "README.md"

{telegram_libs-0.1.14 → telegram_libs-0.1.16}/src/telegram_libs/__init__.py

@@ -2,4 +2,4 @@
 Telegram Libs - Common libraries for Telegram bots
 """
 
-__version__ = "0.1.14"
+__version__ = "0.1.15"

{telegram_libs-0.1.14 → telegram_libs-0.1.16}/src/telegram_libs/constants.py

@@ -5,6 +5,7 @@ required_constants = []
 BOTS_AMOUNT = os.getenv("BOTS_AMOUNT")
 MONGO_URI = os.getenv("MONGO_URI")
 SUBSCRIPTION_DB_NAME = os.getenv("SUBSCRIPTION_DB_NAME")
+DEBUG = os.getenv("DEBUG", "False").lower() in ("true", "1", "yes")
 
 required_constants.append(("BOTS_AMOUNT", BOTS_AMOUNT))
 required_constants.append(("MONGO_URI", MONGO_URI))

telegram_libs-0.1.16/src/telegram_libs/mongo.py

@@ -0,0 +1,54 @@
+from pymongo.mongo_client import MongoClient
+from pymongo.server_api import ServerApi
+from telegram_libs.constants import MONGO_URI, DEBUG
+
+mongo_client = MongoClient(MONGO_URI, server_api=ServerApi("1"))
+
+
+class MongoManager:
+    def __init__(self, mongo_database_name: str, **kwargs):
+        self.client = kwargs.get("client") or mongo_client
+        self.db = self.client[mongo_database_name]
+        self.users_collection = self.db["users_test"] if DEBUG else self.db["users"]
+        self.payments_collection = self.db["order_test"] if DEBUG else self.db["order"]
+        self.user_schema = {"user_id": None, **kwargs.get("user_schema")}
+
+    def create_user(self, user_id: int) -> None:
+        """Create a new user in the database."""
+        user_data = self.user_schema.copy()
+        user_data["user_id"] = user_id
+        self.users_collection.insert_one(user_data)
+        return user_data
+
+    def get_user_data(self, user_id: int) -> dict:
+        """Retrieve user data from the database."""
+        user_data = self.users_collection.find_one({"user_id": user_id})
+        if not user_data:
+            # Initialize user data if not found
+            return self.create_user(user_id)
+        return user_data
+
+    def update_user_data(self, user_id: int, updates: dict) -> None:
+        """Update user data in the database."""
+        result = self.users_collection.update_one({"user_id": user_id}, {"$set": updates})
+        if result.matched_count == 0:
+            # If no document was matched, create a new user
+            self.create_user(user_id)
+            self.users_collection.update_one({"user_id": user_id}, {"$set": updates})
+
+    def add_order(self, user_id: int, order: dict) -> None:
+        """Add an order to the user's data."""
+        self.payments_collection.insert_one({"user_id": user_id, **order})
+
+
+    def get_orders(self, user_id: int) -> list:
+        """Get all orders for a user."""
+        orders = self.payments_collection.find({"user_id": user_id})
+        return list(orders)
+
+
+    def update_order(self, user_id: int, order_id: int, updates: dict) -> None:
+        """Update an order for a user."""
+        self.payments_collection.update_one(
+            {"user_id": user_id, "order_id": order_id}, {"$set": updates}
+        )

{telegram_libs-0.1.14 → telegram_libs-0.1.16}/src/telegram_libs/support_handlers.py

@@ -4,6 +4,7 @@ from telegram import Update
 from telegram.ext import ContextTypes, Application, CommandHandler, MessageHandler, filters
 from telegram.ext.filters import BaseFilter
 from telegram_libs.mongo import mongo_client
+from telegram_libs.constants import DEBUG
 from telegram_libs.translation import t
 
 
@@ -22,7 +23,7 @@ async def _handle_user_response(update: Update, context: ContextTypes.DEFAULT_TY
     """Handle user's support message"""
     if context.user_data.get(SUPPORT_WAITING):
         db_name = "support"
-        collection_name = "support"
+        collection_name = "support" if not DEBUG else "support_test"
         message_key = "support.response"
         doc_field_name = "message"
         context_key = SUPPORT_WAITING

{telegram_libs-0.1.14 → telegram_libs-0.1.16}/src/telegram_libs/utils.py

@@ -7,7 +7,6 @@ from telegram import Update
 from telegram.ext import ContextTypes, Application, CommandHandler
 from telegram_libs.constants import BOTS_AMOUNT
 from telegram_libs.translation import t
-from telegram_libs.mongo import mongo_client
 from telegram_libs.support_handlers import register_support_handlers
 
 

telegram_libs-0.1.14/src/telegram_libs/mongo.py

@@ -1,6 +0,0 @@
-from pymongo.mongo_client import MongoClient
-from pymongo.server_api import ServerApi
-from telegram_libs.constants import MONGO_URI
-
-# Create a new client and connect to the server
-mongo_client = MongoClient(MONGO_URI, server_api=ServerApi("1"))