Update app.py

app.py

@@ -51,6 +51,7 @@ CREATE TABLE IF NOT EXISTS sales (
     product_id INTEGER NOT NULL,
     quantity INTEGER NOT NULL,
     sale_date TIMESTAMP NOT NULL,
+    total_amount REAL NOT NULL,
     FOREIGN KEY(product_id) REFERENCES products(id)
 )
 ''')
@@ -88,16 +89,25 @@ def record_sales():
     c.execute('SELECT product_id, quantity FROM cart')
     cart_items = c.fetchall()
     
+    total_amount = 0.0
     for product_id, quantity in cart_items:
+        # Получение цены за единицу товара
+        c.execute('SELECT sale_price FROM products WHERE id=?', (product_id,))
+        sale_price = c.fetchone()[0]
+        
+        # Расчет общей суммы для этого товара
+        amount = quantity * sale_price
+        total_amount += amount
+        
         c.execute('''
-        INSERT INTO sales (sale_id, product_id, quantity, sale_date)
-        VALUES (?, ?, ?, ?)
-        ''', (sale_id, product_id, quantity, sale_date))
-        sales_details.append((product_id, quantity, sale_date))
+        INSERT INTO sales (sale_id, product_id, quantity, sale_date, total_amount)
+        VALUES (?, ?, ?, ?, ?)
+        ''', (sale_id, product_id, quantity, sale_date, amount))
+        sales_details.append((product_id, quantity, amount, sale_date))
         
     c.execute('DELETE FROM cart')
     conn.commit()
-    return sales_details, sale_id
+    return sales_details, sale_id, total_amount
 
 # Функция для генерации отчета за месяц
 def generate_monthly_report():
@@ -123,8 +133,9 @@ def generate_monthly_report():
 # Функция для получения всех сделок
 def get_all_sales():
     c.execute('''
-    SELECT DISTINCT sale_id, sale_date
+    SELECT DISTINCT sale_id, sale_date, SUM(total_amount) AS total_amount
     FROM sales
+    GROUP BY sale_id, sale_date
     ORDER BY sale_date DESC
     ''')
     sales_data = c.fetchall()
@@ -210,10 +221,13 @@ def edit_products():
         new_quantity_in_stock = st.number_input("Новое количество на складе", min_value=0, step=1, value=product[5])
         
         if st.button("Сохранить изменения"):
-            c.execute("UPDATE products SET name=?, description=?, purchase_price=?, sale_price=?, quantity_in_stock=? WHERE id=?", 
-                      (new_name, new_description, new_purchase_price, new_sale_price, new_quantity_in_stock, product[0]))
+            c.execute('''
+            UPDATE products
+            SET name = ?, description = ?, purchase_price = ?, sale_price = ?, quantity_in_stock = ?
+            WHERE id = ?
+            ''', (new_name, new_description, new_purchase_price, new_sale_price, new_quantity_in_stock, product[0]))
             conn.commit()
-            st.success("Изменения успешно сохранены!")
+            st.success("Товар успешно обновлен!")
 
         if st.button("Удалить товар"):
             c.execute("DELETE FROM products WHERE id=?", (product[0],))
@@ -298,11 +312,12 @@ def add_to_cart():
             
             # Добавляем кнопку "Пробить" для оформления продажи
             if st.button("Пробить"):
-                sales_details, sale_id = record_sales()
+                sales_details, sale_id, total_amount = record_sales()
                 if sales_details:
                     st.write("**Детали сделок:**")
-                    sale_details_df = pd.DataFrame(sales_details, columns=["ID товара", "Количество", "Дата и время"])
-                    st.dataframe(sale_details_df.style.format({"Дата и время": lambda x: pd.to_datetime(x).strftime('%d-%m-%Y %H:%M:%S')}), use_container_width=True)
+                    sale_details_df = pd.DataFrame(sales_details, columns=["ID товара", "Количество", "Сумма", "Дата и время"])
+                    st.dataframe(sale_details_df.style.format({"Сумма": "{:.2f}", "Дата и время": lambda x: pd.to_datetime(x).strftime('%d-%m-%Y %H:%M:%S')}), use_container_width=True)
+                    st.write(f"**Общая сумма сделки:** {total_amount:.2f}")
                     st.success("Корзина успешно пробита! Все товары добавлены в отчет и корзина очищена.")
                 
             # Кнопка для отображения сделок
@@ -310,8 +325,8 @@ def add_to_cart():
                 sales_data = get_all_sales()
                 if sales_data:
                     sale_ids = [sale[0] for sale in sales_data]
-                    for sale_id in sale_ids:
-                        st.write(f"**Сделка ID:** {sale_id} ({next(sale for sale in sales_data if sale[0] == sale_id)[1]})")
+                    for sale_id, sale_date, total_amount in sales_data:
+                        st.write(f"**Сделка ID:** {sale_id} ({sale_date}) - **Общая сумма:** {total_amount:.2f}")
                         sale_details = get_sale_details(sale_id)
                         if sale_details:
                             df_sales = pd.DataFrame(sale_details, columns=["Название товара", "Количество", "Цена за единицу", "Дата и время"])
@@ -341,8 +356,8 @@ def monthly_report():
             sales_data = get_all_sales()
             if sales_data:
                 sale_ids = [sale[0] for sale in sales_data]
-                for sale_id in sale_ids:
-                    st.write(f"**Сделка ID:** {sale_id} ({next(sale for sale in sales_data if sale[0] == sale_id)[1]})")
+                for sale_id, sale_date, total_amount in sales_data:
+                    st.write(f"**Сделка ID:** {sale_id} ({sale_date}) - **Общая сумма:** {total_amount:.2f}")
                     sale_details = get_sale_details(sale_id)
                     if sale_details:
                         df_sales = pd.DataFrame(sale_details, columns=["Название товара", "Количество", "Цена за единицу", "Дата и время"])