314 lines
11 KiB
Python
314 lines
11 KiB
Python
import os
|
||
import asyncio
|
||
from functools import wraps
|
||
from dotenv import load_dotenv
|
||
from aiogram import Bot, Dispatcher, types
|
||
from aiogram.types import Message, InlineKeyboardMarkup, InlineKeyboardButton, WebAppInfo
|
||
from aiogram.filters import Command
|
||
from aiohttp import web
|
||
import database as db
|
||
from phone_utils import normalize_phone
|
||
from qr_utils import decode_qr_from_image, generate_qr_code
|
||
|
||
# Загружаем переменные из .env
|
||
load_dotenv()
|
||
|
||
# ========== КОНФИГУРАЦИЯ ИЗ .ENV ==========
|
||
BOT_TOKEN = os.getenv("BOT_TOKEN")
|
||
WEBAPP_HOST = os.getenv("WEBAPP_HOST", "0.0.0.0")
|
||
WEBAPP_PORT = int(os.getenv("WEBAPP_PORT", "8085"))
|
||
ADMIN_ID = int(os.getenv("ADMIN_ID"))
|
||
DOMAIN = os.getenv("DOMAIN")
|
||
|
||
# Проверяем, что все переменные загружены
|
||
if not BOT_TOKEN:
|
||
raise ValueError("BOT_TOKEN не найден в .env файле")
|
||
if not ADMIN_ID:
|
||
raise ValueError("ADMIN_ID не найден в .env файле")
|
||
if not DOMAIN:
|
||
raise ValueError("DOMAIN не найден в .env файле")
|
||
|
||
bot = Bot(token=BOT_TOKEN)
|
||
dp = Dispatcher()
|
||
|
||
|
||
# ========== ДЕКОРАТОР ПРОВЕРКИ ДОСТУПА ==========
|
||
def allowed_only(func):
|
||
@wraps(func)
|
||
async def wrapper(message: Message, *args, **kwargs):
|
||
if not db.is_user_allowed(message.from_user.id) and message.from_user.id != ADMIN_ID:
|
||
await message.answer("⛔ Доступ запрещён, брат. Обратись к админу.")
|
||
return
|
||
return await func(message, *args, **kwargs)
|
||
|
||
return wrapper
|
||
|
||
|
||
# ========== ОСНОВНЫЕ КОМАНДЫ ==========
|
||
@dp.message(Command("start"))
|
||
async def cmd_start(message: Message):
|
||
user_id = message.from_user.id
|
||
if not db.is_user_allowed(user_id) and user_id != ADMIN_ID:
|
||
await message.answer("⛔ Тебя нет в белом списке.")
|
||
return
|
||
|
||
keyboard = InlineKeyboardMarkup(inline_keyboard=[
|
||
[InlineKeyboardButton(text="📋 Панель управления", web_app=WebAppInfo(url=f"https://{DOMAIN}/get_webapp"))]
|
||
])
|
||
await message.answer(
|
||
"🌑 *МракоЗвон*\n\n"
|
||
"📥 Отправь номер телефона — сохраню.\n"
|
||
"🃏 Отправь QR-код бонусной карты — привяжу.\n"
|
||
"📎 Отправь номер + QR вместе — свяжу их.\n\n"
|
||
"🎲 /get — случайный номер из базы\n"
|
||
"📊 /stats — статистика\n"
|
||
"🗑 /del <номер> — удалить номер\n\n"
|
||
"🌐 Кнопка ниже — веб-морда",
|
||
reply_markup=keyboard,
|
||
parse_mode="Markdown"
|
||
)
|
||
|
||
|
||
@dp.message(Command("get"))
|
||
@allowed_only
|
||
async def cmd_get(message: Message):
|
||
record = db.get_random_record()
|
||
if not record:
|
||
await message.answer("❌ База пуста, брат!")
|
||
return
|
||
|
||
record_type = record['type']
|
||
phone = record.get('phone')
|
||
qr_code = record.get('qr_code')
|
||
|
||
if record_type == 'both' and qr_code:
|
||
await message.answer_photo(
|
||
photo=qr_code,
|
||
caption=f"🎲 *МракоЗвон*\n📞 Номер: `{phone}`\n🃏 Бонусная карта привязана\n🔄 Использован: {record['usage_count']} раз",
|
||
parse_mode="Markdown"
|
||
)
|
||
elif record_type == 'phone':
|
||
await message.answer(
|
||
f"🎲 *МракоЗвон*\n📞 Номер: `{phone}`\n🔄 Использован: {record['usage_count']} раз",
|
||
parse_mode="Markdown"
|
||
)
|
||
elif record_type == 'qr' and qr_code:
|
||
await message.answer_photo(
|
||
photo=qr_code,
|
||
caption=f"🎲 *МракоЗвон*\n🃏 Бонусная карта\n🔄 Использована: {record['usage_count']} раз",
|
||
parse_mode="Markdown"
|
||
)
|
||
|
||
|
||
@dp.message(Command("stats"))
|
||
@allowed_only
|
||
async def cmd_stats(message: Message):
|
||
all_nums = db.get_all_numbers()
|
||
total = len(all_nums)
|
||
used_total = sum(n["usage_count"] for n in all_nums)
|
||
with_card = sum(1 for n in all_nums if n.get("bonus_card"))
|
||
await message.answer(
|
||
f"📊 *Статистика МракоЗвона*\n\n"
|
||
f"📞 Всего номеров: {total}\n"
|
||
f"🃏 Из них с бонусными картами: {with_card}\n"
|
||
f"🔄 Всего использований: {used_total}",
|
||
parse_mode="Markdown"
|
||
)
|
||
|
||
|
||
@dp.message(Command("del"))
|
||
@allowed_only
|
||
async def cmd_delete_number(message: Message):
|
||
args = message.text.split()
|
||
if len(args) < 2:
|
||
await message.answer("ℹ️ Использование: /del <номер>\nПример: /del 89231234567")
|
||
return
|
||
|
||
normalized = normalize_phone(args[1])
|
||
if not normalized:
|
||
await message.answer("⚠️ Некорректный формат номера")
|
||
return
|
||
|
||
all_nums = db.get_all_numbers()
|
||
for num in all_nums:
|
||
if num["phone"] == normalized:
|
||
db.delete_number(num["id"])
|
||
await message.answer(f"✅ Номер {normalized} удалён")
|
||
return
|
||
|
||
await message.answer(f"❌ Номер {normalized} не найден")
|
||
|
||
|
||
# ========== АДМИН-КОМАНДЫ ==========
|
||
@dp.message(Command("adduser"))
|
||
async def cmd_add_user(message: Message):
|
||
if message.from_user.id != ADMIN_ID:
|
||
await message.answer("⛔ Только админ.")
|
||
return
|
||
|
||
args = message.text.split()
|
||
if len(args) < 2:
|
||
await message.answer("ℹ️ /adduser <user_id> [username]\nПример: /adduser 123456789 брат")
|
||
return
|
||
|
||
try:
|
||
user_id = int(args[1])
|
||
username = args[2] if len(args) > 2 else None
|
||
|
||
if db.add_allowed_user(user_id, username, str(message.from_user.id)):
|
||
await message.answer(f"✅ Пользователь {user_id} добавлен в белый список!")
|
||
else:
|
||
await message.answer(f"⚠️ Пользователь {user_id} уже есть в списке.")
|
||
except ValueError:
|
||
await message.answer("❌ user_id должен быть числом.")
|
||
|
||
|
||
@dp.message(Command("removeuser"))
|
||
async def cmd_remove_user(message: Message):
|
||
if message.from_user.id != ADMIN_ID:
|
||
await message.answer("⛔ Только админ.")
|
||
return
|
||
|
||
args = message.text.split()
|
||
if len(args) < 2:
|
||
await message.answer("ℹ️ /removeuser <user_id>")
|
||
return
|
||
|
||
try:
|
||
user_id = int(args[1])
|
||
if user_id == ADMIN_ID:
|
||
await message.answer("❌ Нельзя удалить админа")
|
||
return
|
||
|
||
if db.remove_allowed_user(user_id):
|
||
await message.answer(f"✅ Пользователь {user_id} удалён из белого списка")
|
||
else:
|
||
await message.answer("⚠️ Пользователь не найден")
|
||
except ValueError:
|
||
await message.answer("❌ user_id должен быть числом.")
|
||
|
||
|
||
@dp.message(Command("whitelist"))
|
||
async def cmd_whitelist(message: Message):
|
||
if message.from_user.id != ADMIN_ID:
|
||
await message.answer("⛔ Только админ.")
|
||
return
|
||
|
||
users = db.get_allowed_users()
|
||
if not users:
|
||
await message.answer("📭 Белый список пуст")
|
||
return
|
||
|
||
text = "📋 *Белый список пользователей:*\n\n"
|
||
for u in users:
|
||
text += f"• ID: `{u['user_id']}`"
|
||
if u['username']:
|
||
text += f" (@{u['username']})"
|
||
text += f"\n Добавлен: {u['added_at'][:19]}\n\n"
|
||
|
||
if len(text) > 4000:
|
||
await message.answer("📋 Слишком много записей, смотри в БД.")
|
||
else:
|
||
await message.answer(text, parse_mode="Markdown")
|
||
|
||
|
||
# ========== ОБРАБОТЧИК СООБЩЕНИЙ (НОМЕРА + QR) ==========
|
||
@dp.message()
|
||
@allowed_only
|
||
async def handle_message(message: Message):
|
||
phone = None
|
||
qr_raw_data = None
|
||
qr_image = None
|
||
|
||
# 1. Распознаём QR из фото
|
||
if message.photo:
|
||
photo = message.photo[-1]
|
||
file = await bot.get_file(photo.file_id)
|
||
file_bytes = await bot.download_file(file.file_path)
|
||
qr_raw_data = decode_qr_from_image(file_bytes.read())
|
||
if qr_raw_data:
|
||
qr_image = generate_qr_code(qr_raw_data)
|
||
|
||
# 2. Извлекаем номер из текста
|
||
if message.text:
|
||
phone = normalize_phone(message.text.strip())
|
||
|
||
# 3. Определяем тип записи и сохраняем
|
||
if phone and qr_raw_data:
|
||
# Тип: номер + QR
|
||
db.add_record('both', phone=phone, qr_code=qr_image, qr_data=qr_raw_data)
|
||
await message.answer(f"✅ Сохранено: номер `{phone}` + бонусная карта", parse_mode="Markdown")
|
||
elif phone:
|
||
# Тип: только номер
|
||
db.add_record('phone', phone=phone)
|
||
await message.answer(f"✅ Сохранён номер: `{phone}`", parse_mode="Markdown")
|
||
elif qr_raw_data:
|
||
# Тип: только QR
|
||
db.add_record('qr', qr_code=qr_image, qr_data=qr_raw_data)
|
||
await message.answer(f"🃏 Сохранена бонусная карта: `{qr_raw_data[:50]}...`", parse_mode="Markdown")
|
||
else:
|
||
await message.answer("⚠️ Не распознано. Отправь номер, QR-код или оба вместе.")
|
||
|
||
|
||
# ========== ВЕБ-СЕРВЕР ==========
|
||
routes = web.RouteTableDef()
|
||
|
||
|
||
@routes.get("/get_records")
|
||
async def get_records(request):
|
||
records = db.get_all_records()
|
||
return web.json_response({"records": records})
|
||
|
||
@routes.get("/get_qr_code/{record_id}")
|
||
async def get_qr_code(request):
|
||
record_id = int(request.match_info.get('record_id', 0))
|
||
with sqlite3.connect(DB_NAME) as conn:
|
||
conn.row_factory = sqlite3.Row
|
||
result = conn.execute("SELECT qr_code FROM records WHERE id = ?", (record_id,)).fetchone()
|
||
if result and result['qr_code']:
|
||
return web.json_response({"qr_code": result['qr_code']})
|
||
return web.json_response({"qr_code": None})
|
||
|
||
@routes.post("/delete_record")
|
||
async def delete_record(request):
|
||
data = await request.json()
|
||
db.delete_record(data["id"])
|
||
return web.json_response({"status": "ok"})
|
||
|
||
|
||
@routes.get("/get_bonus_card/{record_id}")
|
||
async def get_bonus_card(request):
|
||
record_id = int(request.match_info.get('record_id', 0))
|
||
bonus_card = db.get_bonus_card(record_id)
|
||
|
||
if bonus_card:
|
||
return web.json_response({"bonus_card": bonus_card})
|
||
|
||
return web.json_response({"bonus_card": None})
|
||
|
||
|
||
async def start_webapp():
|
||
app = web.Application()
|
||
app.add_routes(routes)
|
||
runner = web.AppRunner(app)
|
||
await runner.setup()
|
||
site = web.TCPSite(runner, WEBAPP_HOST, WEBAPP_PORT)
|
||
await site.start()
|
||
print(f"✅ Веб-морда запущена на порту {WEBAPP_PORT}")
|
||
|
||
|
||
# ========== ЗАПУСК ==========
|
||
async def main():
|
||
db.init_db()
|
||
db.add_allowed_user(ADMIN_ID, "admin", "system")
|
||
|
||
asyncio.create_task(start_webapp())
|
||
|
||
print(f"🚀 МракоЗвон запущен!")
|
||
print(f"📱 Домен: https://{DOMAIN}")
|
||
print(f"🔌 Порт: {WEBAPP_PORT}")
|
||
await dp.start_polling(bot)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(main()) |