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
from werkzeug.utils import secure_filename
app = Flask(__name__)
DATA_FILE = 'products.json'
# Настройки Hugging Face
REPO_ID = "flpolprojects/Clients"
HF_TOKEN_WRITE = os.getenv("HF_TOKEN")
HF_TOKEN_READ = os.getenv("HF_TOKEN_READ")
# Настройка логирования
logging.basicConfig(level=logging.DEBUG)
def load_data():
try:
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):
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():
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():
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
except Exception as e:
logging.error(f"Ошибка при скачивании JSON базы: {e}")
raise
def periodic_backup():
while True:
upload_db_to_hf()
time.sleep(15)
@app.route('/')
def catalog():
products = load_data()
catalog_html = '''
Каталог
Каталог товаров
{% for product in products %}
{% if product.get('photo') %}
{% endif %}
{{ product['name'] }}
{{ product['price'] }} ₽
{{ product['description'] }}
Подробнее
{% endfor %}
'''
return render_template_string(catalog_html, products=products, repo_id=REPO_ID)
@app.route('/admin', methods=['GET', 'POST'])
def admin():
products = load_data()
if request.method == 'POST':
action = request.form.get('action')
if action == 'add':
name = request.form.get('name')
price = request.form.get('price')
description = request.form.get('description')
photo = request.files.get('photo')
photo_filename = None
if photo and photo.filename:
photo_filename = secure_filename(photo.filename)
uploads_dir = 'uploads'
os.makedirs(uploads_dir, exist_ok=True)
temp_path = os.path.join(uploads_dir, photo_filename)
photo.save(temp_path)
try:
api = HfApi()
api.upload_file(
path_or_fileobj=temp_path,
path_in_repo=f"photos/{photo_filename}",
repo_id=REPO_ID,
repo_type="dataset",
token=HF_TOKEN_WRITE,
commit_message=f"Добавлено фото для товара {name}"
)
except Exception as e:
logging.error(f"Ошибка при загрузке фото: {e}")
return f"Ошибка при загрузке фото: {e}", 500
finally:
os.remove(temp_path)
if name and price and description:
try:
price = float(price.replace(',', '.'))
except ValueError:
return "Ошибка: Цена должна быть числом.", 400
product = {
'name': name,
'price': price,
'description': description
}
if photo_filename:
product['photo'] = photo_filename
products.append(product)
save_data(products)
return redirect(url_for('admin'))
elif action == 'edit':
index = int(request.form.get('index'))
name = request.form.get('name')
price = request.form.get('price')
description = request.form.get('description')
photo = request.files.get('photo')
if photo:
photo_filename = secure_filename(photo.filename)
uploads_dir = 'uploads'
os.makedirs(uploads_dir, exist_ok=True)
temp_path = os.path.join(uploads_dir, photo_filename)
photo.save(temp_path)
try:
api = HfApi()
api.upload_file(
path_or_fileobj=temp_path,
path_in_repo=f"photos/{photo_filename}",
repo_id=REPO_ID,
repo_type="dataset",
token=HF_TOKEN_WRITE,
commit_message=f"Обновлено фото для товара {name}"
)
products[index]['photo'] = photo_filename
except Exception as e:
logging.error(f"Ошибка при загрузке фото: {e}")
return f"Ошибка при загрузке фото: {e}", 500
finally:
os.remove(temp_path)
products[index]['name'] = name
try:
price = float(price.replace(',', '.'))
except ValueError:
return "Ошибка: Цена должна быть числом.", 400
products[index]['price'] = price
products[index]['description'] = description
save_data(products)
return redirect(url_for('admin'))
elif action == 'delete':
index = int(request.form.get('index'))
del products[index]
save_data(products)
return redirect(url_for('admin'))
admin_html = '''
Админ-панель
Добавление товара
Управление базой данных
Список товаров
{% for product in products %}
{% endfor %}
'''
return render_template_string(admin_html, products=products, repo_id=REPO_ID)
@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}")
app.run(debug=True, host='0.0.0.0', port=7860)