Baseflask4

Paused

File size: 18,155 Bytes

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)

# --- Функции для работы с JSON ---
def load_data():
    """Загружает данные из JSON-файла или скачивает их из репозитория Hugging Face."""
    try:
        # Пытаемся скачать базу данных из Hugging Face
        download_db_from_hf()
        with open(DATA_FILE, 'r', encoding='utf-8') as file:
            return json.load(file)
    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):
    """Сохраняет данные в JSON-файл."""
    try:
        with open(DATA_FILE, 'w', encoding='utf-8') as file:
            json.dump(data, file, ensure_ascii=False, indent=4)
    except Exception as e:
        logging.error(f"Ошибка при сохранении данных: {e}")
        raise

# --- Функции для резервного копирования ---
def upload_db_to_hf():
    """Загружает JSON-файл базы данных в репозиторий Hugging Face."""
    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():
    """Скачивает JSON-файл базы данных из репозитория Hugging Face."""
    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  # Прокидываем исключение выше, чтобы load_data обработал его
    except Exception as e:
        logging.error(f"Ошибка при скачивании JSON базы: {e}")
        raise

def periodic_backup():
    """Периодически вызывает функцию upload_db_to_hf() каждые 15 секунд."""
    while True:
        upload_db_to_hf()
        time.sleep(15)

