I add command /admin /vadmin /iadmin /hello I create database and I improve code
It's version 0.3.0
This commit is contained in:
@@ -0,0 +1,7 @@
|
||||
def register(dp, state, bot):
|
||||
from . import handlers
|
||||
handlers.register_handlers(dp, state, bot)
|
||||
|
||||
def unregister(dp):
|
||||
# Здесь можно удалить хендлеры, если нужно
|
||||
dp.message_handlers.handlers.clear()
|
||||
@@ -0,0 +1,133 @@
|
||||
from config import Config
|
||||
import aiohttp
|
||||
from aiogram.types import BufferedInputFile
|
||||
from utils.antispam import admin_required
|
||||
from aiogram import Dispatcher, Bot
|
||||
from aiogram.types import Message
|
||||
from models.state import BotState
|
||||
from aiogram.filters import Command
|
||||
from logging import getLogger
|
||||
|
||||
logger = getLogger(__name__)
|
||||
API_URL = "http://127.0.0.1:7700/speak"
|
||||
|
||||
def register_handlers(dp: Dispatcher, state: BotState, bot: Bot):
|
||||
@dp.message(Command("vadmin"))
|
||||
@admin_required(0)
|
||||
async def vadmin(message: Message):
|
||||
parts = message.text.split(maxsplit=1)
|
||||
if len(parts) < 2:
|
||||
await message.reply("❗ Укажи текст после /vadmin")
|
||||
return
|
||||
phrase = parts[1]
|
||||
|
||||
# Запрос к TTS API
|
||||
async with aiohttp.ClientSession() as session:
|
||||
async with session.post(API_URL, json={"text": phrase}) as resp:
|
||||
if resp.status != 200:
|
||||
await message.reply("Ошибка генерации аудио")
|
||||
return
|
||||
audio_bytes = await resp.read()
|
||||
audio_file = BufferedInputFile(audio_bytes, filename="speech.wav")
|
||||
|
||||
# Рассылка в чаты
|
||||
for chat_id in Config.CHAT_IDS:
|
||||
try:
|
||||
await bot.send_audio(chat_id, audio=audio_file, caption="🎙 Текст")
|
||||
except Exception as e:
|
||||
await message.reply(f"Не удалось отправить в {chat_id}: {e}")
|
||||
|
||||
await message.reply("✅ Озвучка разослана.")
|
||||
|
||||
@dp.message(Command("admin"))
|
||||
@admin_required(0)
|
||||
async def admin(message: Message):
|
||||
raw_text = message.text or message.caption
|
||||
if not raw_text and not (message.photo or message.document or message.audio or message.video):
|
||||
await message.reply("❌ Укажи текст или прикрепи файл/медиа: /admin <сообщение>")
|
||||
return
|
||||
|
||||
# Отрезаем саму команду (/admin)
|
||||
args = raw_text.split(maxsplit=1) if raw_text else []
|
||||
text_to_send = args[1] if len(args) > 1 else ""
|
||||
|
||||
for chat_id in Config.CHAT_IDS:
|
||||
try:
|
||||
if message.photo:
|
||||
# Фото
|
||||
photo = message.photo[-1].file_id
|
||||
await bot.send_photo(chat_id, photo, caption=text_to_send)
|
||||
|
||||
elif message.document:
|
||||
# Документ
|
||||
await bot.send_document(chat_id, message.document.file_id, caption=text_to_send)
|
||||
|
||||
elif message.audio:
|
||||
# Аудио (музыка)
|
||||
await bot.send_audio(chat_id, message.audio.file_id, caption=text_to_send)
|
||||
|
||||
elif message.video:
|
||||
# Видео
|
||||
await bot.send_video(chat_id, message.video.file_id, caption=text_to_send)
|
||||
|
||||
else:
|
||||
# Только текст
|
||||
await bot.send_message(chat_id, text_to_send)
|
||||
|
||||
logger.info(f"Сообщение отправлено в чат {chat_id}")
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка при отправке в чат {chat_id}: {e}")
|
||||
|
||||
await message.answer("✅ Сообщение отправлено.")
|
||||
|
||||
@dp.message(Command("iadmin"))
|
||||
@admin_required(0)
|
||||
async def id_admin(message: Message):
|
||||
raw_text = message.text or message.caption
|
||||
if not raw_text and not (message.photo or message.document or message.audio or message.video):
|
||||
await message.reply("❌ Укажи ID чата и текст или прикрепи файл/медиа: /iadmin <chat_id> <сообщение>")
|
||||
return
|
||||
|
||||
# Отрезаем саму команду (/iadmin)
|
||||
args = raw_text.split(maxsplit=2) if raw_text else []
|
||||
if len(args) < 2:
|
||||
await message.reply("❌ Укажи ID чата: /iadmin <chat_id> <сообщение>")
|
||||
return
|
||||
|
||||
try:
|
||||
chat_id = int(args[1]) # первый аргумент — ID чата
|
||||
except ValueError:
|
||||
await message.reply("❌ Неверный формат chat_id")
|
||||
return
|
||||
|
||||
text_to_send = args[2] if len(args) > 2 else ""
|
||||
|
||||
try:
|
||||
if message.photo:
|
||||
# Фото
|
||||
photo = message.photo[-1].file_id
|
||||
await bot.send_photo(chat_id, photo, caption=text_to_send)
|
||||
|
||||
elif message.document:
|
||||
# Документ
|
||||
await bot.send_document(chat_id, message.document.file_id, caption=text_to_send)
|
||||
|
||||
elif message.audio:
|
||||
# Аудио (музыка)
|
||||
await bot.send_audio(chat_id, message.audio.file_id, caption=text_to_send)
|
||||
|
||||
elif message.video:
|
||||
# Видео
|
||||
await bot.send_video(chat_id, message.video.file_id, caption=text_to_send)
|
||||
|
||||
else:
|
||||
# Только текст
|
||||
await bot.send_message(chat_id, text_to_send)
|
||||
|
||||
logger.info(f"Сообщение отправлено в чат {chat_id}")
|
||||
await message.answer("✅ Сообщение отправлено.")
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка при отправке в чат {chat_id}: {e}")
|
||||
await message.answer(f"❌ Ошибка при отправке: {e}")
|
||||
Reference in New Issue
Block a user