Update app.py

app.py

@@ -5,7 +5,6 @@ import json
 import random
 import string
 from datetime import datetime, timedelta
-import pytz
 import time
 import threading
 from huggingface_hub import HfApi, hf_hub_download
@@ -13,32 +12,34 @@ from huggingface_hub import HfApi, hf_hub_download
 # --- Настройки Hugging Face ---
 REPO_ID = "Testbase1/testsett"  # Замените на ваш репозиторий
 DB_FILENAME = "auth_system.json"  # JSON база данных
-HF_TOKEN_WRITE = os.getenv("HF_TOKEN")      # Токен для записи (загрузки)
-HF_TOKEN_READ = os.getenv("HF_TOKEN_READ")    # Токен для чтения (скачивания)
+HF_TOKEN_WRITE = os.getenv("HF_TOKEN")  # Токен для записи (загрузки)
+HF_TOKEN_READ = os.getenv("HF_TOKEN_READ")  # Токен для чтения (скачивания)
 
 # Глобальный блокировщик для доступа к JSON-файлу
 db_lock = threading.Lock()
 
-# Функция загрузки базы данных из JSON-файла
+# --- Кеширование функций для повышения производительности ---
+@st.cache_data
 def load_db():
+    """Загружает базу данных из JSON-файла."""
     if not os.path.exists(DB_FILENAME):
         data = {"users": [], "products": [], "cart": [], "sales": []}
         save_db(data)
         return data
-    with open(DB_FILENAME, "r", encoding="utf8") as f:
-        try:
+    try:
+        with open(DB_FILENAME, "r", encoding="utf8") as f:
             data = json.load(f)
-        except json.JSONDecodeError:
-            data = {"users": [], "products": [], "cart": [], "sales": []}
-            save_db(data)
+    except json.JSONDecodeError:
+        data = {"users": [], "products": [], "cart": [], "sales": []}
+        save_db(data)
     return data
 
-# Функция сохранения базы данных в JSON-файл
 def save_db(data):
-    with open(DB_FILENAME, "w", encoding="utf8") as f:
-        json.dump(data, f, ensure_ascii=False, indent=4)
+    """Сохраняет базу данных в JSON-файл."""
+    with db_lock:
+        with open(DB_FILENAME, "w", encoding="utf8") as f:
+            json.dump(data, f, ensure_ascii=False, indent=4)
 
-# --- Функция резервного копирования (загрузка базы на Hugging Face) ---
 def upload_db_to_hf():
     """Загружает JSON-файл базы данных в репозиторий Hugging Face."""
     try:
@@ -48,13 +49,13 @@ def upload_db_to_hf():
             path_in_repo=DB_FILENAME,
             repo_id=REPO_ID,
             repo_type="dataset",
-            token=HF_TOKEN_WRITE
+            token=HF_TOKEN_WRITE,
+            commit_message=f"Автоматическое резервное копирование базы данных {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
         )
         print("Резервная копия JSON базы успешно загружена на Hugging Face.")
     except Exception as e:
         print("Ошибка при загрузке резервной копии:", e)
 
-# --- Функция скачивания базы данных из Hugging Face ---
 def download_db_from_hf():
     """Скачивает JSON-файл базы данных из репозитория Hugging Face."""
     try:
@@ -72,10 +73,9 @@ def download_db_from_hf():
         print("Ошибка при скачивании JSON базы:", e)
         return False
 
-# --- Функция периодического резервного копирования ---
 def periodic_backup():
     """
-    Каждые 30 минут (1800 секунд) вызывается функция upload_db_to_hf().
+    Периодически, каждые 15 секунд, вызывается функция upload_db_to_hf().
     """
     while True:
         upload_db_to_hf()
@@ -83,16 +83,16 @@ def periodic_backup():
 
 # --- Вспомогательные функции ---
 def generate_token():
+    """Генерирует случайный токен."""
     return ''.join(random.choices(string.ascii_letters + string.digits, k=13))
 
 def get_new_id(items):
+    """Возвращает новый ID для элемента."""
     return max([item.get("id", 0) for item in items], default=0) + 1
 
 # --- Функции для работы с продажами ---
 def record_sales():
