Init commit

utils/database/base.py

@@ -0,0 +1,30 @@
+from .model import Admin, Operator
+
+
+def get_admins():
+    users = []
+    for i in Admin.select():
+        users.append(i.user_id)
+    return users
+
+
+def get_operator():
+    users = []
+    for i in Operator.select():
+        users.append(i.user_id)
+    return users
+
+def get_full_admin():
+    usr = []
+    for i in Admin.select():
+        usr.append({
+            "user_id": i.user_id,
+            "first_name": i.first_name,
+            "last_name": i.last_name,
+            "username": i.username
+        })
+    return usr
+
+
+def del_admin(user_id:int):
+    Admin.delete().where(Admin.user_id == user_id).execute()

utils/database/cart.py

@@ -0,0 +1,23 @@
+from .model import Cart
+
+
+def add_to_cart(user_id:int, item_id: int):
+    if Cart.select().where(Cart.user_id==user_id, Cart.product_id==item_id).exists():
+        count = Cart.get(Cart.user_id==user_id, Cart.product_id==item_id).count
+        Cart.update(count=count+1).where(Cart.user_id==user_id, Cart.product_id==item_id).execute()
+        return
+    Cart.insert(user_id=user_id, product_id=item_id, count=1).execute()
+
+
+def del_from_cart(user_id: int, item_id: int):
+    Cart.delete().where(Cart.user_id==user_id, Cart.product_id==item_id).execute()
+
+
+def clean_cart(user_id: int):
+    Cart.delete().where(Cart.user_id==user_id).execute()
+
+def get_user_cart(user_id:int):
+    cart = []
+    for i in Cart.select().where(Cart.user_id == user_id):
+        cart.append((i.product_id, i.count))
+    return cart

utils/database/market.py

@@ -0,0 +1,49 @@
+from ast import Mod
+import typing as t
+
+from .model import Catalog as Model
+
+
+class Catalog():
+    @staticmethod
+    def __get_item(item_id: int) -> dict:
+        item = {}
+        if item_id is not None:
+            i = Model.get_or_none(Model.id == item_id)
+            if i is not None:
+                item.update(
+                    {
+                        "id": i.id,
+                        "name": i.name,
+                        "description": i.description,
+                        "price": i.price,
+                        "image": i.image
+                    }
+                )
+        return item
+    
+    @classmethod
+    def get_catalog(self, item_id:int = None, get_count:bool = False) -> t.Union[list, dict]:
+        if item_id:
+            if get_count:
+                return self.__get_item(item_id), Model.select().count()
+            return self.__get_item(item_id)
+
+        items = []
+        for i in Model.select():
+            items.append(self.__get_item(i.id))
+        return items
+
+    
+    @staticmethod
+    def delete_post(item_id: int):
+        return Model.delete_by_id(item_id)
+
+
+def add_item(name: str, description: str, price:t.Union[int, float], image:bytes, **kw):
+    Model.insert(
+        name=name,
+        description=description,
+        price=price,
+        image=image
+    ).execute()

utils/database/model.py

@@ -0,0 +1,48 @@
+from peewee import (Model, BigIntegerField, TextField, BlobField, 
+                    IntegerField, CharField, FloatField, ForeignKeyField)
+
+from load import db
+
+
+class BaseModel(Model):
+    '''Base model. Abstract Class'''
+    class Meta:
+        database = db
+
+
+class User(BaseModel):
+    user_id = BigIntegerField(null=False, unique=True)
+    first_name = CharField(null=False, max_length=64)
+    last_name = CharField(null=True, max_length=64)
+    username = CharField(null=True, max_length=32)
+
+
+class Admin(User):
+    pass
+
+
+class Operator(User):
+    pass
+
+
+class Catalog(BaseModel):
+    name = TextField()
+    description = TextField()
+    image = BlobField()
+    price = FloatField()
+
+
+class Cart(BaseModel):
+    user_id = BigIntegerField()
+    product_id = IntegerField()
+    count = IntegerField(default=1)
+
+
+class UserInfo(BaseModel):
+    user_id = BigIntegerField(null=False, unique=True)
+    first_name = TextField()
+    last_name = TextField()
+    phone_number = CharField(15)
+    address = TextField()
+
+db.create_tables([Cart, Catalog, Operator, Admin, User, UserInfo])

utils/database/ordering.py

@@ -0,0 +1,23 @@
+from .model import UserInfo
+
+
+def save_info(user_id: int, last_name:str, first_name:str, phone_number:str, address:str, **kw):
+    UserInfo.insert(
+        user_id=user_id,
+        first_name=first_name,
+        last_name=last_name,
+        phone_number=phone_number,
+        address=address
+    ).on_conflict_replace().execute()
+
+def get_info(user_id: int):
+    output = None
+    if UserInfo.select().where(UserInfo.user_id == user_id).exists():
+        user = UserInfo.get(UserInfo.user_id == user_id)
+        output = {
+            "last_name": user.last_name, 
+            "first_name": user.first_name, 
+            "phone_number": user.phone_number, 
+            "address": user.address
+        }
+    return output

