from flask import Flask, render_template_string, request, redirect, url_for
import json
import os
import logging
import threading
import time
from datetime import datetime
from huggingface_hub import HfApi, hf_hub_download
from huggingface_hub.utils import RepositoryNotFoundError
app = Flask(__name__)
DATA_FILE = 'products.json'
# --- Настройки Hugging Face ---
REPO_ID = "Testbase1/testsett" # Замените на ваш репозиторий
HF_TOKEN_WRITE = os.getenv("HF_TOKEN") # Токен для записи (загрузки)
HF_TOKEN_READ = os.getenv("HF_TOKEN_READ") # Токен для чтения (скачивания)
# Настройка логирования
logging.basicConfig(level=logging.DEBUG)
# --- Функции для работы с JSON ---
def load_data():
"""Загружает данные из JSON-файла или скачивает их из репозитория Hugging Face."""
try:
# Пытаемся скачать базу данных из Hugging Face
download_db_from_hf()
with open(DATA_FILE, 'r', encoding='utf-8') as file:
return json.load(file)
except FileNotFoundError:
logging.warning("Локальный файл базы данных не найден после скачивания.")
return []
except json.JSONDecodeError:
logging.error("Ошибка: Невозможно декодировать JSON файл.")
return []
except RepositoryNotFoundError:
logging.error("Репозиторий не найден. Создание локальной базы данных.")
return [] # Возвращаем пустой список, чтобы начать с пустой базы
except Exception as e:
logging.error(f"Произошла ошибка при загрузке данных: {e}")
return []
def save_data(data):
"""Сохраняет данные в JSON-файл."""
try:
with open(DATA_FILE, 'w', encoding='utf-8') as file:
json.dump(data, file, ensure_ascii=False, indent=4)
except Exception as e:
logging.error(f"Ошибка при сохранении данных: {e}")
raise
# --- Функции для резервного копирования ---
def upload_db_to_hf():
"""Загружает JSON-файл базы данных в репозиторий Hugging Face."""
try:
api = HfApi()
api.upload_file(
path_or_fileobj=DATA_FILE,
path_in_repo=DATA_FILE,
repo_id=REPO_ID,
repo_type="dataset",
token=HF_TOKEN_WRITE,
commit_message=f"Автоматическое резервное копирование базы данных {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
)
logging.info("Резервная копия JSON базы успешно загружена на Hugging Face.")
except Exception as e:
logging.error(f"Ошибка при загрузке резервной копии: {e}")
def download_db_from_hf():
"""Скачивает JSON-файл базы данных из репозитория Hugging Face."""
try:
hf_hub_download(
repo_id=REPO_ID,
filename=DATA_FILE,
repo_type="dataset",
token=HF_TOKEN_READ,
local_dir=".",
local_dir_use_symlinks=False
)
logging.info("JSON база успешно скачана из Hugging Face.")
except RepositoryNotFoundError as e:
logging.error(f"Репозиторий не найден: {e}")
raise # Прокидываем исключение выше, чтобы load_data обработал его
except Exception as e:
logging.error(f"Ошибка при скачивании JSON базы: {e}")
raise # Прокидываем исключение выше, чтобы load_data обработал его
def periodic_backup():
"""Периодически вызывает функцию upload_db_to_hf() каждые 100 секунд."""
while True:
upload_db_to_hf()
time.sleep(15)
# --- Маршруты приложения ---
@app.route('/')
def catalog():
products = load_data()
catalog_html = '''
Каталог
Каталог товаров
{% for product in products %}
{{ product['name'] }}
Цена: {{ product['price'] }} руб.
Описание: {{ product['description'] }}
{% endfor %}
'''
return render_template_string(catalog_html, products=products)
@app.route('/admin', methods=['GET', 'POST'])
def admin():
if request.method == 'POST':
name = request.form.get('name')
price = request.form.get('price')
description = request.form.get('description')
logging.debug(f"Полученные данные из формы: name={name}, price={price}, description={description}")
if name and price and description:
try:
price = float(price.replace(',', '.'))
except ValueError:
logging.error("Ошибка: Цена должна быть числом.")
return "Ошибка: Цена должна быть числом.", 400
products = load_data()
products.append({
'name': name,
'price': price,
'description': description
})
try:
save_data(products)
except Exception as e:
return f"Ошибка при сохранении данных: {e}", 500
return redirect(url_for('admin'))
products = load_data()
admin_html = '''
Админ-панель
Добавление товара
Управление базой данных
'''
return render_template_string(admin_html)
@app.route('/backup', methods=['POST'])
def backup():
"""Маршрут для создания резервной копии."""
upload_db_to_hf()
return "Резервная копия успешно создана.", 200
@app.route('/download', methods=['GET'])
def download():
"""Маршрут для скачивания базы данных."""
download_db_from_hf()
return "База данных успешно скачана.", 200
if __name__ == '__main__':
# Запуск фонового потока для периодического резервного копирования
backup_thread = threading.Thread(target=periodic_backup, daemon=True)
backup_thread.start()
# Попытка загрузить базу данных из репозитория перед запуском приложения
try:
load_data()
except Exception as e:
logging.error(f"Не удалось загрузить базу данных при запуске: {e}")
# Здесь можно добавить логику для создания пустой базы данных, если это необходимо
# Например: save_data([])
app.run(debug=True, host='0.0.0.0', port=7860)