-    """
-    Записывает информацию о продажах, обновляет базу данных и очищает корзину.
-    """
+    """Записывает информацию о продажах, обновляет базу данных и очищает корзину."""
     with db_lock:
         data = load_db()
         cart_items = data["cart"]
@@ -111,7 +111,6 @@ def record_sales():
                 total_amount += sale_amount
                 sales_details.append([product["id"], item["quantity"], sale_amount, now])
 
-                # Добавляем ��нформацию о продаже в базу данных
                 new_sale_id = get_new_id(data["sales"])
                 data["sales"].append({
                     "id": new_sale_id,
@@ -122,15 +121,14 @@ def record_sales():
                     "sale_date": now
                 })
 
-        # Очищаем корзину
         data["cart"] = []
         save_db(data)
 
         return sales_details, sale_id, total_amount
 
 # --- Страницы приложения ---
-
 def register():
+    """Страница регистрации."""
     st.title('Регистрация')
     username = st.text_input("Введите ваше имя пользователя")
     admin_password = st.text_input("Введите пароль администратора", type="password")
@@ -151,13 +149,13 @@ def register():
             st.error("Неверный пароль администратора!")
 
 def login():
+    """Страница авторизации."""
     st.title('Авторизация')
     token_input = st.text_input("Введите ваш токен")
 
     if st.button("Войти"):
-        with db_lock:
-            data = load_db()
-            user = next((u for u in data["users"] if u["token"] == token_input), None)
+        data = load_db()
+        user = next((u for u in data["users"] if u["token"] == token_input), None)
         if user:
             st.session_state.logged_in = True
             st.session_state.username = user["username"]
@@ -167,6 +165,7 @@ def login():
             st.error("Неверный токен!")
 
 def add_product():
+    """Страница добавления товара."""
     st.title("Добавление товара")
     product_name = st.text_input("Название товара").strip().lower()
     product_description = st.text_area("Описание товара")
@@ -194,10 +193,10 @@ def add_product():
             st.error("Пожалуйста, введите все обязательные данные.")
 
 def edit_products():
+    """Страница редактирования товара."""
     st.title("Редактирование товара")
-    with db_lock:
-        data = load_db()
-        products = [p for p in data["products"] if p["user_id"] == st.session_state.user_id]
+    data = load_db()
+    products = [p for p in data["products"] if p["user_id"] == st.session_state.user_id]
     if products:
         product_names = [p["name"] for p in products]
         selected_product_name = st.selectbox("Выберите товар", product_names)
@@ -233,11 +232,11 @@ def edit_products():
         st.info("У вас пока нет добавленных товаров.")
 
 def add_to_cart():
+    """Страница добавления товара в корзину."""
     st.title("Отпуск товара")
     search_term = st.text_input("Поиск товара").strip().lower()
-    with db_lock:
-        data = load_db()
-        products = [p for p in data["products"] if p["user_id"] == st.session_state.user_id and search_term in p["name"].lower()]
+    data = load_db()
+    products = [p for p in data["products"] if p["user_id"] == st.session_state.user_id and search_term in p["name"].lower()]
     if products:
         for product in products:
             cols = st.columns(4)
@@ -253,7 +252,6 @@ def add_to_cart():
                     if quantity <= product["quantity_in_stock"]:
                         with db_lock:
                             data = load_db()
-                            # Обновляем корзину: если товар уже есть – увеличиваем количество, иначе добавляем новый
                             cart = data["cart"]
                             existing = next((item for item in cart if item["product_id"] == product["id"]), None)
                             if existing:
@@ -261,7 +259,6 @@ def add_to_cart():
                             else:
                                 new_cart_id = get_new_id(cart)
                                 cart.append({"id": new_cart_id, "product_id": product["id"], "quantity": quantity})
-                            # Уменьшаем остаток товара
                             for p in data["products"]:
                                 if p["id"] == product["id"]:
                                     p["quantity_in_stock"] -= quantity
