Кэширование списка заготовок

App/Waybill/model.py

@@ -1,7 +1,11 @@
 from .db.sql_provider import SQLProvider
 from .db.work import select_list
+from db.DBconnect import DBContextManager
+
 from flask import current_app, session
 from dataclasses import dataclass
+from datetime import date
+from .cache.wrapper import fetch_from_cache
 import os
 
 @dataclass
@@ -23,4 +27,100 @@ def workpiece_list() -> InfoRespronse:
         return InfoRespronse((), 
                              error_message = 'Ошибка в подключении к базе данных. Свяжитесь с администратором', 
                              status=False)
-    return InfoRespronse(result, error_message='', status=True)
+    return InfoRespronse(result, error_message='', status=True)
+
+def form_basket() -> list:
+    current_basket = session.get('basket',{})
+    basket = []
+    for k,v in current_basket.items():
+        _sql = sql_provider.get('one_good.sql', dict(prod_id=k))
+        product = select_list(current_app.config['db_config'], _sql)[0]
+        product['amount'] = v
+        basket.append(product)
+    return basket
+
+def index_basket() -> list:
+    db_config = current_app.config['db_config']
+    cache_config = current_app.config['cache_config']
+
+    cache_select = fetch_from_cache('items_cached', cache_config)(select_list)
+    _sql = sql_provider.get('all_goods.sql', {})
+    products = cache_select(db_config, _sql)
+ 
+    return products
+
+def button_click(request):
+    db_config = current_app.config['db_config']
+    data = dict(prod_id=int(request.form['product_display']))
+
+    _sql = sql_provider.get('one_good.sql', data)
+    result = select_list(db_config, _sql)
+    if result == None:
+        return False
+        
+    product = result[0]
+
+    if request.form.get('buy'):
+        if 'basket' not in session:
+            session['basket'] = dict()
+        
+        if str(product['prod_id']) in session['basket']:
+            pr_id = product['prod_id']
+            amount = int(session['basket'][str(pr_id)])
+            session['basket'][str(pr_id)] = str(amount+1)
+            session.modified = True
+        else:
+            print("NEW PRODUCT")
+            pr_id = product['prod_id']
+            session['basket'][str(pr_id)] = '1'
+            print(session['basket'])
+            session.modified = True
+
+    elif request.form.get('product_display_plus'):
+        # increasing count in basket
+
+        amount = int(session['basket'][str(product['prod_id'])])
+        session['basket'][str(product['prod_id'])] = str(amount + 1)
+        session.modified = True
+
+    elif request.form.get('product_display_minus'):
+        # decreasing count in basket
+        
+        amount = int(session['basket'][str(product['prod_id'])])
+        if amount == 1:
+            session['basket'].pop(str(product['prod_id']))
+        else:
+            session['basket'][str(product['prod_id'])] = str(amount-1)
+        session.modified = True
+    return True
+
+def transaction_order_model(user_id: int, current_date: date):
+
+    db_config = current_app.config['db_config']
+    basket = session.get('basket',{})
+
+    # Чтобы всё это шло как одна транзакция
+    with DBContextManager(db_config) as cursor:
+
+        data = dict(e_user_id=user_id, e_order_date=current_date)
+        
+        _sql = sql_provider.get('create_order.sql', data)
+        try:
+            cursor.execute(_sql)
+        except:
+            return InfoRespronse(tuple(), error_message="Заказ не был создан", status=False)
+        
+        order_id = cursor.lastrowid
+        for key, value in basket.items():
+            _sql = sql_provider.get('insert_order_line.sql',
+                                    dict(e_order_id = order_id,
+                                    e_prod_id = int(key),
+                                    e_amount = int(value)))
+            try:
+                cursor.execute(_sql)
+            except:
+                return InfoRespronse(tuple(), error_message="Заказ не был создан", status=False)
+
+    result = tuple([order_id])
+    clear()
+    return InfoRespronse(result, error_message="", status=True)