363 lines
14 KiB
Python
363 lines
14 KiB
Python
import asyncio
|
||
from asyncio import Lock
|
||
import asyncpg
|
||
from asyncpg import create_pool
|
||
from aiogram import Bot, Dispatcher, types, F
|
||
from aiogram.filters import Command
|
||
from aiogram.types import ReplyKeyboardMarkup, KeyboardButton, ReplyKeyboardRemove
|
||
from dotenv import load_dotenv
|
||
import uuid
|
||
import hashlib
|
||
import ast
|
||
import os
|
||
import logging
|
||
|
||
logging.basicConfig(level=logging.DEBUG) # DEBUG показывает максимум информации
|
||
|
||
# Загрузка данных из .env
|
||
load_dotenv()
|
||
|
||
db_lock = Lock()
|
||
|
||
db_pool2 = None
|
||
|
||
|
||
# Асинхронная функция для логирования состояния пула
|
||
async def log_pool_state():
|
||
try:
|
||
active_connections = len(db_pool2._holders) # Занятые соединения
|
||
free_connections = db_pool2._queue.qsize() # Свободные соединения
|
||
logging.info(
|
||
f"Пул соединений db_pool2: Активных соединений: {active_connections}, Свободных соединений: {free_connections}"
|
||
)
|
||
except Exception as e:
|
||
logging.error(f"Ошибка при логировании состояния пула db_pool2: {e}")
|
||
|
||
|
||
DATABASE_URL = os.getenv("DATABASE_URL")
|
||
|
||
|
||
# Асинхронная функция для создания пула подключения
|
||
async def get_db_pool2():
|
||
try:
|
||
if not DATABASE_URL:
|
||
raise ValueError("DATABASE_URL не задана.")
|
||
|
||
return await asyncpg.create_pool(DATABASE_URL, max_size=10)
|
||
except Exception as e:
|
||
logging.error(f"Ошибка при подключении к базе данных: {e}")
|
||
raise
|
||
|
||
|
||
BOT_TOKEN = os.getenv("BOT_TOKEN")
|
||
GROUP_ID = os.getenv("GROUP_ID") # Telegram ID администратора
|
||
|
||
# Инициализация бота и диспетчера
|
||
bot = Bot(token=BOT_TOKEN)
|
||
dp = Dispatcher()
|
||
|
||
|
||
# Регистрация пользователя с созданием нового топика
|
||
async def register_user(telegram_id):
|
||
"""
|
||
Регистрирует пользователя, создавая анонимный ID и топик для взаимодействия.
|
||
"""
|
||
try:
|
||
logging.info(f"Регистрация пользователя с Telegram ID: {telegram_id}")
|
||
async with db_pool2.acquire() as conn:
|
||
# Проверяем, зарегистрирован ли пользователь
|
||
result = await conn.fetchrow(
|
||
"SELECT anon_id, topic_id FROM an_users WHERE telegram_id = $1",
|
||
telegram_id,
|
||
)
|
||
if result:
|
||
logging.info(f"Пользователь {telegram_id} уже зарегистрирован.")
|
||
return result["anon_id"], result["topic_id"]
|
||
|
||
# Генерация анонимного ID
|
||
anon_id = str(uuid.uuid4())
|
||
|
||
# Создание нового топика
|
||
topic_title = f"Чат {anon_id[:4]}"
|
||
topic_result = await bot.create_forum_topic(
|
||
chat_id=GROUP_ID, name=topic_title
|
||
)
|
||
topic_id = topic_result.message_thread_id
|
||
|
||
# Сохранение в базе данных
|
||
await conn.execute(
|
||
"INSERT INTO an_users (telegram_id, anon_id, topic_id) VALUES ($1, $2, $3)",
|
||
telegram_id,
|
||
anon_id,
|
||
topic_id,
|
||
)
|
||
logging.info(
|
||
f"Создан новый топик с ID {topic_id} для пользователя {telegram_id}."
|
||
)
|
||
return anon_id, topic_id
|
||
|
||
except Exception as e:
|
||
logging.error(f"Ошибка при регистрации пользователя {telegram_id}: {e}")
|
||
return None, None
|
||
|
||
|
||
# Получение Telegram ID по анонимному ID
|
||
async def get_telegram_id(anon_id):
|
||
async with db_pool2.acquire() as conn:
|
||
result = await conn.fetchrow(
|
||
"SELECT telegram_id FROM an_users WHERE anon_id = $1", anon_id
|
||
)
|
||
return result["telegram_id"] if result else None
|
||
|
||
|
||
@dp.message(Command("start"))
|
||
async def start_command(message: types.Message):
|
||
|
||
anon_id, topic_id = await register_user(message.from_user.id)
|
||
|
||
# Создаем кнопки
|
||
keyboard = ReplyKeyboardMarkup(
|
||
keyboard=[
|
||
[KeyboardButton(text="Написать волонтер_ке")],
|
||
[KeyboardButton(text="Заполнить форму")],
|
||
],
|
||
resize_keyboard=True,
|
||
)
|
||
|
||
await message.answer(
|
||
"Привет 👋! Выбери ниже то, что тебе нужно. ✨\n Если мы не ответили тебе втечение 24 ч, то напиши нам на почту: rloveplus@proton.me или в Matrix Element: @br:bark.lgbt \n А так же желаем хорошего тебе дня 💖",
|
||
reply_markup=keyboard,
|
||
)
|
||
|
||
|
||
# Хэндлер для кнопки "Заполнить форму"
|
||
@dp.message(F.text == "Заполнить форму")
|
||
async def fill_form(message: types.Message):
|
||
form_url = "https://t.me/InstantFormsBot/form?startapp=9dd0908d-3c39-45cb-afe1-7f4004aa8fc6&startApp=9dd0908d-3c39-45cb-afe1-7f4004aa8fc6" # Ссылка на вашу форму
|
||
await message.answer(f"[Заполнить форму]({form_url})", parse_mode="Markdown")
|
||
|
||
|
||
@dp.message(F.text == "Написать волонтер_ке")
|
||
async def contact_volunteer(message: types.Message):
|
||
try:
|
||
|
||
# Получаем номер строки в базе
|
||
async with db_pool2.acquire() as conn:
|
||
result = await conn.fetchrow(
|
||
"SELECT id FROM an_users WHERE telegram_id = $1", message.from_user.id
|
||
)
|
||
|
||
if result:
|
||
user_number = result["id"]
|
||
topic_name = f"{user_number}"
|
||
|
||
await message.answer(
|
||
f"Добро пожаловать! Здесь ты можешь написать нам. Ты будешь общаться с администраторами через бота, поэтому ты останешься для них анонимными.",
|
||
reply_markup=ReplyKeyboardRemove(),
|
||
)
|
||
except Exception as e:
|
||
logging.error(f"Ошибка при создании топика: {e}")
|
||
|
||
|
||
# Обработчик сообщений от пользователя
|
||
@dp.message(F.chat.type == "private")
|
||
async def handle_user_message(message: types.Message):
|
||
logging.info(
|
||
f"Сообщение от {message.from_user.id}: {message.text or 'мультимедиа'}"
|
||
)
|
||
|
||
# Регистрируем пользователя и получаем данные
|
||
anon_id, topic_id = await register_user(message.from_user.id)
|
||
|
||
try:
|
||
# Формируем идентификатор для анонимности
|
||
user_tag = f"Сообщение от {str(anon_id)[:4]}:"
|
||
|
||
# Отправляем разные типы сообщений
|
||
if message.text:
|
||
forward_message = f"{user_tag}\n{message.text}"
|
||
await bot.send_message(
|
||
chat_id=GROUP_ID, message_thread_id=topic_id, text=forward_message
|
||
)
|
||
elif message.photo:
|
||
await bot.send_photo(
|
||
chat_id=GROUP_ID,
|
||
message_thread_id=topic_id,
|
||
photo=message.photo[-1].file_id,
|
||
caption=f"{user_tag}\n{message.caption or ''}",
|
||
)
|
||
elif message.video:
|
||
await bot.send_video(
|
||
chat_id=GROUP_ID,
|
||
message_thread_id=topic_id,
|
||
video=message.video.file_id,
|
||
caption=f"{user_tag}\n{message.caption or ''}",
|
||
)
|
||
elif message.document:
|
||
await bot.send_document(
|
||
chat_id=GROUP_ID,
|
||
message_thread_id=topic_id,
|
||
document=message.document.file_id,
|
||
caption=f"{user_tag}\n{message.caption or ''}",
|
||
)
|
||
elif message.audio:
|
||
await bot.send_audio(
|
||
chat_id=GROUP_ID,
|
||
message_thread_id=topic_id,
|
||
audio=message.audio.file_id,
|
||
caption=f"{user_tag}\n{message.caption or ''}",
|
||
)
|
||
elif message.voice:
|
||
await bot.send_voice(
|
||
chat_id=GROUP_ID,
|
||
message_thread_id=topic_id,
|
||
voice=message.voice.file_id,
|
||
caption=user_tag,
|
||
)
|
||
else:
|
||
await bot.send_message(
|
||
chat_id=GROUP_ID,
|
||
message_thread_id=topic_id,
|
||
text=f"{user_tag}\nТип сообщения пока не поддерживается.",
|
||
)
|
||
|
||
# Уведомляем пользователя
|
||
await message.answer("Сообщение отправлено!")
|
||
logging.info(f"Сообщение от {message.from_user.id} успешно переслано в топик.")
|
||
|
||
except Exception as e:
|
||
logging.error(f"Ошибка при обработке сообщения от пользователя: {e}")
|
||
await message.answer("Произошла ошибка при отправке сообщения.")
|
||
|
||
|
||
# Обработка новых сообщений администратора
|
||
@dp.message(F.chat.type.in_(["group", "supergroup"]) & ~F.text.startswith("/"))
|
||
async def handle_admin_reply(message: types.Message):
|
||
"""
|
||
Обрабатывает только текстовые сообщения от администратора в топиках группы,
|
||
игнорируя команды.
|
||
"""
|
||
if not any(
|
||
[
|
||
message.text,
|
||
message.photo,
|
||
message.video,
|
||
message.document,
|
||
message.audio,
|
||
message.voice,
|
||
]
|
||
):
|
||
logging.info(f"Игнорирование пустого сообщения или команды: {message.text}")
|
||
return
|
||
|
||
# Обрабатываем сообщение
|
||
await process_admin_message(message)
|
||
|
||
|
||
# Обработка редактирования сообщений администратора
|
||
@dp.edited_message(F.chat.type.in_(["group", "supergroup"]))
|
||
async def handle_admin_edited_message(message: types.Message):
|
||
"""
|
||
Обрабатывает редактированные сообщения от администратора в топиках группы.
|
||
"""
|
||
topic_id = message.message_thread_id
|
||
|
||
# Проверяем, существует ли topic_id в базе
|
||
async with db_pool2.acquire() as conn:
|
||
result = await conn.fetchrow(
|
||
"SELECT telegram_id FROM an_users WHERE topic_id = $1", topic_id
|
||
)
|
||
|
||
if not result:
|
||
logging.warning(
|
||
f"Редактирование сообщения в несуществующем топике: {topic_id}. Игнорируем."
|
||
)
|
||
return # Игнорируем редактирование, если пользователь не зарегистрирован
|
||
|
||
# Если пользователь найден, продолжаем обработку
|
||
await process_admin_message(message)
|
||
|
||
|
||
# Общая функция обработки сообщений
|
||
async def process_admin_message(message: types.Message):
|
||
"""
|
||
Обрабатывает как новые, так и редактированные сообщения от администратора.
|
||
"""
|
||
logging.info(
|
||
f"Обработка сообщения от администратора: {message.text}, чат: {message.chat.id}"
|
||
)
|
||
# Получаем topic_id из текущего чата
|
||
topic_id = message.message_thread_id
|
||
|
||
# Находим telegram_id пользователя, связанного с этим topic_id
|
||
async with db_pool2.acquire() as conn:
|
||
result = await conn.fetchrow(
|
||
"SELECT telegram_id FROM an_users WHERE topic_id = $1", topic_id
|
||
)
|
||
|
||
try:
|
||
telegram_id = result["telegram_id"]
|
||
|
||
# Пересылаем сообщение пользователю в зависимости от типа контента
|
||
if message.photo:
|
||
await bot.send_photo(
|
||
chat_id=telegram_id,
|
||
photo=message.photo[-1].file_id,
|
||
caption=message.caption,
|
||
)
|
||
elif message.video:
|
||
await bot.send_video(
|
||
chat_id=telegram_id,
|
||
video=message.video.file_id,
|
||
caption=message.caption,
|
||
)
|
||
elif message.document:
|
||
await bot.send_document(
|
||
chat_id=telegram_id,
|
||
document=message.document.file_id,
|
||
caption=message.caption,
|
||
)
|
||
elif message.audio:
|
||
await bot.send_audio(
|
||
chat_id=telegram_id,
|
||
audio=message.audio.file_id,
|
||
caption=message.caption,
|
||
)
|
||
elif message.voice:
|
||
await bot.send_voice(
|
||
chat_id=telegram_id,
|
||
voice=message.voice.file_id,
|
||
caption=message.caption,
|
||
)
|
||
elif message.text:
|
||
await bot.send_message(
|
||
chat_id=telegram_id, text=f"Ответ волонтёр_ки:\n\n{message.text}"
|
||
)
|
||
|
||
logging.info(f"Ответ успешно отправлен пользователю с ID {telegram_id}")
|
||
|
||
except Exception as e:
|
||
logging.error(f"Ошибка при обработке ответа администратора: {e}")
|
||
await message.reply("Произошла ошибка при отправке ответа пользователю.")
|
||
|
||
|
||
# Инициализация пула и запуск бота
|
||
async def main():
|
||
global db_pool2
|
||
try:
|
||
db_pool2 = await get_db_pool2() # Создание пула
|
||
logging.info("Пул db_pool2 успешно создан")
|
||
|
||
# Тестовый запрос
|
||
async with db_pool2.acquire() as conn:
|
||
await conn.execute("SELECT 1")
|
||
logging.info("Подключение к базе данных успешно установлено")
|
||
|
||
await log_pool_state() # Логирование состояния пула
|
||
await dp.start_polling(bot)
|
||
except Exception as e:
|
||
logging.error(f"Ошибка при подключении к базе данных: {e}")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(main()) |