Вторая часть: Небольшие дополнения и структура
This commit is contained in:
6
BotLibrary/system/__init__.py
Normal file
6
BotLibrary/system/__init__.py
Normal file
@@ -0,0 +1,6 @@
|
||||
# BotLibrary/system/__init__.py
|
||||
# Инициализация пакета system, для библиотек запуска
|
||||
|
||||
# Экспортирование модулей во внешние слои проекта
|
||||
from .bots import *
|
||||
from .bot_edit import *
|
||||
90
BotLibrary/system/bot_edit.py
Normal file
90
BotLibrary/system/bot_edit.py
Normal file
@@ -0,0 +1,90 @@
|
||||
# BotLibrary/system/edit_bot.py
|
||||
# Библиотека установки настроек бота через проект и конфиги
|
||||
|
||||
from aiogram.types import ChatAdministratorRights
|
||||
from ProjectsFiles import BotEdit
|
||||
from .bots import bot
|
||||
from ..loggers import logger
|
||||
|
||||
# Настройка логирования
|
||||
log_type = "Edit"
|
||||
|
||||
# Функция для выполнения всех настроек, если они не совпадают
|
||||
async def set_all():
|
||||
await set_adm_rights()
|
||||
await set_bot_name()
|
||||
await set_bot_description()
|
||||
await set_bot_short_description()
|
||||
|
||||
|
||||
# Функция установки прав администратора
|
||||
async def set_adm_rights():
|
||||
# Применить права администратора для бота
|
||||
rights = ChatAdministratorRights(
|
||||
is_anonymous=BotEdit.is_anonymous,
|
||||
can_manage_chat=BotEdit.manage_chat,
|
||||
can_delete_messages=BotEdit.delete_messages,
|
||||
can_manage_video_chats=BotEdit.manage_video_chats,
|
||||
can_restrict_members=BotEdit.restrict_members,
|
||||
can_promote_members=BotEdit.promote_members,
|
||||
can_change_info=BotEdit.change_info,
|
||||
can_invite_users=BotEdit.invite_users,
|
||||
can_post_stories=BotEdit.post_stories,
|
||||
can_edit_stories=BotEdit.edit_stories,
|
||||
can_delete_stories=BotEdit.delete_stories,
|
||||
can_post_messages=BotEdit.post_messages,
|
||||
can_edit_messages=BotEdit.edit_messages,
|
||||
can_pin_messages=BotEdit.pin_messages,
|
||||
can_manage_topics=BotEdit.manage_topics,
|
||||
)
|
||||
|
||||
# Применяем права только в случае изменения
|
||||
current_rights = await bot.get_my_default_administrator_rights()
|
||||
if current_rights != rights:
|
||||
await bot.set_my_default_administrator_rights(rights)
|
||||
|
||||
|
||||
# Функция установки имени бота с проверкой на ограничения
|
||||
async def set_bot_name():
|
||||
# Получаем текущее имя бота
|
||||
current_name = (await bot.get_me()).first_name
|
||||
|
||||
# Проверка длины имени
|
||||
if len(BotEdit.name) < 1 or len(BotEdit.name) > 32:
|
||||
# Логируем ошибку, если имя не соответствует ограничению
|
||||
(logger.bind(log_type=log_type, user="NAME_BOT")
|
||||
.error("Имя бота должно быть от 1 до 32 символов."))
|
||||
|
||||
# Проверяем, совпадает ли текущее имя с тем, которое мы хотим установить
|
||||
if current_name != BotEdit.name:
|
||||
await bot.set_my_name(BotEdit.name)
|
||||
|
||||
|
||||
# Функция установки описания бота с проверкой на ограничения
|
||||
async def set_bot_description():
|
||||
# Получаем текущее описание бота
|
||||
current_description = await bot.get_my_description()
|
||||
|
||||
# Проверка длины описания
|
||||
if len(BotEdit.description) > 255:
|
||||
(logger.bind(log_type=log_type, user="DISCRIPT")
|
||||
.error("Короткое описание бота не может превышать 255 символов."))
|
||||
|
||||
# Проверяем, совпадает ли текущее описание с тем, которое мы хотим установить
|
||||
if current_description != BotEdit.description:
|
||||
await bot.set_my_description(description=BotEdit.description)
|
||||
|
||||
|
||||
# Функция установки короткого описания бота с проверкой на ограничения
|
||||
async def set_bot_short_description():
|
||||
# Получаем текущее короткое описание бота
|
||||
current_short_description = await bot.get_my_short_description()
|
||||
|
||||
# Проверка длины короткого описания
|
||||
if len(BotEdit.short_description) > 512:
|
||||
(logger.bind(log_type=log_type, user="SHORT_DISCRIPT")
|
||||
.error("Описание виджета не может превышать 512 символов."))
|
||||
|
||||
# Проверяем, совпадает ли текущее короткое описание с тем, которое мы хотим установить
|
||||
if current_short_description != BotEdit.short_description:
|
||||
await bot.set_my_short_description(short_description=BotEdit.short_description)
|
||||
113
BotLibrary/system/bots.py
Normal file
113
BotLibrary/system/bots.py
Normal file
@@ -0,0 +1,113 @@
|
||||
# BotLibrary/system/bots.py
|
||||
# Создание и настройка бота в одном файле
|
||||
|
||||
from aiogram import Dispatcher, Bot, F
|
||||
from aiogram.enums import ParseMode
|
||||
from aiogram.client.default import DefaultBotProperties
|
||||
from aiogram.utils.keyboard import InlineKeyboardBuilder, ReplyKeyboardBuilder
|
||||
|
||||
from ..timer import *
|
||||
from ProjectsFiles import *
|
||||
|
||||
# Создание экземпляра диспатчера, строителей кнопок
|
||||
dp = Dispatcher()
|
||||
rkb = ReplyKeyboardBuilder()
|
||||
ikb = InlineKeyboardBuilder()
|
||||
|
||||
|
||||
# Настройка параметров диспатчера
|
||||
dp["started_at"] = get_host_time()
|
||||
dp["started_at_msk"] = get_moscow_time()
|
||||
dp["is_active"] = True # Флаг активности бота
|
||||
dp["logs"] = []
|
||||
dp["users"] = {}
|
||||
dp["sessions"] = {}
|
||||
dp["task_queue"] = []
|
||||
dp["configs"] = {"max_connections": 100, "retry_interval": 5, "time_format": BotVar.time_format}
|
||||
dp["metrics"] = {"messages_received": 0, "messages_sent": 0, "errors": 0}
|
||||
dp["modules"] = {}
|
||||
dp["state"] = {}
|
||||
dp["scheduler"] = []
|
||||
dp["handlers"] = {"on_message": [], "on_error": []}
|
||||
dp["storage"] = {}
|
||||
dp["database"] = None
|
||||
|
||||
|
||||
# Объявление экземпляров и переменных
|
||||
bot_properties = DefaultBotProperties(
|
||||
parse_mode=ParseMode.HTML, # Устанавливаем формат HTML для всех сообщений
|
||||
disable_notification=True, # Отключаем уведомления при отправке сообщений
|
||||
protect_content=True, # Защищаем содержимое сообщений от копирования
|
||||
allow_sending_without_reply=True, # Разрешаем отправлять сообщения без ответа на другое сообщение
|
||||
link_preview_is_disabled=True, # Отключаем предварительный просмотр ссылок
|
||||
show_caption_above_media=False, # Показываем подпись выше медиа
|
||||
)
|
||||
bot = Bot(token=bot_token, default=bot_properties) # Объявление бота
|
||||
F_Media = F.photo | F.files | F.video | F.animation | F.voice | F.video_note # Фильтр-медиа
|
||||
F_All = F.text | F.photo | F.files | F.video | F.animation | F.voice | F.video_note # Фильтр на все
|
||||
|
||||
|
||||
# Класс для хранения данных о боте (некоторые переменные даны как шаблон)
|
||||
class BotInfo:
|
||||
# Статические переменные для хранения данных
|
||||
id = None
|
||||
first_name = None
|
||||
last_name = None
|
||||
username = None
|
||||
description = None
|
||||
short_description = None
|
||||
can_join_groups = None
|
||||
can_read_all_group_messages = None
|
||||
language_code = BotVar.language
|
||||
prefix = BotVar.prefix
|
||||
is_premium = None
|
||||
added_to_attachment_menu = None
|
||||
supports_inline_queries = None
|
||||
can_connect_to_business = None
|
||||
has_main_web_app = None
|
||||
|
||||
# Метод для обновления данных
|
||||
@classmethod
|
||||
def update(cls, bot_info):
|
||||
cls.id = bot_info.id
|
||||
cls.first_name = bot_info.first_name
|
||||
cls.last_name = bot_info.last_name
|
||||
cls.username = bot_info.username
|
||||
cls.description = getattr(bot_info, 'description', '')
|
||||
cls.short_description = getattr(bot_info, 'description', '')
|
||||
cls.language_code = bot_info.language_code
|
||||
cls.is_premium = bot_info.is_premium
|
||||
cls.added_to_attachment_menu = bot_info.added_to_attachment_menu
|
||||
cls.supports_inline_queries = bot_info.supports_inline_queries
|
||||
cls.can_connect_to_business = bot_info.can_connect_to_business
|
||||
cls.has_main_web_app = bot_info.has_main_web_app
|
||||
cls.can_join_groups = getattr(bot_info, 'can_join_groups', None)
|
||||
cls.can_read_all_group_messages = getattr(bot_info, 'can_read_all_group_messages', None)
|
||||
|
||||
|
||||
# Функция получения данных о боте
|
||||
async def bot_get_info():
|
||||
# Получение информации о боте
|
||||
bot_info_data = await bot.get_me()
|
||||
|
||||
# Обновляем данные о боте в BotInfo
|
||||
BotInfo.update(bot_info_data)
|
||||
|
||||
# Возвращаем обновленные данные
|
||||
return {
|
||||
'bot_info': bot_info_data,
|
||||
'id': bot_info_data.id,
|
||||
'first_name': bot_info_data.first_name,
|
||||
'last_name': bot_info_data.last_name,
|
||||
'username': bot_info_data.username,
|
||||
'description': getattr(bot_info_data, 'description', ''),
|
||||
'short_description': getattr(bot_info_data, 'description', ''),
|
||||
'language_code': bot_info_data.language_code,
|
||||
'prefix': BotVar.prefix,
|
||||
'is_premium': bot_info_data.is_premium,
|
||||
'added_to_attachment_menu': bot_info_data.added_to_attachment_menu,
|
||||
'supports_inline_queries': bot_info_data.supports_inline_queries,
|
||||
'can_connect_to_business': bot_info_data.can_connect_to_business,
|
||||
'has_main_web_app': bot_info_data.has_main_web_app,
|
||||
'can_join_groups': getattr(bot_info_data, 'can_join_groups', None),
|
||||
}
|
||||
Reference in New Issue
Block a user