telegram_libs 0.1.15__py3-none-any.whl → 0.1.17__py3-none-any.whl

telegram_libs/mongo.py

@@ -1,6 +1,54 @@
 from pymongo.mongo_client import MongoClient
 from pymongo.server_api import ServerApi
-from telegram_libs.constants import MONGO_URI
+from telegram_libs.constants import MONGO_URI, DEBUG
 
-# Create a new client and connect to the server
-mongo_client = MongoClient(MONGO_URI, server_api=ServerApi("1"))
+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/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.15.dist-info → telegram_libs-0.1.17.dist-info}/METADATA

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

{telegram_libs-0.1.15.dist-info → telegram_libs-0.1.17.dist-info}/RECORD

@@ -2,12 +2,12 @@ telegram_libs/__init__.py,sha256=xrsD5r6ZiJxPapHf1UhQ61z2gHtCCWrzW0CZHvlvXRc,82
 telegram_libs/constants.py,sha256=7-L1uK9UKCVAZNKYnqmM4B18JDxfxHMpZgTIjh0rATg,606
 telegram_libs/locales/en.json,sha256=TDueYazTaytYCs8_6Z7-HeulGyqz5ze3Cm4mXYKFX8Q,549
 telegram_libs/locales/ru.json,sha256=bHIuq9MFIhqIEvbf5j5HM1E9egtsjrTTIx50s6C9PpY,785
-telegram_libs/mongo.py,sha256=7UOy_cE0ofIbH7QiiirAjOgo_FM9JImtgxQ8ouEsFeo,245
+telegram_libs/mongo.py,sha256=TzCOB4rXiYQ6Y0ptDXqrJmoGcnZuxYPRPdC7GkDZiDs,2278
 telegram_libs/subscription.py,sha256=d7xmzplUrm1nNlWlkqW6dddOYa3t_7PAM3iPme0K5F0,1690
-telegram_libs/support_handlers.py,sha256=T0ukR2gak0XknkA5RZ8Kjd1KeZAxH1BftQNg6bub5kw,2165
+telegram_libs/support_handlers.py,sha256=SrJGmP9WJp6vpsCE6k0lgs5ftlZR_mRBf-NQThIXkmU,2240
 telegram_libs/translation.py,sha256=8Kb2cgqKKZH4X_i2Le0V_K1imZdoaCzYAca831DOBig,2049
 telegram_libs/utils.py,sha256=0eBy7psB0XJQL9dPUeq5c2Ymg7ZPA54sdYJW3x2fI4M,2231
-telegram_libs-0.1.15.dist-info/LICENSE,sha256=ZXkWPZbCc61L29Gz6ZHPwn1c4Pm0TnfIqtx8jGWi9F4,1069
-telegram_libs-0.1.15.dist-info/METADATA,sha256=vwa1lBhRpsx6KwevbYqQ-kXyaTudqR3FPZvSE4J0FyM,804
-telegram_libs-0.1.15.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-telegram_libs-0.1.15.dist-info/RECORD,,
+telegram_libs-0.1.17.dist-info/LICENSE,sha256=ZXkWPZbCc61L29Gz6ZHPwn1c4Pm0TnfIqtx8jGWi9F4,1069
+telegram_libs-0.1.17.dist-info/METADATA,sha256=fSyfAOLoJ2DeWIDWDdLW7y9P5HCyKolPbJGf3nhCJFE,804
+telegram_libs-0.1.17.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+telegram_libs-0.1.17.dist-info/RECORD,,