Update app.py

app.py

@@ -223,142 +223,180 @@ def edit_products():
         if st.button("Сохранить изменения"):
             c.execute('''
             UPDATE products
-            SET name=?, description=?, purchase_price=?, sale_price=?, quantity_in_stock=?
-            WHERE id=?
+            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],))
+            c.execute("DELETE FROM products WHERE id=?", (product[0],))
             conn.commit()
-            st.success("Товар удален!")
-    else:
-        st.write("Нет товаров для редактирования.")
+            st.success("Товар успешно удален!")
 
-# Форма для добавления товаров в корзину
+# Форма отпуска товара
 def add_to_cart():
-    st.title("Добавление товара в корзину")
-    products = c.execute("SELECT id, name, sale_price FROM products WHERE user_id=?", (st.session_state.user_id,)).fetchall()
-
+    st.title("Отпуск товара")
+    products = c.execute("SELECT id, name, sale_price, quantity_in_stock FROM products WHERE user_id=?", 
+                         (st.session_state.user_id,)).fetchall()
+    
     if products:
-        product_names = [p[1] for p in products]
-        product_name = st.selectbox("Выберите товар", product_names)
-        quantity = st.number_input("Количество", min_value=1, step=1)
-
-        if st.button("Добавить в корзину"):
-            product = next(p for p in products if p[1] == product_name)
-            c.execute("INSERT OR REPLACE INTO cart (product_id, quantity) VALUES (?, ?)", (product[0], quantity))
-            conn.commit()
-            st.success("Товар добавлен в корзину!")
-    else:
-        st.write("Нет товаров для добавления в корзину.")
-
-# Форма для оформления продажи
-def checkout():
-    st.title("Оформление продажи")
-    if st.button("Оформить"):
-        sales_details, sale_id, total_amount = record_sales()
-        st.write("Продажа оформлена!")
-        st.write(f"Идентификатор продажи: {sale_id}")
-        st.write(f"Общая сумма: {total_amount:.2f}")
-        st.write("Подробности продажи:")
-        for detail in sales_details:
-            st.write(f"Товар: {detail[0]}, Количество: {detail[1]}, Сумма: {detail[2]:.2f}, Дата: {detail[3]}")
+        for product in products:
+            cols = st.columns(4)
+            with cols[0]:
+                st.write(product[1])  # Название товара
+                
+            with cols[1]:
+                quantity = st.number_input(f"Количество для '{product[1]}'", min_value=1, max_value=product[3], key=f"quantity_{product[0]}")
+                
+            with cols[2]:
+                add_button = st.button(f"Добавить в корзину", key=f"add_{product[0]}")
+                
+                if add_button:
+                    if quantity <= product[3]:
+                        # Обработка добавления товара в корзину
+                        c.execute('''
+                        INSERT INTO cart (product_id, quantity) 
+                        VALUES (?, ?)
+                        ON CONFLICT(product_id) 
+                        DO UPDATE SET quantity = quantity + excluded.quantity
+                        ''', (product[0], quantity))
+
+                        # Уменьшаем количество на складе
+                        c.execute("UPDATE products SET quantity_in_stock = quantity_in_stock - ? WHERE id=?", 
+                                  (quantity, product[0]))
+                        conn.commit()
+                        st.success(f"Товар '{product[1]}' успешно добавлен в корзину!")
+                    else:
+                        st.error(f"Недостаточное количество товара '{product[1]}' на складе!")
+
+            with cols[3]:
+                remove_button = st.button(f"Удалить из корзины", key=f"remove_{product[0]}")
+                
+                if remove_button:
+                    if quantity > 0:
+                        # Обработка удаления товара из корзины
+                        current_quantity = c.execute("SELECT quantity FROM cart WHERE product_id=?", (product[0],)).fetchone()
+                        if current_quantity:
+                            new_quantity = current_quantity[0] - quantity
+                            if new_quantity > 0:
+                                c.execute('''
+                                INSERT INTO cart (product_id, quantity) 
+                                VALUES (?, ?)
+                                ON CONFLICT(product_id) 
+                                DO UPDATE SET quantity = excluded.quantity
+                                ''', (product[0], new_quantity))
+                            else:
+                                c.execute("DELETE FROM cart WHERE product_id=?", (product[0],))
+
+                            # Увеличиваем количество на складе
+                            c.execute("UPDATE products SET quantity_in_stock = quantity_in_stock + ? WHERE id=?", 
+                                      (quantity, product[0]))
+                            conn.commit()
+                            st.success(f"Товар '{product[1]}' успешно удалён из корзины!")
+                        else:
+                            st.error(f"Товар '{product[1]}' не найден в корзине.")
+                    else:
+                        st.error(f"Введите количество для удаления товара '{product[1]}'.")
+
+        # Отображение состояния корзины
+        st.subheader("Состояние корзины")
+        items, total_quantity, total_price = get_cart_summary(st.session_state.user_id)
+        
+        if items:
+            # Создание DataFrame для таблицы
+            df = pd.DataFrame(items, columns=["Название", "Количество", "Цена за единицу", "Итого"])
+            st.dataframe(df.style.format({"Цена за единицу": "{:.2f}", "Итого": "{:.2f}"}), use_container_width=True)
+            
+            st.write(f"Общее количество: {total_quantity}, Общая стоимость: {total_price:.2f}")
+            
+            # Добавляем кнопку "Пробить" для оформления продажи
+            if st.button("Пробить"):
+                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({"Сумма": "{:.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("Корзина успешно пробита! Все товары добавлены в отчет и корзина очищена.")
+                
+            # Кнопка для отображения сделок
+            if st.button("Сделки"):
+                sales_data = get_all_sales()
+                if sales_data:
+                    sale_ids = [sale[0] for sale in sales_data]
+                    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=["Название товара", "Количество", "Цена за единицу", "Дата и время"])
+                            st.dataframe(df_sales.style.format({"Дата и время": lambda x: pd.to_datetime(x).strftime('%d-%m-%Y %H:%M:%S')}), use_container_width=True)
+                        else:
+                            st.info("Нет деталей для отображения.")
+                else:
+                    st.info("Нет сделок для отображения.")
+        else:
+            st.info("Корзина пуста.")
 
-# Форма для генерации отчета за месяц
+# Страница отчета о продажах за месяц
 def monthly_report():
-    st.title("Ежемесячный отчет")
+    st.title("Отчет о продажах за месяц")
     sales_data, total_sales, total_profit = generate_monthly_report()
-    st.write(f"Общая сумма продаж: {total_sales:.2f}")
-    st.write(f"Общая прибыль: {total_profit:.2f}")
-    st.write("Детали продаж:")
-    for data in sales_data:
-        st.write(f"Товар: {data[0]}, Количество: {data[1]}, Сумма продаж: {data[2]:.2f}, Прибыль: {data[4]:.2f}")
-
-# Форма для просмотра всех сделок
-def view_sales():
-    st.title("Просмотр всех сделок")
-    sales_data = get_all_sales()
-    for sale in sales_data:
-        st.write(f"Продажа ID: {sale[0]}, Дата: {sale[1]}, Сумма: {sale[2]:.2f}")
-
-# Форма для просмотра подробностей сделки
-def view_sale_details():
-    st.title("Детали сделки")
-    sale_id = st.text_input("Введите ID сделки")
-
-    if st.button("Показать детали"):
-        sale_details = get_sale_details(sale_id)
-        if sale_details:
-            st.write(f"Детали сделки ID: {sale_id}")
-            for detail in sale_details:
-                st.write(f"Товар: {detail[0]}, Количество: {detail[1]}, Цена: {detail[2]:.2f}, Дата: {detail[3]}")
-        else:
-            st.error("Сделка не найдена.")
+    if sales_data:
+        # Создание DataFrame для отчета
+        df = pd.DataFrame(sales_data, columns=["Название", "Общее количество", "Отпускная цена", "Общие продажи", "Прибыль"])
+        st.write(f"**Отчет за {datetime.now().strftime('%B %Y')}**")
+        st.dataframe(df.style.format({"Отпускная цена": "{:.2f}", "Общие продажи": "{:.2f}", "Прибыль": "{:.2f}"}), use_container_width=True)
+        
+        st.write(f"**Общая сумма продаж**: {total_sales:.2f}")
+        st.write(f"**Общая сумма прибыли**: {total_profit:.2f}")
+        
+        # Кнопка для отображения сделок
+        if st.button("Сделки"):
+            sales_data = get_all_sales()
+            if sales_data:
+                sale_ids = [sale[0] for sale in sales_data]
+                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=["Название товара", "Количество", "Цена за единицу", "Дата и время"])
+                        st.dataframe(df_sales.style.format({"Дата и время": lambda x: pd.to_datetime(x).strftime('%d-%m-%Y %H:%M:%S')}), use_container_width=True)
+                    else:
+                        st.info("Нет деталей для отображения.")
+            else:
+                st.info("Нет сделок для отображения.")
+    else:
+        st.info("Нет данных о продажах за этот месяц.")
 
-# Основной блок
+# Главная функция приложения
 def main():
-    st.sidebar.title("Меню")
-    options = ["Авторизация", "Регистрация", "Добавление товара", "Редактирование товара", "Добавление в корзину", "Оформление продажи", "Ежемесячный отчет", "Просмотр всех сделок", "Детали сделки"]
-    page = st.sidebar.radio("Выберите страницу", options)
-
-    if "logged_in" not in st.session_state:
+    if 'logged_in' not in st.session_state:
         st.session_state.logged_in = False
 
-    if page == "Авторизация":
-        if not st.session_state.logged_in:
-            login()
-        else:
-            st.sidebar.write(f"Вы уже авторизованы как {st.session_state.username}")
-
-    elif page == "Регистрация":
-        if not st.session_state.logged_in:
-            register()
-        else:
-            st.sidebar.write(f"Вы уже авторизованы как {st.session_state.username}")
+    if st.session_state.logged_in:
+        st.sidebar.title(f"Привет, {st.session_state.username}!")
+        option = st.sidebar.selectbox("Выберите действие", ["Добавить товар", "Отпуск товара", "Редактировать товары", "Отчет за месяц", "Выйти"])
 
-    elif page == "Добавление товара":
-        if st.session_state.logged_in:
+        if option == "Добавить товар":
             add_product()
-        else:
-            st.error("Пожалуйста, авторизуйтесь для добавления товара.")
-
-    elif page == "Редактирование товара":
-        if st.session_state.logged_in:
-            edit_products()
-        else:
-            st.error("Пожалуйста, авторизуйтесь для редактирования товара.")
-
-    elif page == "Добавление в корзину":
-        if st.session_state.logged_in:
+        elif option == "Отпуск товара":
             add_to_cart()
-        else:
-            st.error("Пожалуйста, авторизуйтесь для добавления товара в корзину.")
-
-    elif page == "Оформление продажи":
-        if st.session_state.logged_in:
-            checkout()
-        else:
-            st.error("Пожалуйста, авторизуйтесь для оформления продажи.")
-
-    elif page == "Ежемесячный отчет":
-        if st.session_state.logged_in:
+        elif option == "Редактировать товары":
+            edit_products()
+        elif option == "Отчет за месяц":
             monthly_report()
-        else:
-            st.error("Пожалуйста, авторизуйтесь для просмотра отчета.")
-
-    elif page == "Просмотр всех сделок":
-        if st.session_state.logged_in:
-            view_sales()
-        else:
-            st.error("Пожалуйста, авторизуйтесь для просмотра сделок.")
-
-    elif page == "Детали сделки":
-        if st.session_state.logged_in:
-            view_sale_details()
-        else:
-            st.error("Пожалуйста, авторизуйтесь для просмотра деталей сделки.")
+        elif option == "Выйти":
+            st.session_state.logged_in = False
+            st.session_state.username = None
+            st.session_state.user_id = None
+            st.success("Вы вышли из системы!")
+    else:
+        page = st.sidebar.selectbox("Выберите страницу", ["Регистрация", "Авторизация"])
+        if page == "Регистрация":
+            register()
+        elif page == "Авторизация":
+            login()
 
 if __name__ == "__main__":
     main()