from flask import Flask, render_template_string, request, redirect, url_for, session
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
import requests
app = Flask(__name__)
app.secret_key = os.getenv("FLASK_SECRET_KEY", "a_very_secret_key_for_development")
DATA_FILE = 'data_lizabrand.json'
REPO_ID = "Kgshop/clients"
HF_TOKEN_WRITE = os.getenv("HF_TOKEN")
HF_TOKEN_READ = os.getenv("HF_TOKEN_READ")
VK_APP_ID = os.getenv("VK_APP_ID")
VK_APP_SECRET = os.getenv("VK_APP_SECRET")
ADMIN_VK_IDS = os.getenv("ADMIN_VK_IDS", "").split(',')
REDIRECT_URI = "https://lizabrand-optom.hf.space/vk_callback"
LOGO_URL = "https://cdn-avatars.huggingface.co/v1/production/uploads/67c280ccb9d3dfdee58ecfdd/hyc6QWsSZ6FejTAAqDGUS.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:
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.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():
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(800)
@app.route('/login')
def login():
if not VK_APP_ID:
return "VK_APP_ID не настроен на сервере.", 500
vk_auth_url = f"https://oauth.vk.com/authorize?client_id={VK_APP_ID}&display=page&redirect_uri={REDIRECT_URI}&scope=offline&response_type=code&v=5.131"
login_html = f'''
Вход
Войти через ВКонтакте
'''
return render_template_string(login_html)
@app.route('/vk_callback')
def vk_callback():
code = request.args.get('code')
if not code:
return "Ошибка: код авторизации не получен.", 400
if not VK_APP_ID or not VK_APP_SECRET:
return "Учетные данные приложения VK не настроены на сервере.", 500
token_url = 'https://oauth.vk.com/access_token'
params = {
'client_id': VK_APP_ID,
'client_secret': VK_APP_SECRET,
'redirect_uri': REDIRECT_URI,
'code': code
}
try:
response = requests.get(token_url, params=params)
response.raise_for_status()
data = response.json()
except requests.RequestException as e:
return f"Сетевая ошибка при аутентификации: {e}", 500
if 'access_token' in data and 'user_id' in data:
session['vk_user_id'] = data['user_id']
return redirect(url_for('admin'))
else:
error_description = data.get('error_description', 'Неизвестная ошибка')
return f"Ошибка при аутентификации: {error_description}", 400
@app.route('/logout')
def logout():
session.pop('vk_user_id', None)
return redirect(url_for('catalog'))
@app.route('/')
def catalog():
data = load_data()
products = data['products']
categories = data['categories']
catalog_html = '''
Liza Brand - женская одежда оптом
{% for category in categories %}
{% endfor %}
{% for product in products %}
{% if product.get('photos') and product['photos']|length > 0 %}
{% endif %}
{{ product['name'] }}
{{ product['price'] }} с
{{ product['description'][:50] }}{% if product['description']|length > 50 %}...{% endif %}
{% endfor %}
×
Корзина
Итого: 0 с
'''
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'] %}
{% endfor %}
{% else %}
{% 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():
if 'vk_user_id' not in session or str(session.get('vk_user_id')) not in ADMIN_VK_IDS:
return redirect(url_for('login'))
data = load_data()
products = data['products']
categories = data['categories']
if request.method == 'POST':
action = request.form.get('action')
if action == 'add_category':
category_name = request.form.get('category_name')
if category_name and category_name not in categories:
categories.append(category_name)
save_data(data)
return redirect(url_for('admin'))
return "Ошибка: Категория уже существует или не указано название", 400
elif action == 'delete_category':
category_index = int(request.form.get('category_index'))
deleted_category = categories.pop(category_index)
for product in products:
if product.get('category') == deleted_category:
product['category'] = 'Без категории'
save_data(data)
return redirect(url_for('admin'))
elif action == 'add':
name = request.form.get('name')
price = request.form.get('price')
description = request.form.get('description')
category = request.form.get('category')
photos_files = request.files.getlist('photos')
colors = request.form.getlist('colors')
photos_list = []
if photos_files:
for photo in photos_files[:10]:
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)
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)
if os.path.exists(temp_path):
os.remove(temp_path)
if not name or not price or not description:
return "Ошибка: Заполните все обязательные поля", 400
price = float(price.replace(',', '.'))
new_product = {
'name': name,
'price': price,
'description': description,
'category': category if category in categories else 'Без категории',
'photos': photos_list,
'colors': colors if colors else []
}
products.append(new_product)
save_data(data)
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')
category = request.form.get('category')
photos_files = request.files.getlist('photos')
colors = request.form.getlist('colors')
if photos_files and any(photo.filename for photo in photos_files):
new_photos_list = []
for photo in photos_files[:10]:
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)
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)
if os.path.exists(temp_path):
os.remove(temp_path)
products[index]['photos'] = new_photos_list
products[index]['name'] = name
products[index]['price'] = float(price.replace(',', '.'))
products[index]['description'] = description
products[index]['category'] = category if category in categories else 'Без категории'
products[index]['colors'] = colors if colors else []
save_data(data)
return redirect(url_for('admin'))
elif action == 'delete':
index = int(request.form.get('index'))
del products[index]
save_data(data)
return redirect(url_for('admin'))
admin_html = '''
Админ-панель
Добавление товара
Управление категориями
Список категорий
{% for category in categories %}
{{ category }}
{% endfor %}
Управление базой данных
Список товаров
{% for product in products %}
{% endfor %}
'''
return render_template_string(admin_html, products=products, categories=categories, 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)