@@ -283,7 +280,6 @@ def add_to_cart():
                                     existing["quantity"] -= quantity
                                 else:
                                     cart.remove(existing)
-                                # Возвращаем товар на склад
                                 for p in data["products"]:
                                     if p["id"] == product["id"]:
                                         p["quantity_in_stock"] += quantity
@@ -294,14 +290,13 @@ def add_to_cart():
                         st.error(f"Введите количество для удаления товара '{product['name']}'.")
 
         st.subheader("Состояние корзины")
-        with db_lock:
-            data = load_db()
-            cart_items = []
-            for item in data["cart"]:
-                prod = next((p for p in data["products"] if p["id"] == item["product_id"]), None)
-                if prod:
-                    total = item["quantity"] * prod["sale_price"]
-                    cart_items.append([prod["name"], item["quantity"], prod["sale_price"], total])
+        data = load_db()
+        cart_items = []
+        for item in data["cart"]:
+            prod = next((p for p in data["products"] if p["id"] == item["product_id"]), None)
+            if prod:
+                total = item["quantity"] * prod["sale_price"]
+                cart_items.append([prod["name"], item["quantity"], prod["sale_price"], total])
         if cart_items:
             df = pd.DataFrame(cart_items, columns=["Название", "Количество", "Цена за единицу", "Итого"])
             st.dataframe(df.style.format({"Цена за единицу": "{:.2f}", "Итого": "{:.2f}"}), use_container_width=True)
@@ -314,7 +309,7 @@ def add_to_cart():
                 if sales_details:
                     st.write("**Детали сделок:**")
                     sale_details_df = pd.DataFrame(sales_details, columns=["ID товара", "Количество", "Сумма", "Дата и время"])
