New features

utils/database/base.py

@@ -27,10 +27,26 @@ def get_full_admin():
     return usr
 
 
-def del_admin(user_id:int):
+def get_full_operator():
+    usr = []
+    for i in Operator.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()
 
 
+def del_operator(user_id: int):
+    Operator.delete().where(Operator.user_id == user_id).execute()
+
+
 def get_active_operator():
     usr = []
     for i in Operator.select().where(Operator.active==True):

utils/database/cart.py

@@ -1,7 +1,7 @@
 from .model import Cart
 
 
-def add_to_cart(user_id:int, item_id: int):
+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()
@@ -10,15 +10,15 @@ def add_to_cart(user_id:int, item_id: int):
 
 
 def del_from_cart(user_id: int, item_id: int):
-    Cart.delete().where(Cart.user_id==user_id, Cart.product_id==item_id).execute()
+    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()
+    Cart.delete().where(Cart.user_id == user_id).execute()
 
 
-def get_user_cart(user_id:int):
+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
+    return cart

utils/database/market.py

@@ -22,7 +22,7 @@ class Catalog:
         return item
     
     @classmethod
-    def get_catalog(cls, item_id:int = None, get_count:bool = False) -> t.Union[list, dict]:
+    def get_catalog(cls, item_id:int = None, get_count:bool = False) -> t.Union[t.List[t.Any], t.Dict[t.Any, t.Any]]:
         if item_id:
             if get_count:
                 return cls.__get_item(item_id), Model.select().count()

utils/database/ordering.py

@@ -1,7 +1,7 @@
 from .model import UserInfo
 
 
-def save_info(user_id: int, last_name:str, first_name:str, phone_number:str, address:str, **kw):
+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,

utils/messages.py

@@ -15,6 +15,7 @@ contact = "☎️ Предоставить номер телефона"
 
 save_and_continue = "Сохранить данные и продолжить"
 continue_ = "Продолжить"
+order = "Заявка была отправлена!"
 
 rewrite_data = "Ввести заново"
 load_data = "Загрузить данные"
@@ -30,6 +31,8 @@ add_post = "Добавить пост"
 delete_post = "🗑️ Удалить пост"
 add_admin = "Добавить админа"
 del_admin = "Удалить админа"
+add_operator = "Добавить оператора"
+del_operator = "Удалить оператора"
 admin_user = (
     'Имя: [{first_name}](tg://user?id={user_id})\n'
     'Фамилия: [{last_name}](tg://user?id={user_id})\n'