Spaces:

Testbase1
/

Baseflask

Paused

App Files Files Community

Baseflask / app.py

Aleksmorshen

Update app.py

7c6b9af verified about 1 year ago

raw

history blame contribute delete

10.7 kB

	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

	app = Flask(__name__)
	DATA_FILE = 'products.json'

	# --- Настройки Hugging Face ---
	REPO_ID = "Testbase1/testsett" # Замените на ваш репозиторий
	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 # Прокидываем исключение выше, чтобы load_data обработал его

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

	# --- Маршруты приложения ---
	@app.route('/')
	def catalog():
	products = load_data()
	catalog_html = '''
	<!DOCTYPE html>
	<html lang="en">
	<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;
	}
	</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>
	</div>
	{% endfor %}
	</body>
	</html>
	'''
	return render_template_string(catalog_html, products=products)

	@app.route('/admin', methods=['GET', 'POST'])
	def admin():
	if request.method == 'POST':
	name = request.form.get('name')
	price = request.form.get('price')
	description = request.form.get('description')

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

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

	products = load_data()
	products.append({
	'name': name,
	'price': price,
	'description': description
	})

	try:
	save_data(products)
	except Exception as e:
	return f"Ошибка при сохранении данных: {e}", 500

	return redirect(url_for('admin'))

	products = load_data()
	admin_html = '''
	<!DOCTYPE html>
	<html lang="en">
	<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;
	}
	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;
	}
	</style>
	</head>
	<body>
	<h1>Добавление товара</h1>
	<form method="POST">
	<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>

	<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>

	</body>
	</html>
	'''
	return render_template_string(admin_html)

	@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)