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 from flask import Flask, render_template_string, request, redirect, url_for app = Flask(__name__) DATA_FILE = 'data.json' REPO_ID = "Kgshop/zalkar" HF_TOKEN_WRITE = os.getenv("HF_TOKEN") HF_TOKEN_READ = os.getenv("HF_TOKEN_READ") LOGO_URL = "https://cdn-avatars.huggingface.co/v1/production/uploads/67c280ccb9d3dfdee58ecfdd/4IIqefkClYrsa2euUC_Nh.jpeg" 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") if not isinstance(data, dict) or 'products' not in data or 'categories' not in data: # Handle cases where the data structure is not as expected products = data if isinstance(data, list) else [] categories = [] # Attempt to find categories if products have a 'category' key if products: all_categories = {p.get('category') for p in products if p.get('category')} categories = sorted(list(all_categories)) return {'products': products, 'categories': categories} return data except FileNotFoundError: logging.warning("Локальный файл базы данных не найден после скачивания.") return {'products': [], 'categories': []} except json.JSONDecodeError: logging.error("Ошибка: Невозможно декодировать JSON файл.") return {'products': [], 'categories': []} except RepositoryNotFoundError: logging.error("Репозиторий не найден. Создание локальной базы данных.") 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 or not REPO_ID: logging.warning("HF_TOKEN_WRITE или REPO_ID не установлены. Пропуск загрузки резервной копии.") 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 Exception as e: logging.error(f"Ошибка при загрузке резервной копии: {e}") def download_db_from_hf(): if not REPO_ID or not (HF_TOKEN_READ or HF_TOKEN_WRITE): logging.warning("REPO_ID и/или токены Hugging Face не установлены. Пропуск скачивания базы данных.") return try: hf_hub_download( repo_id=REPO_ID, filename=DATA_FILE, repo_type="dataset", token=HF_TOKEN_READ if HF_TOKEN_READ else HF_TOKEN_WRITE, local_dir=".", local_dir_use_symlinks=False ) logging.info("JSON база успешно скачана из Hugging Face.") except RepositoryNotFoundError as e: logging.error(f"Репозиторий не найден: {e}") # Decide if you want to create an empty local file or raise if not os.path.exists(DATA_FILE): logging.info(f"Создание пустого локального файла {DATA_FILE}.") with open(DATA_FILE, 'w', encoding='utf-8') as f: json.dump({'products': [], 'categories': []}, f) raise # Re-raise to be caught by load_data except Exception as e: logging.error(f"Ошибка при скачивании JSON базы: {e}") raise # Re-raise to be caught by load_data def periodic_backup(): while True: upload_db_to_hf() time.sleep(800) @app.route('/') def catalog(): data = load_data() products = data['products'] categories = data['categories'] catalog_html = ''' Zalkar Textile - ткани на заказ

Zalkar Textile

{% for category in categories %} {% endfor %}
{% for product in products %}
{% if product.get('photos') and product['photos']|length > 0 %}
{{ product['name'] }}
{% else %}
No image available
{% endif %}

{{ product['name'] }}

{{ product['price'] }} $

{{ product['description'] }}

{% endfor %}
''' return render_template_string(catalog_html, products=products, categories=categories, repo_id=REPO_ID) @app.route('/product/') def product_detail(index): data = load_data() products = data['products'] try: product = products[index] except IndexError: return "Продукт не найден", 404 detail_html = '''

{{ product['name'] }}

{% if product.get('photos') %} {% for photo in product['photos'] %}
{{ product['name'] }}
{% endfor %} {% else %}
No Image
{% endif %}

Категория: {{ product.get('category', 'Без категории') }}

Цена: {{ product['price'] }} $

Описание: {{ product['description'] }}

Доступные цвета: {{ product.get('colors', ['Не указаны'])|join(', ') }}

