Закачка аватаров канала (в разработке)
This commit is contained in:
63
Test/commands/download_channel_avatar.py
Normal file
63
Test/commands/download_channel_avatar.py
Normal file
@@ -0,0 +1,63 @@
|
||||
# BotCode/routers/downloads/download_channel_avatar.py
|
||||
# Закачка аватарок канала (в разработке + сделать логирование!!!)
|
||||
|
||||
import requests
|
||||
from aiogram import Router
|
||||
from aiogram.types import Message
|
||||
from BotLibrary import *
|
||||
|
||||
# Создание роутера и настройка экспорта
|
||||
__all__ = ("router", "download_channel_avatar",)
|
||||
router = Router(name="avatar_channel_router")
|
||||
type_messages = "AvatarChannel"
|
||||
|
||||
|
||||
# Функция для скачивания аватарки канала
|
||||
@router.message(lambda message: message.chat.type == 'channel')
|
||||
async def download_channel_avatar(message: Message):
|
||||
try:
|
||||
# Логирование для получения сообщения из канала
|
||||
logger.bind(custom_variable=type_messages, user_var=f"{message.chat.id}").info(f"Получено сообщение из канала ID {message.chat.id}: {message.text}")
|
||||
|
||||
# Получаем информацию о чате канала
|
||||
chat_info = await bot.get_chat(message.chat.id) # message.chat.id для получения ID канала
|
||||
|
||||
# Логирование полученной информации
|
||||
logger.bind(custom_variable=type_messages, user_var=f"{chat_info.id}").info(f"Информация о канале: {chat_info.title}")
|
||||
|
||||
# Проверка наличия аватара
|
||||
if not chat_info.photo:
|
||||
text_error = f"Канал с ID {message.chat.id} не имеет аватара."
|
||||
logger.error(text_error)
|
||||
return text_error
|
||||
|
||||
# Получаем file_id для фото (используем big_file_id для лучшего качества)
|
||||
file_info = await bot.get_file(chat_info.photo.big_file_id)
|
||||
|
||||
# Строим URL для скачивания файла
|
||||
file_url = f"https://api.telegram.org/file/bot{bot.token}/{file_info.file_path}"
|
||||
|
||||
# Формируем имя и путь для сохранения фото
|
||||
save_dir = f"{ImportantPath.channel_avatar}/{message.chat.id}"
|
||||
os.makedirs(save_dir, exist_ok=True)
|
||||
file_name = file_info.file_path.split("/")[-1] # Имя файла (например, "abc.jpg")
|
||||
save_path = os.path.join(save_dir, file_name)
|
||||
|
||||
# Скачиваем аватар
|
||||
response = requests.get(file_url)
|
||||
if response.status_code == 200:
|
||||
with open(save_path, "wb") as file:
|
||||
file.write(response.content)
|
||||
text = f"Фото аватара канала с ID {message.chat.id} успешно скачано и сохранено как {save_path}"
|
||||
logger.info(text)
|
||||
return text
|
||||
|
||||
else:
|
||||
text_error = f"Не удалось скачать аватар канала с ID {message.chat.id}. Статус: {response.status_code}"
|
||||
logger.bind(custom_variable=type_messages, user_var=f"{message.chat.id}").error(text_error)
|
||||
return text_error
|
||||
|
||||
except Exception as e:
|
||||
text_error = f"Ошибка при скачивании фото аватара канала с ID {message.chat.id}: {e}"
|
||||
logger.bind(custom_variable=type_messages, user_var=f"{message.chat.id}").error(text_error)
|
||||
return text_error
|
||||
Reference in New Issue
Block a user