Initial local state

tgbot/database/db_category.py

@@ -0,0 +1,47 @@
+# - *- coding: utf- 8 - *-
+from typing import Any, Dict, Optional, Union
+
+from sqlalchemy import BigInteger, Integer, String
+from sqlalchemy.orm import Mapped, mapped_column
+
+from tgbot.database.core import Base
+from tgbot.database.entities import Category
+from tgbot.database.repository import BaseRepository
+from tgbot.utils.const_functions import get_unix
+
+
+class CategoryModel(Base):
+    __tablename__ = "storage_category"
+
+    increment: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
+    category_id: Mapped[int] = mapped_column(BigInteger, nullable=False, index=True)
+    category_name: Mapped[str] = mapped_column(String(255), nullable=False)
+    category_unix: Mapped[int] = mapped_column(Integer, nullable=False, default=get_unix)
+
+
+ModelBase = Category
+BaseModel = Category
+
+
+class Categoryx(BaseRepository[CategoryModel, Category]):
+    # Подключение модели категорий
+    def __init__(self):
+        super().__init__()
+        self.table_model = CategoryModel
+        self.entity_model = Category
+        self.storage_name = CategoryModel.__tablename__
+
+    # Добавление категории товара
+    async def add(self, category_id: int, category_name: str) -> Category:
+        return await self._insert(
+            category_id=category_id,
+            category_name=category_name,
+            category_unix=get_unix(),
+        )
+
+    # Обновление категории по ID или фильтру
+    async def update(self, where: Optional[Union[Dict[str, Any], int]] = None, **kwargs) -> int:
+        if isinstance(where, int):
+            where = {"category_id": where}
+
+        return await self._update(where=where, **kwargs)