day of hard work with xkirsch #2

web/db.py

@@ -1,4 +1,3 @@
-
 import sqlite3
 from flask import current_app as app
 
@@ -62,4 +61,41 @@ def fetch_employees():
     employees = [dict_from_row(employee) for employee in employees]
     conn.close()
     app.logger.debug(f"Fetched employees: {employees}")
-    return employees
+    return employees
+
+def fetch_products():
+    conn = get_db_connection()
+    products = conn.execute('SELECT * FROM Produkty').fetchall()
+    products = [dict_from_row(product) for product in products]
+    conn.close()
+    app.logger.debug(f"Fetched products: {products}")
+    return products
+
+def update_product(product_id, nazev, popis, momentalni_zasoba, minimalni_zasoba):
+    conn = get_db_connection()
+    conn.execute('''
+        UPDATE Produkty
+        SET Nazev = ?, Popis = ?, Momentalni_Zasoba = ?, Minimalni_Zasoba = ?
+        WHERE ID_Produktu = ?
+    ''', (nazev, popis, momentalni_zasoba, minimalni_zasoba, product_id))
+    conn.commit()
+    conn.close()
+
+def add_product_stock(product_id, quantity):
+    conn = get_db_connection()
+    conn.execute('''
+        UPDATE Produkty
+        SET Momentalni_Zasoba = Momentalni_Zasoba + ?
+        WHERE ID_Produktu = ?
+    ''', (quantity, product_id))
+    conn.commit()
+    conn.close()
+
+def create_product(nazev, popis, momentalni_zasoba, minimalni_zasoba):
+    conn = get_db_connection()
+    conn.execute('''
+        INSERT INTO Produkty (Nazev, Popis, Momentalni_Zasoba, Minimalni_Zasoba)
+        VALUES (?, ?, ?, ?)
+    ''', (nazev, popis, momentalni_zasoba, minimalni_zasoba))
+    conn.commit()
+    conn.close()