# --- Маршруты приложения ---
@app.route('/')
def catalog():
    products = load_data()
    catalog_html = '''
    <!DOCTYPE html>
    <html lang="ru">
    <head>
        <meta charset="UTF-8">
        <meta name="viewport" content="width=device-width, initial-scale=1.0">
        <title>Каталог</title>
        <style>
            body {
                font-family: Arial, sans-serif;
                margin: 20px;
                background-color: #f9f9f9;
            }
            h1 {
                color: #333;
            }
            .product {
                background-color: #fff;
                border: 1px solid #ddd;
                padding: 15px;
                margin-bottom: 10px;
                border-radius: 5px;
            }
            .product h2 {
                margin-top: 0;
                color: #555;
            }
            .product p {
                color: #777;
            }
            .product img {
                max-width: 200px;
                margin-top: 10px;
                border: 1px solid #ccc;
                border-radius: 4px;
            }
        </style>
    </head>
    <body>
        <h1>Каталог товаров</h1>
        {% for product in products %}
        <div class="product">
            <h2>{{ product['name'] }}</h2>
            <p><strong>Цена:</strong> {{ product['price'] }} руб.</p>
            <p><strong>Описание:</strong> {{ product['description'] }}</p>
            {% if product.get('photo') %}
                <img src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/photos/{{ product['photo'] }}" alt="{{ product['name'] }}">
            {% endif %}
        </div>
        {% endfor %}
    </body>
    </html>
    '''
    return render_template_string(catalog_html, products=products, 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')
            photo = request.files.get('photo')

            logging.debug(f"Полученные данные из формы: name={name}, price={price}, description={description}")

            photo_filename = None
            if photo and photo.filename:
                photo_filename = secure_filename(photo.filename)
                # Сохраняем файл временно в папку uploads
                uploads_dir = 'uploads'
                os.makedirs(uploads_dir, exist_ok=True)
                temp_path = os.path.join(uploads_dir, photo_filename)
                photo.save(temp_path)

                # Загружаем фото в репозиторий в папку "photos"
                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} {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
                    )
                    logging.info("Фото успешно загружено в репозиторий.")
                except Exception as e:
                    logging.error(f"Ошибка при загрузке фото: {e}")
                    return f"Ошибка при загрузке фото: {e}", 500
                finally:
                    # Удаляем временный файл
                    os.remove(temp_path)

            if name and price and description:
                try:
                    price = float(price.replace(',', '.'))
                except ValueError:
                    logging.error("Ошибка: Цена должна быть числом.")
                    return "Ошибка: Цена должна быть числом.", 400

                product_entry = {
                    'name': name,
                    'price': price,
                    'description': description
                }
                if photo_filename:
                    product_entry['photo'] = photo_filename

                products.append(product_entry)
                save_data(products)
                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')
            photo = request.files.get('photo')

            # Логика обновления фотографии
            if photo:
                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} {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
                    )
                    logging.info("Фото успешно обновлено в репозитории.")
                    products[index]['photo'] = photo_filename  # Обновляем имя файла
                except Exception as e:
                    logging.error(f"Ошибка при загрузке фото: {e}")
                    return f"Ошибка при загрузке фото: {e}", 500
                finally:
                    os.remove(temp_path)
            
            # Обновление остальных полей
            products[index]['name'] = name
            try:
                price = float(price.replace(',', '.'))
            except ValueError:
                logging.error("Ошибка: Цена должна быть числом.")
                return "Ошибка: Цена должна быть числом.", 400
            products[index]['price'] = price
            products[index]['description'] = description

            save_data(products)
            return redirect(url_for('admin'))

        elif action == 'delete':
            index = int(request.form.get('index'))
            del products[index]
            save_data(products)
            return redirect(url_for('admin'))

    admin_html = '''
    <!DOCTYPE html>
    <html lang="ru">
    <head>
        <meta charset="UTF-8">
        <meta name="viewport" content="width=device-width, initial-scale=1.0">
        <title>Админ-панель</title>
        <style>
            body {
                font-family: Arial, sans-serif;
                margin: 20px;
                background-color: #f9f9f9;
            }
            h1 {
                color: #333;
            }
            form {
                background-color: #fff;
                padding: 20px;
                border: 1px solid #ddd;
                border-radius: 5px;
                max-width: 400px;
                margin-bottom: 20px;
            }
            label {
                display: block;
                margin-top: 10px;
                color: #555;
            }
            input, textarea {
                width: 100%;
                padding: 8px;
                margin-top: 5px;
                border: 1px solid #ddd;
                border-radius: 4px;
            }
            button {
                margin-top: 15px;
                padding: 10px 15px;
                background-color: #28a745;
                color: white;
                border: none;
                border-radius: 4px;
                cursor: pointer;
            }
            button:hover {
                background-color: #218838;
            }
            .product-list {
                margin-top: 20px;
            }
            .product-item {
                background-color: #fff;
                border: 1px solid #ddd;
                padding: 15px;
                margin-bottom: 10px;
                border-radius: 5px;
            }
            .edit-form {
                margin-top: 10px;
                padding: 10px;
                border: 1px solid #ddd;
                border-radius: 5px;
                background-color: #f9f9f9;
            }
        </style>
    </head>
    <body>
        <h1>Добавление товара</h1>
        <form method="POST" enctype="multipart/form-data">
            <input type="hidden" name="action" value="add">
            <label for="name">Название товара:</label>
            <input type="text" id="name" name="name" required>

            <label for="price">Цена:</label>
            <input type="number" id="price" name="price" step="0.01" required>

            <label for="description">Описание:</label>
            <textarea id="description" name="description" rows="4" required></textarea>

            <label for="photo">Фотография товара (необязательно):</label>
            <input type="file" id="photo" name="photo" accept="image/*">

            <button type="submit">Добавить товар</button>
        </form>

        <h2>Управление базой данных</h2>
        
        <!-- Кнопки для резервной копии и скачивания -->
        <form method="POST" action="{{ url_for('backup') }}">
          <button type="submit">Создать резервную копию</button>
        </form>

        <form method="GET" action="{{ url_for('download') }}">
          <button type="submit">Скачать базу данных</button>
        </form>

        <h2>Список товаров</h2>
        <div class="product-list">
            {% for product in products %}
            <div class="product-item">
                <h3>{{ product['name'] }}</h3>
                <p><strong>Цена:</strong> {{ product['price'] }} руб.</p>
                <p><strong>Описание:</strong> {{ product['description'] }}</p>
                {% if product.get('photo') %}
                    <img src="https://huggingface.co/datasets/{{ repo_id }}/resolve/main/photos/{{ product['photo'] }}" alt="{{ product['name'] }}" style="max-width: 100px;">
                {% endif %}

                <details>
                    <summary>Редактировать</summary>
                    <form method="POST" enctype="multipart/form-data" class="edit-form">
                        <input type="hidden" name="action" value="edit">
                        <input type="hidden" name="index" value="{{ loop.index0 }}">
                        <label for="name">Название товара:</label>
                        <input type="text" id="name" name="name" value="{{ product['name'] }}" required>

                        <label for="price">Цена:</label>
                        <input type="number" id="price" name="price" step="0.01" value="{{ product['price'] }}" required>

                        <label for="description">Описание:</label>
                        <textarea id="description" name="description" rows="4" required>{{ product['description'] }}</textarea>

                        <label for="photo">Фотография товара (необязательно):</label>
                        <input type="file" id="photo" name="photo" accept="image/*">

                        <button type="submit">Сохранить изменения</button>
                    </form>
                </details>

                <form method="POST">
                    <input type="hidden" name="action" value="delete">
                    <input type="hidden" name="index" value="{{ loop.index0 }}">
                    <button type="submit">Удалить</button>
                </form>
            </div>
            {% endfor %}
        </div>

    </body>
    </html>
    '''
    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}")
        # Здесь можно добавить логику для создания пустой базы данных, если это необходимо
        # Например: save_data([])

    app.run(debug=True, host='0.0.0.0', port=7860)