Create app.py

app.py

@@ -0,0 +1,361 @@
+import streamlit as st
+import pandas as pd
+import sqlite3
+import random
+import string
+from datetime import datetime, timedelta
+import pytz
+
+# Подключение к базе данных SQLite3
+conn = sqlite3.connect('auth_system.db', check_same_thread=False)
+c = conn.cursor()
+
+# Создание таблиц, если они не существуют
+c.execute('''
+CREATE TABLE IF NOT EXISTS users (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    username TEXT NOT NULL UNIQUE,
+    token TEXT NOT NULL
+)
+''')
+conn.commit()
+
+c.execute('''
+CREATE TABLE IF NOT EXISTS products (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    name TEXT NOT NULL,
+    description TEXT,
+    purchase_price REAL NOT NULL,
+    sale_price REAL NOT NULL,
+    quantity_in_stock INTEGER NOT NULL DEFAULT 0,
+    user_id INTEGER NOT NULL,
+    FOREIGN KEY(user_id) REFERENCES users(id)
+)
+''')
+conn.commit()
+
+c.execute('''
+CREATE TABLE IF NOT EXISTS cart (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    product_id INTEGER NOT NULL,
+    quantity INTEGER NOT NULL,
+    UNIQUE(product_id)
+)
+''')
+conn.commit()
+
+c.execute('''
+CREATE TABLE IF NOT EXISTS sales (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    product_id INTEGER NOT NULL,
+    quantity INTEGER NOT NULL,
+    sale_date DATE NOT NULL,
+    FOREIGN KEY(product_id) REFERENCES products(id)
+)
+''')
+conn.commit()
+
+# Функция для генерации случайного 13-значного токена
+def generate_token():
+    return ''.join(random.choices(string.ascii_letters + string.digits, k=13))
+
+# Функция для получения идентификатора пользователя по токену
+def get_user_id_by_token(token):
+    c.execute("SELECT id FROM users WHERE token=?", (token,))
+    user = c.fetchone()
+    return user[0] if user else None
+
+# Функция для получения товаров из корзины
+def get_cart_summary(user_id):
+    c.execute('''
+    SELECT p.name, c.quantity, p.sale_price, c.quantity * p.sale_price
+    FROM cart c
+    JOIN products p ON c.product_id = p.id
+    WHERE p.user_id = ?
+    ''', (user_id,))
+    items = c.fetchall()
+    total_quantity = sum(item[1] for item in items)
+    total_price = sum(item[3] for item in items)
+    return items, total_quantity, total_price
+
+# Функция для добавления продажи в отчет
+def record_sales():
+    c.execute('''
+    SELECT product_id, quantity FROM cart
+    ''')
+    cart_items = c.fetchall()
+    for product_id, quantity in cart_items:
+        c.execute('''
+        INSERT INTO sales (product_id, quantity, sale_date)
+        VALUES (?, ?, ?)
+        ''', (product_id, quantity, datetime.now(pytz.timezone('Asia/Bishkek')).strftime('%Y-%m-%d %H:%M:%S')))
+    c.execute('DELETE FROM cart')
+    conn.commit()
+
+# Функция для генерации отчета за месяц
+def generate_monthly_report():
+    start_date = (datetime.now().replace(day=1)).strftime('%Y-%m-%d')
+    end_date = (datetime.now() + timedelta(days=31)).replace(day=1).strftime('%Y-%m-%d')
+    
+    c.execute('''
+    SELECT p.name, SUM(s.quantity) AS total_quantity, p.sale_price, SUM(s.quantity) * p.sale_price AS total_sales, 
+           SUM(s.quantity) * (p.sale_price - p.purchase_price) AS profit
+    FROM sales s
+    JOIN products p ON s.product_id = p.id
+    WHERE s.sale_date BETWEEN ? AND ?
+    GROUP BY p.id
+    ''', (start_date, end_date))
+    sales_data = c.fetchall()
+
+    # Расчёт общей суммы продаж и прибыли
+    total_sales = sum(item[3] for item in sales_data)
+    total_profit = sum(item[4] for item in sales_data)
+    
+    return sales_data, total_sales, total_profit
+
+# Функция для получения всех сделок
+def get_all_sales():
+    c.execute('''
+    SELECT s.id, p.name, s.quantity, s.sale_date
+    FROM sales s
+    JOIN products p ON s.product_id = p.id
+    ORDER BY s.sale_date DESC
+    ''')
+    sales_data = c.fetchall()
+    return sales_data
+
+# Страница регистрации
+def register():
+    st.title('Регистрация')
+    username = st.text_input("Введите ваше имя пользователя")
+    admin_password = st.text_input("Введите пароль администратора", type="password")
+
+    if st.button("Зарегистрироваться"):
+        if admin_password == "1234":
+            token = generate_token()
+            try:
+                c.execute("INSERT INTO users (username, token) VALUES (?, ?)", (username, token))
+                conn.commit()
+                st.success(f"Регистрация успешна! Ваш токен: {token}")
+            except sqlite3.IntegrityError:
+                st.error("Это имя пользователя уже занято. Попробуйте другое.")
+        else:
+            st.error("Неверный пароль администратора!")
+
+# Страница авторизации
+def login():
+    st.title('Авторизация')
+    token_input = st.text_input("Введите ваш токен")
+
+    if st.button("Войти"):
+        user_id = get_user_id_by_token(token_input)
+        if user_id:
+            st.session_state.logged_in = True  # Устанавливаем состояние авторизации
+            st.session_state.username = token_input
+            st.session_state.user_id = user_id
+            st.success("Добро пожаловать!")
+        else:
+            st.error("Неверный токен!")
+
+# Форма добавления товара
+def add_product():
+    st.title("Добавление товара")
+    product_name = st.text_input("Название товара").strip().lower()  # Преобразование в нижний регистр
+    product_description = st.text_area("Описание товара")
+    purchase_price = st.number_input("Приходная цена", min_value=0.0, step=0.01)  # Запрос приходной цены
+    sale_price = st.number_input("Отпускная цена", min_value=0.0, step=0.01)      # Запрос отпускной цены
+    product_quantity = st.number_input("Количество на складе", min_value=0, step=1)
+
+    if st.button("Добавить товар"):
+        if product_name and sale_price:
+            c.execute("INSERT INTO products (name, description, purchase_price, sale_price, quantity_in_stock, user_id) VALUES (?, ?, ?, ?, ?, ?)", 
+                      (product_name, product_description, purchase_price, sale_price, product_quantity, st.session_state.user_id))
+            conn.commit()
+            st.success("Товар успешно добавлен!")
+        else:
+            st.error("Пожалуйста, введите все обязательные данные.")
+
+# Форма редактирования и удаления товара
+def edit_products():
+    st.title("Редактирование товара")
+    products = c.execute("SELECT id, name, description, purchase_price, 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_id = st.selectbox("Выберите товар", product_names)
+        product = next(p for p in products if p[1] == product_id)
+        
+        new_name = st.text_input("Новое название товара", product[1])
+        new_description = st.text_area("Новое описание товара", product[2])
+        new_purchase_price = st.number_input("Новая приходная цена", min_value=0.0, step=0.01, value=product[3])
+        new_sale_price = st.number_input("Новая отпускная цена", min_value=0.0, step=0.01, value=product[4])
+        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]))
+            conn.commit()
+            st.success("Изменения успешно сохранены!")
+
+        if st.button("Удалить товар"):
+            c.execute("DELETE FROM products WHERE id=?", (product[0],))
+            c.execute("DELETE FROM cart WHERE product_id=?", (product[0],))  # Удаляем товар из корзины, если он там есть
+            conn.commit()
+            st.success("Товар успешно удалён!")
+    else:
+        st.info("Нет товаров для редактирования.")
+
+# Форма для добавления и удаления товара в корзину с поиском
+def add_to_cart():
+    st.title("Отпуск товара")
+    
+    search_term = st.text_input("Поиск товара").strip().lower()  # Преобразуем ввод в нижний регистр
+    
+    # Поиск товаров по ключевому слову (без учета регистра)
+    c.execute("SELECT id, name, sale_price, quantity_in_stock FROM products WHERE user_id=? AND LOWER(name) LIKE ?", 
+              (st.session_state.user_id, f"%{search_term}%"))
+    products = c.fetchall()
+    
+    if products:
+        product_list = []
+        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("Пробить"):
+                record_sales()
+                st.success("Корзина успешно пробита! Все товары добавлены в отчет и корзина очищена.")
+                
+            # Кнопка для отображения сделок
+            if st.button("Сделки"):
+                sales_data = get_all_sales()
+                if sales_data:
+                    df_sales = pd.DataFrame(sales_data, columns=["ID", "Название товара", "Количество", "Дата и время"])
+                    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("Корзина пуста.")
+
+# Страница отчета о продажах за месяц
+def monthly_report():
+    st.title("Отчет о продажах за месяц")
+    sales_data, total_sales, total_profit = generate_monthly_report()
+    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:
+                df_sales = pd.DataFrame(sales_data, columns=["ID", "Название товара", "Количество", "Дата и время"])
+                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("Нет данных о продажах за этот месяц.")
+
+# Главная функция приложения
+def main():
+    if 'logged_in' not in st.session_state:
+        st.session_state.logged_in = False
+
+    if st.session_state.logged_in:
+        st.sidebar.title(f"Привет, {st.session_state.username}!")
+        option = st.sidebar.selectbox("Выберите действие", ["Добавить товар", "Отпуск товара", "Редактировать товары", "Отчет за месяц", "Выйти"])
+
+        if option == "Добавить товар":
+            add_product()
+        elif option == "Отпуск товара":
+            add_to_cart()
+        elif option == "Редактировать товары":
+            edit_products()
+        elif option == "Отчет за месяц":
+            monthly_report()
+        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()