# --- START OF FILE app (1) (2).py ---
from flask import Flask, render_template_string, request, redirect, url_for, send_file
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 = 'data.json'
REPO_ID = "Kgshop/Aigaz"
HF_TOKEN_WRITE = os.getenv("HF_TOKEN")
HF_TOKEN_READ = os.getenv("HF_TOKEN_READ")
LOGO_URL = "https://huggingface.co/spaces/Kgshop/Aigaz/resolve/main/Picsart_25-04-23_11-45-34-036.jpg"
logging.basicConfig(level=logging.INFO)
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")
if not isinstance(data, dict) or 'products' not in data or 'categories' not in data:
logging.warning("Структура JSON некорректна, инициализация.")
return {'products': [], 'categories': [] if not isinstance(data, list) else data}
return data
except FileNotFoundError:
logging.warning("Локальный файл базы данных не найден. Создание новой.")
return {'products': [], 'categories': []}
except json.JSONDecodeError:
logging.error("Ошибка: Невозможно декодировать JSON файл. Создание новой базы.")
return {'products': [], 'categories': []}
except RepositoryNotFoundError:
logging.warning("Репозиторий HF не найден. Используется/Создается локальная база данных.")
if os.path.exists(DATA_FILE):
try:
with open(DATA_FILE, 'r', encoding='utf-8') as file:
data = json.load(file)
if isinstance(data, dict) and 'products' in data and 'categories' in data:
return data
else:
return {'products': [], 'categories': []}
except (FileNotFoundError, json.JSONDecodeError):
return {'products': [], 'categories': []}
else:
return {'products': [], 'categories': []}
except Exception as e:
logging.error(f"Произошла непредвиденная ошибка при загрузке данных: {e}")
return {'products': [], 'categories': []}
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():
if not HF_TOKEN_WRITE:
logging.warning("HF_TOKEN (write) не установлен. Загрузка на Hugging Face пропущена.")
return
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 RepositoryNotFoundError:
logging.error(f"Репозиторий {REPO_ID} не найден. Не удалось загрузить бэкап.")
except Exception as e:
logging.error(f"Ошибка при загрузке резервной копии: {e}")
def download_db_from_hf():
if not HF_TOKEN_READ:
logging.warning("HF_TOKEN_READ не установлен. Скачивание с Hugging Face пропущено.")
return
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,
force_download=True # Добавлено для принудительной загрузки
)
logging.info("JSON база успешно скачана из Hugging Face.")
except RepositoryNotFoundError as e:
logging.info(f"Репозиторий {REPO_ID} не найден на Hugging Face. Скачивание невозможно.")
raise # Перебрасываем, чтобы load_data обработал
except Exception as e:
if "404 Client Error" in str(e) and "EntryNotFound" in str(e):
logging.info(f"Файл {DATA_FILE} не найден в репозитории {REPO_ID}. Скачивание невозможно.")
# Не перебрасываем, так как это ожидаемо, если файл еще не создан
else:
logging.error(f"Ошибка при скачивании JSON базы: {e}")
raise # Перебрасываем другие ошибки
def periodic_backup():
while True:
time.sleep(800)
logging.info("Запуск периодического резервного копирования...")
try:
# Загружаем актуальные данные перед бэкапом, если кто-то мог их изменить
current_data = load_data() # Используем существующую функцию, она уже скачивает с HF если нужно
if current_data: # Только если данные успешно загружены
save_data(current_data) # save_data вызовет upload_db_to_hf
else:
logging.warning("Пропуск периодического бэкапа из-за ошибки загрузки данных.")
except Exception as e:
logging.error(f"Ошибка в потоке периодического бэкапа: {e}")
@app.route('/')
def catalog():
data = load_data()
products = data.get('products', [])
categories = data.get('categories', [])
catalog_html = '''
Aigaz - котельное оборудование
Каталог
{% for category in categories %}
{% endfor %}
{% for product in products %}
{% if product.get('photos') and product['photos']|length > 0 %}
{% else %}
{% endif %}
{{ product['name'] }}
{{ product['price'] }} с
{{ product['description'][:50] }}{% if product['description']|length > 50 %}...{% endif %}
{% set colors = product.get('colors', []) %}
{% if colors %}
Доступные цвета: {{ colors|join(', ') }}
{% else %}
Доступные цвета: Стандартный
{% endif %}
'''
return render_template_string(detail_html, product=product, repo_id=REPO_ID)
@app.route('/admin', methods=['GET', 'POST'])
def admin():
data = load_data()
products = data.get('products', [])
categories = data.get('categories', [])
if request.method == 'POST':
action = request.form.get('action')
if action == 'add_category':
category_name = request.form.get('category_name', '').strip()
if category_name and category_name not in categories:
categories.append(category_name)
categories.sort() # Keep categories sorted
save_data(data)
return redirect(url_for('admin'))
elif not category_name:
return "Ошибка: Название категории не может быть пустым.", 400
else:
return "Ошибка: Категория с таким названием уже существует.", 400
elif action == 'delete_category':
category_index_str = request.form.get('category_index')
try:
category_index = int(category_index_str)
if 0 <= category_index < len(categories):
deleted_category = categories.pop(category_index)
# Update products using this category
for product in products:
if product.get('category') == deleted_category:
product['category'] = 'Без категории'
save_data(data)
return redirect(url_for('admin'))
else:
return "Ошибка: Неверный индекс категории.", 400
except (ValueError, TypeError):
return "Ошибка: Неверный индекс категории.", 400
elif action == 'add' or action == 'edit':
try:
name = request.form.get('name', '').strip()
price_str = request.form.get('price', '').replace(',', '.')
description = request.form.get('description', '').strip()
category = request.form.get('category')
photos_files = request.files.getlist('photos')
# Filter out empty color inputs
colors = [c.strip() for c in request.form.getlist('colors') if c.strip()]
if not name or not price_str or not description:
return "Ошибка: Заполните все обязательные поля (Название, Цена, Описание).", 400
try:
price = float(price_str)
if price < 0:
raise ValueError("Цена не может быть отрицательной.")
except ValueError:
return "Ошибка: Неверный формат цены.", 400
photos_list = []
if action == 'edit':
index = int(request.form.get('index'))
# Keep existing photos if no new ones are uploaded
photos_list = products[index].get('photos', [])
if photos_files and any(f.filename for f in photos_files):
new_photos_uploaded = []
uploads_dir = 'uploads'
os.makedirs(uploads_dir, exist_ok=True)
api = HfApi() if HF_TOKEN_WRITE else None
for photo in photos_files[:10]: # Limit to 10 photos
if photo and photo.filename:
photo_filename = secure_filename(f"{datetime.now().strftime('%Y%m%d%H%M%S')}_{photo.filename}")
temp_path = os.path.join(uploads_dir, photo_filename)
try:
photo.save(temp_path)
if api:
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_uploaded.append(photo_filename)
else:
logging.warning("Пропуск загрузки фото на HF - токен не задан.")
# Decide if you want to store local paths or just skip if HF isn't available
# For now, we'll add the filename assuming it *could* be uploaded later or handled differently
new_photos_uploaded.append(photo_filename)
except Exception as e:
logging.error(f"Ошибка при обработке/загрузке файла {photo_filename}: {e}")
# Decide how to handle partial failures, maybe return an error?
finally:
if os.path.exists(temp_path):
try:
os.remove(temp_path)
except OSError as e:
logging.error(f"Не удалось удалить временный файл {temp_path}: {e}")
if action == 'edit' and new_photos_uploaded:
# Replace existing photos only if new ones were successfully uploaded
photos_list = new_photos_uploaded
elif action == 'add':
photos_list = new_photos_uploaded
product_data = {
'name': name,
'price': price,
'description': description,
'category': category if category in categories else 'Без категории',
'photos': photos_list,
'colors': colors if colors else []
}
if action == 'add':
products.append(product_data)
elif action == 'edit':
if 0 <= index < len(products):
products[index].update(product_data) # Update existing dict
else:
return "Ошибка: Неверный индекс товара для редактирования.", 400
save_data(data)
return redirect(url_for('admin'))
except ValueError as e:
return f"Ошибка ввода данных: {e}", 400
except Exception as e:
logging.error(f"Непредвиденная ошибка при добавлении/редактировании товара: {e}")
return "Внутренняя ошибка сервера", 500
elif action == 'delete':
index_str = request.form.get('index')
try:
index = int(index_str)
if 0 <= index < len(products):
# Optionally: Delete photos from Hugging Face here if needed
# photos_to_delete = products[index].get('photos', [])
# ... logic to delete files using HfApi ...
del products[index]
save_data(data)
return redirect(url_for('admin'))
else:
return "Ошибка: Неверный индекс товара для удаления.", 400
except (ValueError, TypeError):
return "Ошибка: Неверный индекс товара.", 400
admin_html = '''
Админ-панель Aigaz
{% if product.get('photos') and product['photos']|length > 0 %}
{% for photo in product['photos'] %}
{% endfor %}
{% endif %}
Редактировать
{% endfor %}
{% else %}
Товаров пока нет.
{% endif %}
'''
return render_template_string(admin_html, products=products, categories=categories, repo_id=REPO_ID)
@app.route('/backup', methods=['POST'])
def backup():
try:
data = load_data() # Load potentially modified local data first
save_data(data) # This will save locally and then upload to HF
return "Резервная копия успешно создана и загружена на Hugging Face.", 200
except Exception as e:
logging.error(f"Ошибка при ручном создании резервной копии: {e}")
return f"Ошибка при создании резервной копии: {e}", 500
@app.route('/force_download_hf', methods=['POST'])
def force_download_hf():
try:
download_db_from_hf()
return "База данных успешно скачана из Hugging Face и перезаписала локальную версию.", 200
except RepositoryNotFoundError:
return "Ошибка: Репозиторий не найден на Hugging Face.", 404
except Exception as e:
logging.error(f"Ошибка при принудительном скачивании базы данных: {e}")
return f"Ошибка при скачивании базы данных: {e}", 500
@app.route('/download_local', methods=['GET'])
def download_local():
try:
# Ensure the file exists before attempting to send it
if os.path.exists(DATA_FILE):
return send_file(DATA_FILE, as_attachment=True, download_name='data_local_backup.json')
else:
return "Локальный файл базы данных не найден.", 404
except Exception as e:
logging.error(f"Ошибка при скачивании локального файла: {e}")
return "Ошибка при скачивании файла.", 500
if __name__ == '__main__':
# Initial load on startup
try:
load_data()
except Exception as e:
logging.error(f"Не удалось загрузить базу данных при запуске: {e}")
# Start periodic backup in a separate thread
backup_thread = threading.Thread(target=periodic_backup, daemon=True)
backup_thread.start()
# Run the Flask app
# Use Gunicorn or Waitress in production instead of app.run(debug=True)
app.run(host='0.0.0.0', port=7860, debug=False) # Set debug=False for production/semi-production
# --- END OF FILE app (1) (2).py ---