-                    st.dataframe(sale_details_df.style.format({"Сумма": "{:.2f}", 
+                    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}")
@@ -325,39 +320,40 @@ def add_to_cart():
         st.info("По вашему запросу товары не найдены.")
 
 def monthly_report():
+    """Страница отчета о продажах за месяц."""
     st.title("Отчет о продажах за месяц")
-    with db_lock:
-        data = load_db()
-        start_date = datetime.now().replace(day=1)
-        end_date = (datetime.now() + timedelta(days=31)).replace(day=1)
-        sales = []
-        # Для каждого проданного товара ищем информацию о продукте
-        for sale in data["sales"]:
-            sale_date = datetime.strptime(sale["sale_date"], "%Y-%m-%d %H:%M:%S")
-            if start_date <= sale_date < end_date:
-                prod = next((p for p in data["products"] if p["id"] == sale["product_id"]), None)
-                if prod and prod["user_id"] == st.session_state.user_id:
-                    sales.append({
-                        "name": prod["name"],
-                        "quantity": sale["quantity"],
-                        "sale_price": prod["sale_price"],
-                        "total_sales": sale["total_amount"],
-                        "profit": sale["quantity"] * (prod["sale_price"] - prod["purchase_price"])
-                    })
-        # Группировка по имени товара
-        report = {}
-        for s in sales:
-            key = s["name"]
-            if key not in report:
-                report[key] = {"quantity": 0, "sale_price": s["sale_price"], "total_sales": 0, "profit": 0}
-            report[key]["quantity"] += s["quantity"]
-            report[key]["total_sales"] += s["total_sales"]
-            report[key]["profit"] += s["profit"]
-        report_data = []
-        for name, values in report.items():
-            report_data.append([name, values["quantity"], values["sale_price"], values["total_sales"], values["profit"]])
-        total_sales = sum(item[3] for item in report_data)
-        total_profit = sum(item[4] for item in report_data)
+    data = load_db()
+    start_date = datetime.now().replace(day=1)
+    end_date = (datetime.now() + timedelta(days=31)).replace(day=1)
+    sales = []
+
+    for sale in data["sales"]:
+        sale_date = datetime.strptime(sale["sale_date"], "%Y-%m-%d %H:%M:%S")
+        if start_date <= sale_date < end_date:
+            prod = next((p for p in data["products"] if p["id"] == sale["product_id"]), None)
+            if prod and prod["user_id"] == st.session_state.user_id:
+                sales.append({
+                    "name": prod["name"],
+                    "quantity": sale["quantity"],
+                    "sale_price": prod["sale_price"],
+                    "total_sales": sale["total_amount"],
+                    "profit": sale["quantity"] * (prod["sale_price"] - prod["purchase_price"])
+                })
+
+    report = {}
+    for s in sales:
+        key = s["name"]
+        if key not in report:
+            report[key] = {"quantity": 0, "sale_price": s["sale_price"], "total_sales": 0, "profit": 0}
+        report[key]["quantity"] += s["quantity"]
+        report[key]["total_sales"] += s["total_sales"]
+        report[key]["profit"] += s["profit"]
+    report_data = []
+    for name, values in report.items():
+        report_data.append([name, values["quantity"], values["sale_price"], values["total_sales"], values["profit"]])
+    total_sales = sum(item[3] for item in report_data)
+    total_profit = sum(item[4] for item in report_data)
+
     if report_data:
         df = pd.DataFrame(report_data, columns=["Название", "Общее количество", "Отпускная цена", "Общие продажи", "Прибыль"])
         st.write(f"Отчет за {datetime.now().strftime('%B %Y')}")
@@ -366,17 +362,16 @@ def monthly_report():
         st.write(f"**Общая сумма прибыли**: {total_profit:.2f}")
 
         if st.button("Сделки"):
-            with db_lock:
-                data = load_db()
-                sales_grouped = {}
-                for sale in data["sales"]:
-                    prod = next((p for p in data["products"] if p["id"] == sale["product_id"]), None)
-                    if prod and prod["user_id"] == st.session_state.user_id:
-                        key = sale["sale_id"]
-                        if key not in sales_grouped:
-                            sales_grouped[key] = {"sale_date": sale["sale_date"], "total_amount": 0, "details": []}
-                        sales_grouped[key]["total_amount"] += sale["total_amount"]
-                        sales_grouped[key]["details"].append([prod["name"], sale["quantity"], prod["sale_price"], sale["sale_date"]])
+            data = load_db()
+            sales_grouped = {}
+            for sale in data["sales"]:
+                prod = next((p for p in data["products"] if p["id"] == sale["product_id"]), None)
+                if prod and prod["user_id"] == st.session_state.user_id:
+                    key = sale["sale_id"]
+                    if key not in sales_grouped:
+                        sales_grouped[key] = {"sale_date": sale["sale_date"], "total_amount": 0, "details": []}
+                    sales_grouped[key]["total_amount"] += sale["total_amount"]
+                    sales_grouped[key]["details"].append([prod["name"], sale["quantity"], prod["sale_price"], sale["sale_date"]])
             for key, group in sales_grouped.items():
                 st.write(f"**Сделка ID:** {key} ({group['sale_date']}) - **Общая сумма:** {group['total_amount']:.2f}")
                 df_sales = pd.DataFrame(group["details"], columns=["Название товара", "Количество", "Цена за единицу", "Дата и время"])
@@ -387,6 +382,7 @@ def monthly_report():
 
 # --- Главная функция приложения ---
 def main():
+    """Главная функция приложения."""
     st.warning("Попытка скачивания базы данных из Hugging Face...")
     download_success = download_db_from_hf()
     if download_success:
@@ -403,7 +399,7 @@ def main():
             st.error("Не удалось скачать JSON базу и локальной базы нет. Будет создана новая база.")
             with open(DB_FILENAME, "w", encoding="utf8") as f:
                 json.dump({"users": [], "products": [], "cart": [], "sales": []}, f, ensure_ascii=False, indent=4)
-                
+
     if 'logged_in' not in st.session_state:
         st.session_state.logged_in = False
 
@@ -432,8 +428,7 @@ def main():
             login()
 
 if __name__ == "__main__":
-    # Ждем 15 секунд после инициализации, затем запускаем фоновый поток резервного копирования
-    time.sleep(15)
+    # Запускаем фоновый поток резервного копирования сразу
     backup_thread = threading.Thread(target=periodic_backup, daemon=True)
     backup_thread.start()
     main()