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:
data = json.load(file)
logging.info("Данные успешно загружены из JSON")
return data
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)
logging.info("Данные успешно сохранены в JSON")
upload_db_to_hf()
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('photos') and product['photos']|length > 0 %}
{% endif %}
{{ product['name'] }}
{{ product['price'] }} ₽
{{ product['description'][:100] }}{% if product['description']|length > 100 %}...{% endif %}
{% endfor %}
×
Корзина
Итого: 0 ₽
'''
return render_template_string(catalog_html, products=products, repo_id=REPO_ID)
@app.route('/product/')
def product_detail(index):
products = load_data()
try:
product = products[index]
except IndexError:
return "Продукт не найден", 404
detail_html = '''
{{ product['name'] }}
{% if product.get('photos') %}
{% for photo in product['photos'] %}
{% endfor %}
{% else %}
{% endif %}
Цена: {{ product['price'] }} ₽
Описание: {{ product['description'] }}
'''
return render_template_string(detail_html, product=product, 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')
photos_files = request.files.getlist('photos')
photos_list = []
logging.info(f"Добавление нового товара: name={name}, price={price}, description={description}")
if photos_files:
for i, photo in enumerate(photos_files[:2]):
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}"
)
photos_list.append(photo_filename)
logging.info(f"Фото успешно загружено: {photo_filename}")
except Exception as e:
logging.error(f"Ошибка при загрузке фото: {e}")
return f"Ошибка при загрузке фото: {e}", 500
finally:
if os.path.exists(temp_path):
os.remove(temp_path)
if not name or not price or not description:
logging.error("Ошибка: Не все обязательные поля заполнены")
return "Ошибка: Заполните все обязательные поля", 400
try:
price = float(price.replace(',', '.'))
except ValueError:
logging.error("Ошибка: Цена должна быть числом")
return "Ошибка: Цена должна быть числом", 400
new_product = {
'name': name,
'price': price,
'description': description,
'photos': photos_list
}
products.append(new_product)
logging.info(f"Новый товар добавлен в список: {new_product}")
try:
save_data(products)
logging.info("Товар успешно сохранен в базе данных")
return redirect(url_for('admin'))
except Exception as e:
logging.error(f"Ошибка при сохранении товара: {e}")
return f"Ошибка при сохранении товара: {e}", 500
elif action == 'edit':
index = int(request.form.get('index'))
name = request.form.get('name')
price = request.form.get('price')
description = request.form.get('description')
photos_files = request.files.getlist('photos')
logging.info(f"Редактирование товара с индексом {index}")
if photos_files and any(photo.filename for photo in photos_files):
new_photos_list = []
for i, photo in enumerate(photos_files[:2]):
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}"
)
new_photos_list.append(photo_filename)
logging.info(f"Фото обновлено: {photo_filename}")
except Exception as e:
logging.error(f"Ошибка при загрузке фото: {e}")
return f"Ошибка при загрузке фото: {e}", 500
finally:
if os.path.exists(temp_path):
os.remove(temp_path)
products[index]['photos'] = new_photos_list
products[index]['name'] = name
try:
products[index]['price'] = float(price.replace(',', '.'))
except ValueError:
logging.error("Ошибка: Цена должна быть числом")
return "Ошибка: Цена должна быть числом", 400
products[index]['description'] = description
save_data(products)
logging.info(f"Товар с индексом {index} успешно отредактирован")
return redirect(url_for('admin'))
elif action == 'delete':
index = int(request.form.get('index'))
logging.info(f"Удаление товара с индексом {index}")
del products[index]
save_data(products)
logging.info("Товар успешно удален")
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)