''' 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', []) # Ensure lists exist 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 data['categories'] = categories # Update data dict save_data(data) return redirect(url_for('admin')) return "Ошибка: Категория уже существует или не указано название", 400 elif action == 'delete_category': try: category_index = int(request.form.get('category_index')) if 0 <= category_index < len(categories): deleted_category = categories.pop(category_index) data['categories'] = categories # Update data dict # Reassign products in the deleted category for product in products: if product.get('category') == deleted_category: product['category'] = 'Без категории' data['products'] = products # Update data dict save_data(data) return redirect(url_for('admin')) return "Ошибка: Неверный индекс категории", 400 except ValueError: return "Ошибка: Неверный индекс категории", 400 elif action == 'add': name = request.form.get('name', '').strip() price_str = request.form.get('price', '0').replace(',', '.') description = request.form.get('description', '').strip() category = request.form.get('category') photos_files = request.files.getlist('photos') colors = request.form.getlist('colors') photos_list = [] 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 if photos_files: uploads_dir = 'uploads' os.makedirs(uploads_dir, exist_ok=True) api = HfApi(token=HF_TOKEN_WRITE) # Pass token explicitly for photo in photos_files[:10]: # Limit to 10 photos if photo and photo.filename: original_filename = secure_filename(photo.filename) # Prevent overwriting files with the same name if uploaded at different times photo_filename = f"{datetime.now().strftime('%Y%m%d%H%M%S')}_{original_filename}" temp_path = os.path.join(uploads_dir, photo_filename) try: photo.save(temp_path) 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"Добавлено фото {photo_filename} для товара {name}" ) photos_list.append(photo_filename) except Exception as e: logging.error(f"Ошибка загрузки фото {original_filename} на HF: {e}") # Optionally return an error or log and continue finally: if os.path.exists(temp_path): os.remove(temp_path) # Clean up temporary file new_product = { 'name': name, 'price': price, 'description': description, 'category': category if category in categories else 'Без категории', 'photos': photos_list, 'colors': [c.strip() for c in colors if c.strip()] # Clean up color strings } products.append(new_product) data['products'] = products # Update data dict save_data(data) return redirect(url_for('admin')) elif action == 'edit': try: index = int(request.form.get('index')) if not (0 <= index < len(products)): return "Ошибка: Неверный индекс товара", 400 product = products[index] name = request.form.get('name', '').strip() price_str = request.form.get('price', '0').replace(',', '.') description = request.form.get('description', '').strip() category = request.form.get('category') photos_files = request.files.getlist('photos') colors = request.form.getlist('colors') 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 product['name'] = name product['price'] = price product['description'] = description product['category'] = category if category in categories else 'Без категории' product['colors'] = [c.strip() for c in colors if c.strip()] # Clean up color strings # Handle photo updates only if new photos are provided if photos_files and any(photo.filename for photo in photos_files): new_photos_list = [] uploads_dir = 'uploads' os.makedirs(uploads_dir, exist_ok=True) api = HfApi(token=HF_TOKEN_WRITE) # Pass token explicitly for photo in photos_files[:10]: # Limit to 10 photos if photo and photo.filename: original_filename = secure_filename(photo.filename) photo_filename = f"{datetime.now().strftime('%Y%m%d%H%M%S')}_{original_filename}" temp_path = os.path.join(uploads_dir, photo_filename) try: photo.save(temp_path) 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"Обновлено фото {photo_filename} для товара {name}" ) new_photos_list.append(photo_filename) except Exception as e: logging.error(f"Ошибка загрузки обновленного фото {original_filename} на HF: {e}") finally: if os.path.exists(temp_path): os.remove(temp_path) # Replace old photos list only if new photos were successfully uploaded if new_photos_list: product['photos'] = new_photos_list data['products'] = products # Update data dict save_data(data) return redirect(url_for('admin')) except ValueError: return "Ошибка: Неверный индекс товара", 400 elif action == 'delete': try: index = int(request.form.get('index')) if 0 <= index < len(products): del products[index] data['products'] = products # Update data dict save_data(data) return redirect(url_for('admin')) return "Ошибка: Неверный индекс товара", 400 except ValueError: return "Ошибка: Неверный индекс товара", 400 admin_html = ''' Админ-панель

Админ-панель

Добавление нового товара

Управление категориями

Список категорий ({{ categories|length }})

{% for category in categories %}

{{ category }}

{% endfor %} {% if not categories %}

Нет добавленных категорий.

{% endif %}

Управление базой данных

Список товаров ({{ products|length }})

{% for product in products %}

{{ product.get('name', 'Без названия') }}

Категория: {{ product.get('category', 'Без категории') }}

Цена: {{ product.get('price', 0)|float|round(2) }} $

Описание: {{ product.get('description', 'Без описания') }}

Цвета: {{ product.get('colors', ['Не указаны'])|join(', ') }}

{% if product.get('photos') and product['photos']|length > 0 %}
{% for photo in product['photos'] %} {{ product.get('name', 'Фото товара') }} {% endfor %}
{% endif %}
Редактировать товар
{% for color in product.get('colors', []) %}
{% endfor %} {% if not product.get('colors') %}
{% endif %}
{% endfor %} {% if not products %}

Нет добавленных товаров.

{% endif %}
''' return render_template_string(admin_html, products=products, categories=categories, repo_id=REPO_ID) @app.route('/backup', methods=['POST']) def backup(): try: upload_db_to_hf() return "Резервная копия успешно создана.", 200 except Exception as e: return f"Ошибка при создании резервной копии: {e}", 500 @app.route('/download', methods=['GET']) def download(): try: download_db_from_hf() return "База данных успешно скачана.", 200 except Exception as e: return f"Ошибка при скачивании базы данных: {e}", 500 if __name__ == '__main__': if HF_TOKEN_WRITE and REPO_ID: logging.info("Запуск потока периодического резервного копирования.") backup_thread = threading.Thread(target=periodic_backup, daemon=True) backup_thread.start() else: logging.warning("Отсутствуют токены Hugging Face или REPO_ID. Периодическое резервное копирование отключено.") try: # Attempt initial data load to create local file if not exists/download fails load_data() except Exception as e: logging.error(f"Не удалось выполнить начальную загрузку или создать базу данных: {e}") app.run(debug=True, host='0.0.0.0', port=7860)