utils/database/user.py

@@ -0,0 +1,45 @@
+from aiogram.types import User as UserType
+from peewee import Model
+
+from .model import User as UserModel
+from .model import Admin, Operator
+from utils import types
+
+
+class Register:
+
+    @classmethod
+    def __register_user(self, model: Model, user: UserType):
+        model.insert(
+            user_id=user.id,
+            first_name=user.first_name,
+            last_name=user.last_name,
+            username=user.username
+        ).on_conflict_replace().execute()
+    
+    @classmethod
+    def register_user(self, user: UserType):
+        return self.__register_user(UserModel, user)
+    
+    @classmethod
+    def register_admin(self, user: UserType):
+        return self.__register_user(Admin, user)
+
+    @classmethod
+    def register_operator(self, user: UserType):
+        return self.__register_user(Operator, user)
+
+
+class User:
+
+    @staticmethod
+    def get_user(user_id: int):
+        if not UserModel.select().where(UserModel.user_id==user_id).exists():
+            return None
+        data = UserModel.get(UserModel.user_id==user_id)
+        return types.User(
+            user_id=data.user_id,
+            first_name=data.first_name,
+            last_name=data.last_name,
+            username=data.username
+        )

utils/helper.py

@@ -0,0 +1,13 @@
+import io
+import typing as t
+
+from load import bot, config
+
+async def download_file(file_path, *args, **kw) -> t.Union[io.BytesIO, t.Any]:
+    if config.telegram_api_server == "https://api.telegram.org":
+        return await bot.download_file(file_path, *args, **kw)
+    else:
+        with open(file_path, 'rb') as f:
+            file = f.read()
+            f.close()
+            return io.BytesIO(file)

utils/json/__init__.py

@@ -0,0 +1,5 @@
+from load import permissions as __permissions
+
+from .model import Permission
+
+permission = Permission(**__permissions)

utils/json/model.py

@@ -0,0 +1,5 @@
+from pydantic import BaseModel
+
+class Permission(BaseModel):
+    can_admin_add_admins: bool
+    can_admin_del_admins: bool

utils/messages.py

@@ -0,0 +1,54 @@
+catalog = '🛍️ Каталог'
+cart = '🛒 Корзина'
+back = "👈 Главное меню" #Назад
+checkout = "📦 Оформить заказ"
+add_to_cart = "➕ Добавить в корзину" 
+delete = "🗑️ Удалить"
+clean_cart = "🗑️ Очистить корзину"
+info = "ℹ️ Информация"
+
+confirm_message = '✅ Подтвердить заказ'
+all_right_message = '✅ Все верно'
+cancel_message = '🚫 Отменить'
+skip_message = '⏩ Пропустить'
+contact = "☎️ Предоставить номер телефона"
+
+save_and_continue = "Сохранить данные и продолжить"
+continue_ = "Продолжить"
+
+rewrite_data = "Ввести заново"
+load_data = "Загрузить данные"
+
+
+#Admins \ Moder
+admin_panel = "Админ панель"
+operator_panel = "Панель оператора"
+add_post = "Добавить пост"
+delete_post = "🗑️ Удалить пост"
+add_admin = "Добавить админа"
+del_admin = "Удалить админа"
+admin_user = (
+    'Имя: [{first_name}](tg://user?id={user_id})\n'
+    'Фамилия: [{last_name}](tg://user?id={user_id})\n'
+    'Имя пользователя: @{username}\n'
+)
+
+
+#balance = '💰 Баланс'
+#delivery_status = '🚚 Статус заказа'
+
+product_message = (
+    "*{name}*\n\n"
+    "{description}\n\n\n"
+    "Цена: {price}"
+)
+cart_message = (
+    "*{name}*\n\n"
+    "{description}\n\n\n"
+    "Количество: {count}\n"
+    "Цена: {price}"
+)
+welcome = (
+    "Привет {user}\n"
+    "Добро пожаловать в магазинчик"
+)

utils/types.py

@@ -0,0 +1,23 @@
+import typing as t
+from abc import ABC
+
+
+class BaseUserFileds(ABC):
+    id: int
+    first_name: str
+    last_name: t.Union[str, None] = None
+    username: t.Union[str, None] = None
+
+
+class User(BaseUserFileds):
+
+    def __new__(self, **kw):
+        if "user_id" in kw:
+            self.id = kw["user_id"]
+        if "first_name" in kw:
+            self.first_name = kw["first_name"]
+        if "last_name" in kw:
+            self.last_name = kw["last_name"]
+        if "username" in kw:
+            self.username = kw["username"]
+        return self