import asyncio import json import os import urllib.parse from datetime import datetime from aiogram import Bot, Dispatcher, types, F from aiogram.filters import Command from aiogram.utils.keyboard import ReplyKeyboardBuilder, InlineKeyboardBuilder from flask import Flask, request, jsonify, render_template_string, redirect import logging import threading from huggingface_hub import HfApi, hf_hub_download from huggingface_hub.utils import RepositoryNotFoundError from werkzeug.utils import secure_filename # Настройка логирования logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__name__) # Инициализация бота и Flask BOT_TOKEN = '7595736142:AAHSU3WGItBkebIgjO293J2WjX5qWAne8Y8' bot = Bot(token=BOT_TOKEN) dp = Dispatcher() app = Flask(__name__) # Путь для хранения данных DATA_FILE = 'data2.json' # Настройки Hugging Face REPO_ID = "flpolprojects/Clients" HF_TOKEN_WRITE = os.getenv("HF_TOKEN") HF_TOKEN_READ = os.getenv("HF_TOKEN_READ") # Функции для работы с данными def load_data(): try: download_db_from_hf() with open(DATA_FILE, 'r', encoding='utf-8') as f: loaded_data = json.load(f) if not (isinstance(loaded_data, dict) and 'products' in loaded_data and 'orders' in loaded_data): logger.error("Неверная структура JSON файла") loaded_data = {'products': [], 'orders': []} if "categories" not in loaded_data: loaded_data["categories"] = [] return loaded_data except Exception as e: logger.error(f"Ошибка при загрузке данных: {e}") return {'products': [], 'orders': [], 'categories': []} def save_data(data): try: with open(DATA_FILE, 'w', encoding='utf-8') as f: json.dump(data, f, ensure_ascii=False, indent=4) # upload_db_to_hf() убрано отсюда except Exception as e: logger.error(f"Ошибка при сохранении данных: {e}") 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')}" ) logger.info("База загружена на Hugging Face") except Exception as e: logger.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 ) logger.info("База скачана из Hugging Face") except Exception as e: logger.error(f"Ошибка при скачивании: {e}") raise # Периодическое копирование каждые 30 секунд def start_periodic_backup(): def backup_loop(): upload_db_to_hf() # Запускаем следующий вызов через 30 секунд threading.Timer(30, backup_loop).start() # Запускаем первый вызов threading.Timer(30, backup_loop).start() logger.info("Периодическое копирование каждые 30 секунд запущено") # Загрузка данных data = load_data() # Формирование клавиатур def get_main_keyboard(): builder = ReplyKeyboardBuilder() builder.button(text="📋 Каталог") builder.button(text="🛒 Корзина") builder.button(text="📦 Заказы") builder.adjust(2) return builder.as_markup(resize_keyboard=True) def get_category_keyboard(): builder = InlineKeyboardBuilder() for category in data['categories']: builder.button(text=category['name'], callback_data=f"cat_{category['id']}") builder.adjust(2) return builder.as_markup() def get_product_keyboard(product_id): builder = InlineKeyboardBuilder() builder.button(text="Добавить в корзину", callback_data=f"add_{product_id}") return builder.as_markup() # Обработчики бота @dp.message(Command("start")) async def cmd_start(message: types.Message): await message.answer("Здравствуйте ! это магазин Routine!. Выберите действие:", reply_markup=get_main_keyboard()) # Изменено условие на "📋 Каталог", чтобы соответствовать кнопке @dp.message(F.text == "📋 Каталог") async def show_categories(message: types.Message): if not data['categories']: await message.answer("Нет доступных категорий.") return await message.answer("Выберите категорию:", reply_markup=get_category_keyboard()) @dp.callback_query(F.data.startswith("cat_")) async def show_products_in_category(callback_query: types.CallbackQuery): try: cat_id = int(callback_query.data.split('_')[1]) products_in_cat = [p for p in data['products'] if p.get('category_id') == cat_id] if not products_in_cat: await bot.send_message(callback_query.from_user.id, "В этой категории нет товаров.") await bot.answer_callback_query(callback_query.id) return async def send_product_batch(products_batch): for product in products_batch: photo_url = f"https://huggingface.co/datasets/{REPO_ID}/resolve/main/photos/{product['photo']}" if product.get('photo') else None caption = f"🏷 {product['name']} - {product['price']} сом\nОписание: {product['description']}\n/id: {product['id']}" try: if photo_url: await bot.send_photo(chat_id=callback_query.from_user.id, photo=photo_url, caption=caption, reply_markup=get_product_keyboard(product['id'])) else: await bot.send_message(callback_query.from_user.id, caption, reply_markup=get_product_keyboard(product['id'])) except Exception as e: logger.error(f"Ошибка при отправке: {e}") await bot.send_message(callback_query.from_user.id, caption, reply_markup=get_product_keyboard(product['id'])) batch_size = 5 for i in range(0, len(products_in_cat), batch_size): batch = products_in_cat[i:i + batch_size] await send_product_batch(batch) await asyncio.sleep(0.1) await bot.answer_callback_query(callback_query.id) except Exception as e: logger.error(f"Ошибка в show_products_in_category: {e}") await bot.answer_callback_query(callback_query.id, "Ошибка при загрузке товаров") @dp.message(F.text == "🛒 Корзина") async def show_cart(message: types.Message): user_id = message.from_user.id cart = next((o for o in data['orders'] if o['user_id'] == user_id and not o.get('completed')), None) if not cart or not cart['items']: await message.answer("Ваша корзина пуста.") return total = 0 response = "Ваша корзина:\n" for item in cart['items']: product = next(p for p in data['products'] if p['id'] == item['product_id']) response += f"🏷 {product['name']} - {product['price']} сом x {item['quantity']}\n" total += product['price'] * item['quantity'] response += f"\nИтого: {total} сом" builder = InlineKeyboardBuilder() builder.button(text="Оформить заказ", callback_data=f"complete_{user_id}") await message.answer(response, reply_markup=builder.as_markup()) @dp.callback_query(F.data.startswith("add_")) async def add_to_cart(callback_query: types.CallbackQuery): try: product_id = int(callback_query.data.split('_')[1]) product = next((p for p in data['products'] if p['id'] == product_id), None) if product: user_id = callback_query.from_user.id cart = next((o for o in data['orders'] if o['user_id'] == user_id and not o.get('completed')), None) if not cart: cart = {'user_id': user_id, 'items': [], 'date': datetime.now().isoformat()} data['orders'].append(cart) cart['items'].append({'product_id': product_id, 'quantity': 1}) save_data(data) await bot.answer_callback_query(callback_query.id, "Товар добавлен в корзину!") except Exception as e: logger.error(f"Ошибка при добавлении в корзину: {e}") await bot.answer_callback_query(callback_query.id, "Ошибка") @dp.callback_query(F.data.startswith("complete_")) async def complete_order(callback_query: types.CallbackQuery): try: user_id = int(callback_query.data.split('_')[1]) cart = next((o for o in data['orders'] if o['user_id'] == user_id and not o.get('completed')), None) if cart and cart['items']: total = 0 cart_text = "Привет, я хочу сделать заказ:\n" for item in cart['items']: product = next((p for p in data['products'] if p['id'] == item['product_id']), None) if product: cart_text += f"{product['name']} - {product['price']} сом x {item['quantity']}\n" total += product['price'] * item['quantity'] cart_text += f"\nИтого: {total} сом" encoded_text = urllib.parse.quote(cart_text) whatsapp_link = f"https://wa.me/996709513331?text={encoded_text}" data['orders'].remove(cart) save_data(data) await bot.send_message(user_id, f"Оформите заказ через WhatsApp:\n{whatsapp_link}") await bot.answer_callback_query(callback_query.id) except Exception as e: logger.error(f"Ошибка при оформлении заказа: {e}") await bot.answer_callback_query(callback_query.id, "Ошибка") @dp.message(F.text == "📦 Заказы") async def show_orders(message: types.Message): user_id = message.from_user.id user_orders = [o for o in data['orders'] if o.get('completed')] if not user_orders: await message.answer("У вас нет оформленных заказов.") return for order in user_orders: response = "Ваш заказ:\n" total = 0 for item in order['items']: product = next(p for p in data['products'] if p['id'] == item['product_id']) response += f"🏷 {product['name']} - {product['price']} сом x {item['quantity']}\n" total += product['price'] * item['quantity'] response += f"\nИтого: {total} сом\nДата: {order['date']}" await message.answer(response) # Админ-панель admin_html = """
Нет категорий.
{% endif %}Нет товаров.
{% endif %}Нет заказов.
{% endif %}