from flask import Flask, render_template_string, request, redirect, url_for, send_file, flash, jsonify
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, HfHubHTTPError
from werkzeug.utils import secure_filename
from dotenv import load_dotenv
import requests
import uuid
load_dotenv()
app = Flask(__name__)
app.secret_key = 'your_unique_secret_key_meka_shop_12345_no_login'
DATA_FILE = 'data.json'
PHOTOS_DIR = 'photos'
SYNC_FILES = [DATA_FILE] # Only data.json for now, photos are managed separately
REPO_ID = "Kgshop/nizhbel"
HF_TOKEN_WRITE = os.getenv("HF_TOKEN")
HF_TOKEN_READ = os.getenv("HF_TOKEN_READ")
STORE_ADDRESS = "Рынок Кербент, 6 ряд , 3 контейнер / 5 ряд 25 контейнер "
CURRENCY_CODE = 'KGS'
CURRENCY_NAME = 'Кыргызский сом'
DOWNLOAD_RETRIES = 3
DOWNLOAD_DELAY = 5
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
def download_hf_file(filename, subfolder=None, retries=DOWNLOAD_RETRIES, delay=DOWNLOAD_DELAY):
token_to_use = HF_TOKEN_READ if HF_TOKEN_READ else HF_TOKEN_WRITE
path_in_repo = os.path.join(subfolder, filename) if subfolder else filename
local_path = os.path.join(subfolder, filename) if subfolder else filename
if subfolder and not os.path.exists(subfolder):
os.makedirs(subfolder, exist_ok=True)
for attempt in range(retries + 1):
try:
logging.info(f"Downloading {path_in_repo} (Attempt {attempt + 1}/{retries + 1})...")
downloaded_path = hf_hub_download(
repo_id=REPO_ID,
filename=path_in_repo,
repo_type="dataset",
token=token_to_use,
local_dir=".",
local_dir_use_symlinks=False,
force_download=True,
resume_download=False
)
logging.info(f"Successfully downloaded {path_in_repo} to {downloaded_path}.")
return downloaded_path
except RepositoryNotFoundError:
logging.error(f"Repository {REPO_ID} not found. Download cancelled for {path_in_repo}.")
return None
except HfHubHTTPError as e:
if e.response.status_code == 404:
logging.warning(f"File {path_in_repo} not found in repo {REPO_ID} (404). Skipping.")
return None
else:
logging.error(f"HTTP error downloading {path_in_repo} (Attempt {attempt + 1}): {e}. Retrying in {delay}s...")
except requests.exceptions.RequestException as e:
logging.error(f"Network error downloading {path_in_repo} (Attempt {attempt + 1}): {e}. Retrying in {delay}s...")
except Exception as e:
logging.error(f"Unexpected error downloading {path_in_repo} (Attempt {attempt + 1}): {e}. Retrying in {delay}s...", exc_info=True)
if attempt < retries:
time.sleep(delay)
logging.error(f"Failed to download {path_in_repo} after {retries + 1} attempts.")
return None
def upload_hf_file(local_path, path_in_repo, commit_message):
if not HF_TOKEN_WRITE:
logging.warning("HF_TOKEN (for writing) not set. Skipping upload to Hugging Face.")
return False
try:
api = HfApi()
if os.path.exists(local_path):
api.upload_file(
path_or_fileobj=local_path,
path_in_repo=path_in_repo,
repo_id=REPO_ID,
repo_type="dataset",
token=HF_TOKEN_WRITE,
commit_message=commit_message
)
logging.info(f"File {local_path} successfully uploaded to Hugging Face as {path_in_repo}.")
return True
else:
logging.warning(f"Local file {local_path} not found, skipping upload.")
return False
except Exception as e:
logging.error(f"Error uploading file {local_path} to Hugging Face as {path_in_repo}: {e}", exc_info=True)
return False
def delete_hf_files(paths_in_repo, commit_message):
if not HF_TOKEN_WRITE:
logging.warning("HF_TOKEN (for writing) not set. Skipping file deletion on Hugging Face.")
return False
try:
api = HfApi()
api.delete_files(
repo_id=REPO_ID,
paths_in_repo=paths_in_repo,
repo_type="dataset",
token=HF_TOKEN_WRITE,
commit_message=commit_message
)
logging.info(f"Files {paths_in_repo} successfully deleted from Hugging Face.")
return True
except Exception as e:
logging.error(f"Error deleting files {paths_in_repo} from Hugging Face: {e}", exc_info=True)
return False
def download_data_file():
return download_hf_file(DATA_FILE)
def upload_data_file():
return upload_hf_file(DATA_FILE, DATA_FILE, f"Sync {DATA_FILE} {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
def periodic_backup():
backup_interval = 1800
logging.info(f"Setting up periodic backup every {backup_interval} seconds.")
while True:
time.sleep(backup_interval)
logging.info("Starting periodic backup...")
upload_data_file()
logging.info("Periodic backup finished.")
def load_data():
default_data = {'products': [], 'categories': [], 'orders': {}}
try:
with open(DATA_FILE, 'r', encoding='utf-8') as file:
data = json.load(file)
logging.info(f"Local data loaded successfully from {DATA_FILE}")
if not isinstance(data, dict):
logging.warning(f"Local {DATA_FILE} is not a dictionary. Attempting download.")
raise FileNotFoundError
if 'products' not in data: data['products'] = []
if 'categories' not in data: data['categories'] = []
if 'orders' not in data: data['orders'] = {}
return data
except FileNotFoundError:
logging.warning(f"Local file {DATA_FILE} not found. Attempting download from HF.")
except json.JSONDecodeError:
logging.error(f"Error decoding JSON in local {DATA_FILE}. File might be corrupt. Attempting download.")
if download_data_file():
try:
with open(DATA_FILE, 'r', encoding='utf-8') as file:
data = json.load(file)
logging.info(f"Data loaded successfully from {DATA_FILE} after download.")
if not isinstance(data, dict):
logging.error(f"Downloaded {DATA_FILE} is not a dictionary. Using default.")
return default_data
if 'products' not in data: data['products'] = []
if 'categories' not in data: data['categories'] = []
if 'orders' not in data: data['orders'] = {}
return data
except FileNotFoundError:
logging.error(f"File {DATA_FILE} still not found even after download reported success. Using default.")
return default_data
except json.JSONDecodeError:
logging.error(f"Error decoding JSON in downloaded {DATA_FILE}. Using default.")
return default_data
except Exception as e:
logging.error(f"Unknown error loading downloaded {DATA_FILE}: {e}. Using default.", exc_info=True)
return default_data
else:
logging.error(f"Failed to download {DATA_FILE} from HF after retries. Using empty default data structure.")
if not os.path.exists(DATA_FILE):
try:
with open(DATA_FILE, 'w', encoding='utf-8') as f:
json.dump(default_data, f)
logging.info(f"Created empty local file {DATA_FILE} after failed download.")
except Exception as create_e:
logging.error(f"Failed to create empty local file {DATA_FILE}: {create_e}")
return default_data
def save_data(data):
try:
if not isinstance(data, dict):
logging.error("Attempted to save invalid data structure (not a dict). Aborting save.")
return
if 'products' not in data: data['products'] = []
if 'categories' not in data: data['categories'] = []
if 'orders' not in data: data['orders'] = {}
with open(DATA_FILE, 'w', encoding='utf-8') as file:
json.dump(data, file, ensure_ascii=False, indent=4)
logging.info(f"Data successfully saved to {DATA_FILE}")
upload_data_file()
except Exception as e:
logging.error(f"Error saving data to {DATA_FILE}: {e}", exc_info=True)
CATALOG_TEMPLATE = '''
Meka Shop - Каталог
Meka Shop
Наш адрес: {{ store_address }}
{% for category in categories %}
{% endfor %}
{% for product in products %}
{% if product.get('is_top', False) %}
Топ
{% endif %}
{% if product.get('photos') and product['photos']|length > 0 %}
{% else %}
{% endif %}
{% with messages = get_flashed_messages(with_categories=true) %}
{% if messages %}
{% for category, message in messages %}
{{ message }}
{% endfor %}
{% endif %}
{% endwith %}
Синхронизация с Датацентром
Резервное копирование происходит автоматически каждые 30 минут, а также после каждого сохранения данных. Используйте эти кнопки для немедленной синхронизации.
Управление категориями
Добавить новую категорию
Существующие категории:
{% if categories %}
{% for category in categories %}
{{ category }}
{% endfor %}
{% else %}
Категорий пока нет.
{% endif %}
Информация
Управление пользователями отключено, так как сайт не требует входа.
Заказы создаются анонимно и должны быть подтверждены через WhatsApp.
Управление товарами
Добавить новый товар
Список товаров:
{% if products %}
{% for product in products %}
{% if product.get('photos') %}
{% else %}
{% endif %}
{{ product['name'] }}
{% if product.get('in_stock', True) %}
В наличии
{% else %}
Нет в наличии
{% endif %}
{% if product.get('is_top', False) %}
Топ
{% endif %}
'''
@app.route('/')
def catalog():
data = load_data()
all_products = data.get('products', [])
categories = sorted(data.get('categories', []))
products_in_stock = [p for p in all_products if p.get('in_stock', True)]
products_sorted = sorted(products_in_stock, key=lambda p: (not p.get('is_top', False), p.get('name', '').lower()))
return render_template_string(
CATALOG_TEMPLATE,
products=products_sorted,
categories=categories,
repo_id=REPO_ID,
store_address=STORE_ADDRESS,
currency_code=CURRENCY_CODE
)
@app.route('/product/')
def product_detail(index):
data = load_data()
all_products = data.get('products', [])
products_in_stock = [p for p in all_products if p.get('in_stock', True)]
products_sorted = sorted(products_in_stock, key=lambda p: (not p.get('is_top', False), p.get('name', '').lower()))
try:
product = products_sorted[index]
except IndexError:
logging.warning(f"Attempted access to non-existent or out-of-stock product with index {index}")
return "Товар не найден или отсутствует в наличии.", 404
return render_template_string(
PRODUCT_DETAIL_TEMPLATE,
product=product,
repo_id=REPO_ID,
currency_code=CURRENCY_CODE
)
@app.route('/create_order', methods=['POST'])
def create_order():
order_data = request.get_json()
if not order_data or 'cart' not in order_data or not order_data['cart']:
logging.warning("Create order request missing cart data or cart is empty.")
return jsonify({"error": "Корзина пуста или не передана."}), 400
cart_items = order_data['cart']
total_price = 0
processed_cart = []
for item in cart_items:
if not all(k in item for k in ('name', 'price', 'quantity')):
logging.error(f"Invalid cart item structure received: {item}")
return jsonify({"error": "Неверный формат товара в корзине."}), 400
try:
price = float(item['price'])
quantity = int(item['quantity'])
if price < 0 or quantity <= 0:
raise ValueError("Invalid price or quantity")
total_price += price * quantity
processed_cart.append({
"name": item['name'],
"price": price,
"quantity": quantity,
"color": item.get('color', 'N/A'),
"photo": item.get('photo'),
"photo_url": f"https://huggingface.co/datasets/{REPO_ID}/resolve/main/photos/{item['photo']}" if item.get('photo') else "https://via.placeholder.com/60x60.png?text=N/A"
})
except (ValueError, TypeError) as e:
logging.error(f"Invalid price/quantity in cart item: {item}. Error: {e}")
return jsonify({"error": "Неверная цена или количество в товаре."}), 400
order_id = f"{datetime.now().strftime('%Y%m%d%H%M%S')}-{uuid.uuid4().hex[:6]}"
order_timestamp = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
new_order = {
"id": order_id,
"created_at": order_timestamp,
"cart": processed_cart,
"total_price": round(total_price, 2),
"user_info": None,
"status": "new"
}
try:
data = load_data()
if 'orders' not in data or not isinstance(data.get('orders'), dict):
data['orders'] = {}
data['orders'][order_id] = new_order
save_data(data)
logging.info(f"Order {order_id} created successfully (anonymously).")
return jsonify({"order_id": order_id}), 201
except Exception as e:
logging.error(f"Failed to save order {order_id}: {e}", exc_info=True)
return jsonify({"error": "Ошибка сервера при сохранении заказа."}), 500
@app.route('/order/')
def view_order(order_id):
data = load_data()
order = data.get('orders', {}).get(order_id)
if order:
logging.info(f"Displaying order {order_id}")
else:
logging.warning(f"Order {order_id} not found.")
return render_template_string(ORDER_TEMPLATE,
order=order,
repo_id=REPO_ID,
currency_code=CURRENCY_CODE)
@app.route('/admin', methods=['GET', 'POST'])
def admin():
data = load_data()
products = data.get('products', [])
categories = data.get('categories', [])
if 'orders' not in data or not isinstance(data.get('orders'), dict):
data['orders'] = {}
if request.method == 'POST':
action = request.form.get('action')
logging.info(f"Admin action received: {action}")
try:
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)
data['categories'] = categories
save_data(data)
logging.info(f"Category '{category_name}' added.")
flash(f"Категория '{category_name}' успешно добавлена.", 'success')
elif not category_name:
logging.warning("Attempted to add empty category.")
flash("Название категории не может быть пустым.", 'error')
else:
logging.warning(f"Category '{category_name}' already exists.")
flash(f"Категория '{category_name}' уже существует.", 'error')
elif action == 'delete_category':
category_to_delete = request.form.get('category_name')
if category_to_delete and category_to_delete in categories:
categories.remove(category_to_delete)
updated_count = 0
for product in products:
if product.get('category') == category_to_delete:
product['category'] = 'Без категории'
updated_count += 1
data['categories'] = categories
data['products'] = products
save_data(data)
logging.info(f"Category '{category_to_delete}' deleted. Updated products: {updated_count}.")
flash(f"Категория '{category_to_delete}' удалена. {updated_count} товаров обновлено.", 'success')
else:
logging.warning(f"Attempted to delete non-existent or empty category: {category_to_delete}")
flash(f"Не удалось удалить категорию '{category_to_delete}'.", 'error')
elif action == 'add_product':
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')
colors = [c.strip() for c in request.form.getlist('colors') if c.strip()]
in_stock = 'in_stock' in request.form
is_top = 'is_top' in request.form
if not name or not price_str:
flash("Название и цена товара обязательны.", 'error')
return redirect(url_for('admin'))
try:
price = round(float(price_str), 2)
if price < 0: price = 0
except ValueError:
flash("Неверный формат цены.", 'error')
return redirect(url_for('admin'))
photos_list = []
if photos_files and any(f.filename for f in photos_files):
os.makedirs(PHOTOS_DIR, exist_ok=True)
photo_limit = 10
uploaded_count = 0
for photo in photos_files:
if uploaded_count >= photo_limit:
logging.warning(f"Photo limit ({photo_limit}) reached, ignoring remaining photos.")
flash(f"Загружено только первые {photo_limit} фото.", "warning")
break
if photo and photo.filename:
try:
ext = os.path.splitext(photo.filename)[1].lower()
if ext not in ['.jpg', '.jpeg', '.png', '.gif', '.webp']:
logging.warning(f"Skipping non-image file upload: {photo.filename}")
flash(f"Файл {photo.filename} не является изображением и был пропущен.", "warning")
continue
safe_name = secure_filename(name.replace(' ', '_'))[:50]
photo_filename = f"{safe_name}_{datetime.now().strftime('%Y%m%d%H%M%S%f')}{ext}"
temp_path = os.path.join(PHOTOS_DIR, photo_filename)
photo.save(temp_path)
if upload_hf_file(temp_path, f"photos/{photo_filename}", f"Add photo for product {name}"):
photos_list.append(photo_filename)
else:
flash(f"Ошибка при загрузке фото {photo.filename} на Hugging Face. Проверьте токен.", 'error')
os.remove(temp_path)
uploaded_count += 1
except Exception as e:
logging.error(f"Error processing photo {photo.filename} for product {name}: {e}", exc_info=True)
flash(f"Ошибка при обработке фото {photo.filename}.", 'error')
if os.path.exists(temp_path):
try: os.remove(temp_path)
except OSError: pass
elif photo and not photo.filename:
logging.warning("Received an empty photo file object when adding product.")
try:
if os.path.exists(PHOTOS_DIR) and not os.listdir(PHOTOS_DIR):
os.rmdir(PHOTOS_DIR)
except OSError as e:
logging.warning(f"Could not remove temporary upload directory {PHOTOS_DIR}: {e}")
elif not HF_TOKEN_WRITE and photos_files and any(f.filename for f in photos_files):
flash("HF_TOKEN (write) не настроен. Фотографии не были загружены.", "warning")
new_product = {
'name': name, 'price': price, 'description': description,
'category': category if category in categories else 'Без категории',
'photos': photos_list, 'colors': colors,
'in_stock': in_stock, 'is_top': is_top
}
products.append(new_product)
data['products'] = products
save_data(data)
logging.info(f"Product '{name}' added.")
flash(f"Товар '{name}' успешно добавлен.", 'success')
elif action == 'edit_product':
index_str = request.form.get('index')
if index_str is None:
flash("Ошибка редактирования: индекс товара не передан.", 'error')
return redirect(url_for('admin'))
try:
index = int(index_str)
if not (0 <= index < len(products)):
raise IndexError("Product index out of range")
product_to_edit = products[index]
original_name = product_to_edit.get('name', 'N/A')
except (ValueError, IndexError):
flash(f"Ошибка редактирования: неверный индекс товара '{index_str}'.", 'error')
logging.error(f"Invalid index '{index_str}' for editing. Product list length: {len(products)}")
return redirect(url_for('admin'))
product_to_edit['name'] = request.form.get('name', product_to_edit['name']).strip()
price_str = request.form.get('price', str(product_to_edit['price'])).replace(',', '.')
product_to_edit['description'] = request.form.get('description', product_to_edit.get('description', '')).strip()
category = request.form.get('category')
product_to_edit['category'] = category if category in categories else 'Без категории'
product_to_edit['colors'] = [c.strip() for c in request.form.getlist('colors') if c.strip()]
product_to_edit['in_stock'] = 'in_stock' in request.form
product_to_edit['is_top'] = 'is_top' in request.form
try:
price = round(float(price_str), 2)
if price < 0: price = 0
product_to_edit['price'] = price
except ValueError:
logging.warning(f"Invalid price format '{price_str}' during edit of product {original_name}. Price not changed.")
flash(f"Неверный формат цены для товара '{original_name}'. Цена не изменена.", 'warning')
photos_files = request.files.getlist('photos')
if photos_files and any(f.filename for f in photos_files):
os.makedirs(PHOTOS_DIR, exist_ok=True)
new_photos_list = []
photo_limit = 10
uploaded_count = 0
logging.info(f"Uploading new photos for product {product_to_edit['name']}...")
for photo in photos_files:
if uploaded_count >= photo_limit:
logging.warning(f"Photo limit ({photo_limit}) reached, ignoring remaining photos.")
flash(f"Загружено только первые {photo_limit} фото.", "warning")
break
if photo and photo.filename:
try:
ext = os.path.splitext(photo.filename)[1].lower()
if ext not in ['.jpg', '.jpeg', '.png', '.gif', '.webp']:
logging.warning(f"Skipping non-image file upload during edit: {photo.filename}")
flash(f"Файл {photo.filename} не является изображением и был пропущен.", "warning")
continue
safe_name = secure_filename(product_to_edit['name'].replace(' ', '_'))[:50]
photo_filename = f"{safe_name}_{datetime.now().strftime('%Y%m%d%H%M%S%f')}{ext}"
temp_path = os.path.join(PHOTOS_DIR, photo_filename)
photo.save(temp_path)
if upload_hf_file(temp_path, f"photos/{photo_filename}", f"Update photo for product {product_to_edit['name']}"):
new_photos_list.append(photo_filename)
else:
flash(f"Ошибка при загрузке нового фото {photo.filename} на Hugging Face.", 'error')
os.remove(temp_path)
uploaded_count += 1
except Exception as e:
logging.error(f"Error processing new photo {photo.filename}: {e}", exc_info=True)
flash(f"Ошибка при обработке нового фото {photo.filename}.", 'error')
if os.path.exists(temp_path):
try: os.remove(temp_path)
except OSError: pass
try:
if os.path.exists(PHOTOS_DIR) and not os.listdir(PHOTOS_DIR):
os.rmdir(PHOTOS_DIR)
except OSError as e:
logging.warning(f"Could not remove temporary upload directory {PHOTOS_DIR}: {e}")
if new_photos_list:
logging.info(f"New photo list for product {product_to_edit['name']} generated.")
old_photos = product_to_edit.get('photos', [])
if old_photos:
logging.info(f"Attempting to delete old photos: {old_photos}")
paths_to_delete_on_hf = [f"photos/{p}" for p in old_photos]
if delete_hf_files(paths_to_delete_on_hf, f"Delete old photos for product {product_to_edit['name']}"):
logging.info(f"Old photos for product {product_to_edit['name']} deleted from HF.")
else:
logging.error(f"Failed to delete old photos {old_photos} from HF.")
flash("Не удалось удалить старые фотографии с сервера. Новые фото загружены.", "warning")
product_to_edit['photos'] = new_photos_list
flash("Фотографии товара успешно обновлены.", "success")
elif uploaded_count == 0 and any(f.filename for f in photos_files):
flash("Не удалось загрузить новые фотографии (возможно, неверный формат).", "error")
elif not HF_TOKEN_WRITE and photos_files and any(f.filename for f in photos_files):
flash("HF_TOKEN (write) не настроен. Фотографии не были обновлены.", "warning")
products[index] = product_to_edit
data['products'] = products
save_data(data)
logging.info(f"Product '{original_name}' (index {index}) updated to '{product_to_edit['name']}'.")
flash(f"Товар '{product_to_edit['name']}' успешно обновлен.", 'success')
elif action == 'delete_product':
index_str = request.form.get('index')
if index_str is None:
flash("Ошибка удаления: индекс товара не передан.", 'error')
return redirect(url_for('admin'))
try:
index = int(index_str)
if not (0 <= index < len(products)): raise IndexError("Product index out of range")
deleted_product = products.pop(index)
product_name = deleted_product.get('name', 'N/A')
photos_to_delete = deleted_product.get('photos', [])
if photos_to_delete:
logging.info(f"Attempting to delete photos for product '{product_name}' from HF: {photos_to_delete}")
paths_to_delete_on_hf = [f"photos/{p}" for p in photos_to_delete]
if delete_hf_files(paths_to_delete_on_hf, f"Delete photos for deleted product {product_name}"):
logging.info(f"Photos for product '{product_name}' deleted from HF.")
else:
logging.error(f"Failed to delete photos {photos_to_delete} for product '{product_name}' from HF.")
flash(f"Не удалось удалить фото для товара '{product_name}' с сервера. Товар удален локально.", "warning")
data['products'] = products
save_data(data)
logging.info(f"Product '{product_name}' (original index {index}) deleted.")
flash(f"Товар '{product_name}' удален.", 'success')
except (ValueError, IndexError):
flash(f"Ошибка удаления: неверный индекс товара '{index_str}'.", 'error')
logging.error(f"Invalid index '{index_str}' for deletion. Product list length: {len(products)}")
else:
logging.warning(f"Received unknown admin action: {action}")
flash(f"Неизвестное действие: {action}", 'warning')
return redirect(url_for('admin'))
except Exception as e:
logging.error(f"Произошла внутренняя ошибка при выполнении действия '{action}': {e}", exc_info=True)
flash(f"Произошла внутренняя ошибка при выполнении действия '{action}'. Подробности в логе сервера.", 'error')
return redirect(url_for('admin'))
current_data = load_data()
display_products = sorted(current_data.get('products', []), key=lambda p: p.get('name', '').lower())
display_categories = sorted(current_data.get('categories', []))
return render_template_string(
ADMIN_TEMPLATE,
products=display_products,
categories=display_categories,
repo_id=REPO_ID,
currency_code=CURRENCY_CODE
)
@app.route('/force_upload', methods=['POST'])
def force_upload():
logging.info("Forcing upload to Hugging Face...")
try:
if upload_data_file():
flash("Данные успешно загружены на Hugging Face.", 'success')
else:
flash("Не удалось загрузить данные на Hugging Face. Проверьте логи.", 'error')
except Exception as e:
logging.error(f"Error during forced upload: {e}", exc_info=True)
flash(f"Ошибка при загрузке на Hugging Face: {e}", 'error')
return redirect(url_for('admin'))
@app.route('/force_download', methods=['POST'])
def force_download():
logging.info("Forcing download from Hugging Face...")
try:
if download_data_file():
flash("Данные успешно скачаны с Hugging Face. Локальные файлы обновлены.", 'success')
load_data()
else:
flash("Не удалось скачать данные с Hugging Face после нескольких попыток. Проверьте логи.", 'error')
except Exception as e:
logging.error(f"Error during forced download: {e}", exc_info=True)
flash(f"Ошибка при скачивании с Hugging Face: {e}", 'error')
return redirect(url_for('admin'))
@app.route('/api/db', methods=['GET'])
def api_download_db():
try:
if not os.path.exists(DATA_FILE):
download_data_file() # Ensure it exists locally if not present
if os.path.exists(DATA_FILE):
return send_file(DATA_FILE, as_attachment=True, download_name=DATA_FILE)
else:
return jsonify({"error": "Data file not found locally or on Hugging Face."}), 404
except Exception as e:
logging.error(f"API: Error downloading data file: {e}", exc_info=True)
return jsonify({"error": f"Internal server error: {e}"}), 500
@app.route('/api/db', methods=['POST'])
def api_upload_db():
if 'file' not in request.files:
return jsonify({"error": "No file part in the request."}), 400
file = request.files['file']
if file.filename == '':
return jsonify({"error": "No selected file."}), 400
if file and file.filename.endswith('.json'):
try:
temp_path = DATA_FILE
file.save(temp_path)
if upload_data_file():
return jsonify({"message": "Data file uploaded and synchronized successfully."}), 200
else:
return jsonify({"error": "Failed to upload data file to Hugging Face. Check server logs."}), 500
except Exception as e:
logging.error(f"API: Error uploading data file: {e}", exc_info=True)
return jsonify({"error": f"Internal server error: {e}"}), 500
return jsonify({"error": "Invalid file type. Only JSON files are allowed."}), 400
@app.route('/api/photos', methods=['POST'])
def api_upload_photos():
if not HF_TOKEN_WRITE:
return jsonify({"error": "Hugging Face write token not configured on server."}), 500
if 'photos' not in request.files:
return jsonify({"error": "No 'photos' file part in the request."}), 400
photos = request.files.getlist('photos')
if not photos or all(f.filename == '' for f in photos):
return jsonify({"error": "No selected photos."}), 400
uploaded_filenames = []
failed_uploads = []
os.makedirs(PHOTOS_DIR, exist_ok=True)
for photo in photos:
if photo and photo.filename:
try:
ext = os.path.splitext(photo.filename)[1].lower()
if ext not in ['.jpg', '.jpeg', '.png', '.gif', '.webp']:
failed_uploads.append({"filename": photo.filename, "reason": "Invalid file type"})
continue
safe_name = secure_filename(os.path.splitext(photo.filename)[0])[:50]
photo_filename = f"{safe_name}_{datetime.now().strftime('%Y%m%d%H%M%S%f')}{ext}"
temp_path = os.path.join(PHOTOS_DIR, photo_filename)
photo.save(temp_path)
if upload_hf_file(temp_path, f"photos/{photo_filename}", f"API upload photo: {photo_filename}"):
uploaded_filenames.append(photo_filename)
else:
failed_uploads.append({"filename": photo.filename, "reason": "Hugging Face upload failed"})
os.remove(temp_path)
except Exception as e:
logging.error(f"API: Error processing photo {photo.filename}: {e}", exc_info=True)
failed_uploads.append({"filename": photo.filename, "reason": str(e)})
if os.path.exists(temp_path):
try: os.remove(temp_path)
except OSError: pass
try:
if os.path.exists(PHOTOS_DIR) and not os.listdir(PHOTOS_DIR):
os.rmdir(PHOTOS_DIR)
except OSError as e:
logging.warning(f"Could not remove temporary upload directory {PHOTOS_DIR}: {e}")
if uploaded_filenames:
message = f"Successfully uploaded {len(uploaded_filenames)} photos."
if failed_uploads:
message += f" {len(failed_uploads)} photos failed to upload."
return jsonify({"message": message, "uploaded": uploaded_filenames, "failed": failed_uploads}), 207
return jsonify({"message": message, "uploaded": uploaded_filenames}), 200
else:
return jsonify({"error": "No photos were uploaded.", "failed": failed_uploads}), 400
@app.route('/api/photos/', methods=['GET'])
def api_download_photo(filename):
if not filename:
return jsonify({"error": "Filename is required."}), 400
try:
local_filepath = os.path.join(PHOTOS_DIR, secure_filename(filename))
download_success = download_hf_file(secure_filename(filename), subfolder=PHOTOS_DIR)
if download_success and os.path.exists(local_filepath):
return send_file(local_filepath, as_attachment=True, download_name=filename)
else:
return jsonify({"error": "Photo not found or failed to download."}), 404
except Exception as e:
logging.error(f"API: Error downloading photo {filename}: {e}", exc_info=True)
return jsonify({"error": f"Internal server error: {e}"}), 500
finally:
if os.path.exists(local_filepath):
try:
os.remove(local_filepath)
except OSError as e:
logging.warning(f"Could not remove temporary downloaded photo {local_filepath}: {e}")
@app.route('/api/photos/', methods=['DELETE'])
def api_delete_photo(filename):
if not HF_TOKEN_WRITE:
return jsonify({"error": "Hugging Face write token not configured on server."}), 500
if not filename:
return jsonify({"error": "Filename is required."}), 400
path_in_repo = f"photos/{secure_filename(filename)}"
if delete_hf_files([path_in_repo], f"API delete photo: {secure_filename(filename)}"):
return jsonify({"message": f"Photo '{filename}' deleted successfully from Hugging Face."}), 200
else:
return jsonify({"error": f"Failed to delete photo '{filename}' from Hugging Face. Check server logs."}), 500
if __name__ == '__main__':
logging.info("Application starting up. Performing initial data load/download...")
os.makedirs(PHOTOS_DIR, exist_ok=True) # Ensure photos directory exists
download_data_file() # Only download data.json initially. Photos are fetched on demand or via API.
load_data()
logging.info("Initial data load complete.")
if HF_TOKEN_WRITE:
backup_thread = threading.Thread(target=periodic_backup, daemon=True)
backup_thread.start()
logging.info("Periodic backup thread started.")
else:
logging.warning("Periodic backup will NOT run (HF_TOKEN for writing not set).")
port = int(os.environ.get('PORT', 7860))
logging.info(f"Starting Flask app on host 0.0.0.0 and port {port}")
app.run(debug=False, host='0.0.0.0', port=port)