Вторая часть: Небольшие дополнения и структура
This commit is contained in:
@@ -1,6 +1,9 @@
|
||||
# ProjectsFiles/__init__.py
|
||||
# Инициализация пакета ProjectsFiles, для настройки переменных проекта
|
||||
# BotLibrary/__init__.py
|
||||
# Инициализация пакета BotLibrary, для настройки личных библиотек проекта
|
||||
|
||||
# Экспортирование модулей во внешние слои проекта
|
||||
from .bots import *
|
||||
from .logs import *
|
||||
from .analytics import *
|
||||
from .loggers import *
|
||||
from .system import *
|
||||
from .timer import *
|
||||
from .validators import *
|
||||
|
||||
5
BotLibrary/analytics/__init__.py
Normal file
5
BotLibrary/analytics/__init__.py
Normal file
@@ -0,0 +1,5 @@
|
||||
# BotLibrary/analytics/__init__.py
|
||||
# Инициализация пакета analytics, для аналитики и проверки типов
|
||||
|
||||
# Экспортирование модулей во внешние слои проекта
|
||||
from .type_msg import *
|
||||
63
BotLibrary/analytics/type_msg.py
Normal file
63
BotLibrary/analytics/type_msg.py
Normal file
@@ -0,0 +1,63 @@
|
||||
# BotLibrary/analytics/log_type.py
|
||||
# Определение типа сообщения
|
||||
|
||||
from aiogram.types import ContentType
|
||||
|
||||
def types_message(message):
|
||||
# Словарь для соответствия типов сообщений
|
||||
content_types = {
|
||||
ContentType.TEXT: "Текст",
|
||||
ContentType.PHOTO: "Фото",
|
||||
ContentType.STICKER: "Стикер",
|
||||
ContentType.ANIMATION: "Гиф",
|
||||
ContentType.VOICE: "Голосовое сообщение",
|
||||
ContentType.VIDEO_NOTE: "Видео-сообщение",
|
||||
ContentType.VIDEO: "Видео",
|
||||
ContentType.AUDIO: "Аудио",
|
||||
ContentType.DOCUMENT: "Документ",
|
||||
ContentType.CONTACT: "Контакт",
|
||||
ContentType.LOCATION: "Локация",
|
||||
ContentType.VENUE: "Место",
|
||||
ContentType.DICE: "Бросок кубика",
|
||||
ContentType.STORY: "История",
|
||||
ContentType.GAME: "Игра",
|
||||
ContentType.POLL: "Опрос",
|
||||
ContentType.FORUM_TOPIC_CREATED: "Создание темы на форуме",
|
||||
ContentType.FORUM_TOPIC_EDITED: "Редактирование темы форума",
|
||||
ContentType.FORUM_TOPIC_CLOSED: "Закрытие темы форума",
|
||||
ContentType.FORUM_TOPIC_REOPENED: "Открытие темы форума",
|
||||
ContentType.GENERAL_FORUM_TOPIC_HIDDEN: "Скрытие общей темы форума",
|
||||
ContentType.GENERAL_FORUM_TOPIC_UNHIDDEN: "Раскрытие общей темы форума",
|
||||
ContentType.GIVEAWAY_CREATED: "Создание розыгрыша",
|
||||
ContentType.GIVEAWAY: "Розыгрыш",
|
||||
ContentType.GIVEAWAY_WINNERS: "Победители розыгрыша",
|
||||
ContentType.GIVEAWAY_COMPLETED: "Розыгрыш завершен",
|
||||
ContentType.VIDEO_CHAT_SCHEDULED: "Запланированный видеочат",
|
||||
ContentType.VIDEO_CHAT_STARTED: "Видеочат начат",
|
||||
ContentType.VIDEO_CHAT_ENDED: "Видеочат завершен",
|
||||
ContentType.VIDEO_CHAT_PARTICIPANTS_INVITED: "Участники приглашены в видеочат",
|
||||
ContentType.PINNED_MESSAGE: "Закрепленное сообщение",
|
||||
ContentType.INVOICE: "Счет",
|
||||
ContentType.SUCCESSFUL_PAYMENT: "Успешный платеж",
|
||||
ContentType.REFUNDED_PAYMENT: "Возврат платежа",
|
||||
ContentType.USERS_SHARED: "Пользователи поделились",
|
||||
ContentType.CHAT_SHARED: "Чат был передан",
|
||||
ContentType.CONNECTED_WEBSITE: "Подключенный веб-сайт",
|
||||
ContentType.WRITE_ACCESS_ALLOWED: "Разрешение на запись",
|
||||
ContentType.PASSPORT_DATA: "Данные паспорта",
|
||||
ContentType.PROXIMITY_ALERT_TRIGGERED: "Срабатывание предупреждения о близости",
|
||||
ContentType.BOOST_ADDED: "Буст чата",
|
||||
ContentType.CHAT_BACKGROUND_SET: "Установлен фон чата"
|
||||
}
|
||||
|
||||
# Проверяем тип сообщения и возвращаем описание
|
||||
if message.pinned_message: # Закрепленное сообщение
|
||||
return content_types.get(ContentType.PINNED_MESSAGE, "Закрепленное сообщение")
|
||||
|
||||
# Проверка для обычных сообщений
|
||||
for content_type, description in content_types.items():
|
||||
if getattr(message, str(content_type.value)):
|
||||
return description
|
||||
|
||||
# Если сообщение не соответствует ни одному из типов
|
||||
return "Неизвестный тип"
|
||||
6
BotLibrary/loggers/__init__.py
Normal file
6
BotLibrary/loggers/__init__.py
Normal file
@@ -0,0 +1,6 @@
|
||||
# BotLibrary/loggers/__init__.py
|
||||
# Инициализация пакета loggers, для создания логеров
|
||||
|
||||
# Экспортирование модулей во внешние слои проекта
|
||||
from .logs import *
|
||||
from .start_info_out import *
|
||||
@@ -1,11 +1,10 @@
|
||||
# BotLibrary/logs.py
|
||||
# BotLibrary/system/logs.py
|
||||
# Создание логгеров и их шаблон для проекта
|
||||
|
||||
import sys
|
||||
from loguru import logger
|
||||
from ProjectsFiles.configs.config import BotLogs
|
||||
|
||||
|
||||
# Создание обычного логгера + логгер в файл
|
||||
async def setup_logger():
|
||||
logger.remove() # Удаляем все логгеры
|
||||
43
BotLibrary/loggers/start_info_out.py
Normal file
43
BotLibrary/loggers/start_info_out.py
Normal file
@@ -0,0 +1,43 @@
|
||||
# BotLibrary/loggers/start_info_out.py
|
||||
# Вывод данных бота в консоль для начальной проверки
|
||||
|
||||
from time import sleep
|
||||
from colorama import Fore
|
||||
|
||||
from ProjectsFiles import Permissions
|
||||
from .logs import logger
|
||||
from ..system import BotInfo
|
||||
|
||||
|
||||
# Функция для получения информации о боте и выводе ее в консоль и файл
|
||||
def bot_info_out():
|
||||
try:
|
||||
bot_name = f"Основное имя: {BotInfo.first_name}\n"
|
||||
bot_post_name = f" Доп. имя: {BotInfo.last_name}\n"
|
||||
bot_username = f" Юзернейм: @{BotInfo.username}\n"
|
||||
bot_id = f" ID: {BotInfo.id}\n"
|
||||
bot_language = f" Языковой код: {BotInfo.language_code}\n"
|
||||
bot_can_join_groups = f" Может ли вступать в группы: {BotInfo.can_join_groups}\n"
|
||||
bot_can_read_all_group_messages = f" Чтение всех сообщений: {BotInfo.can_read_all_group_messages}\n"
|
||||
bot_is_premium = f" Является премиум-ботом: {BotInfo.is_premium}\n"
|
||||
bot_added_to_attachment_menu = f" Добавлен в меню вложений: {BotInfo.added_to_attachment_menu}\n"
|
||||
bot_supports_inline_queries = f" Поддерживает инлайн-запросы: {BotInfo.supports_inline_queries}\n"
|
||||
bot_can_connect_to_business = f" Подключение к бизнес-аккаунтам: {BotInfo.can_connect_to_business}\n"
|
||||
bot_has_main_web_app = f" Основное веб-приложение: {BotInfo.has_main_web_app}\n"
|
||||
|
||||
# Формируем полный текст с выводом информации о боте
|
||||
bot_all_info = (f"{bot_name} {bot_post_name} {bot_username} {bot_id} {bot_language} "
|
||||
f"{bot_can_join_groups} {bot_can_read_all_group_messages} {bot_is_premium} "
|
||||
f"{bot_added_to_attachment_menu} {bot_supports_inline_queries} {bot_can_connect_to_business} "
|
||||
f"{bot_has_main_web_app}")
|
||||
|
||||
# Печатаем все данные в консоль с задержкой в 1 секунду
|
||||
sleep(1)
|
||||
if Permissions.start_info_console:
|
||||
print(Fore.CYAN + bot_all_info)
|
||||
|
||||
return bot_all_info
|
||||
|
||||
# Проверка на ошибку и ее логирование
|
||||
except Exception as e:
|
||||
logger.bind(log_type="INFO", user="Start_INFO").error(f"Ошибка при получении ID пользователя: {e}")
|
||||
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)
|
||||
@@ -1,27 +1,29 @@
|
||||
# BotLibrary/library/bots.py
|
||||
# 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
|
||||
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
||||
from aiogram.utils.keyboard import InlineKeyboardBuilder, ReplyKeyboardBuilder
|
||||
|
||||
from ..timer import *
|
||||
from ProjectsFiles import *
|
||||
|
||||
# Создание экземпляра диспатчера, строителей кнопок
|
||||
dp = Dispatcher()
|
||||
rkb = ReplyKeyboardBuilder()
|
||||
ikb = InlineKeyboardBuilder()
|
||||
|
||||
|
||||
# Настройка параметров диспатчера
|
||||
dp["started_at"] = None
|
||||
dp["started_at_msk"] = None
|
||||
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": None}
|
||||
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"] = {}
|
||||
@@ -41,7 +43,6 @@ bot_properties = DefaultBotProperties(
|
||||
show_caption_above_media=False, # Показываем подпись выше медиа
|
||||
)
|
||||
bot = Bot(token=bot_token, default=bot_properties) # Объявление бота
|
||||
scheduler = AsyncIOScheduler(timezone=None) # Создание планировщика
|
||||
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 # Фильтр на все
|
||||
|
||||
5
BotLibrary/timer/__init__.py
Normal file
5
BotLibrary/timer/__init__.py
Normal file
@@ -0,0 +1,5 @@
|
||||
# BotLibrary/timer/__init__.py
|
||||
# Инициализация модуля timers, для проверки данных
|
||||
|
||||
# Экспортирование модулей во внешние слои проекта
|
||||
from .start_time import *
|
||||
23
BotLibrary/timer/start_time.py
Normal file
23
BotLibrary/timer/start_time.py
Normal file
@@ -0,0 +1,23 @@
|
||||
# BotLibrary/timer/start_time.py
|
||||
# Получение времени по
|
||||
|
||||
import pytz
|
||||
from datetime import datetime
|
||||
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
||||
from ProjectsFiles import BotVar
|
||||
|
||||
# Функция получение времени по Московскому времени
|
||||
def get_moscow_time():
|
||||
moscow_tz = pytz.timezone('Europe/Moscow')
|
||||
moscow_time = datetime.now(moscow_tz)
|
||||
return moscow_time.strftime(BotVar.time_format)
|
||||
|
||||
|
||||
# Функция получение времени хоста
|
||||
def get_host_time():
|
||||
host_time = datetime.now()
|
||||
return host_time.strftime(BotVar.time_format)
|
||||
|
||||
|
||||
# Создание планировщика
|
||||
scheduler = AsyncIOScheduler(timezone=get_moscow_time())
|
||||
5
BotLibrary/validators/__init__.py
Normal file
5
BotLibrary/validators/__init__.py
Normal file
@@ -0,0 +1,5 @@
|
||||
# BotLibrary/validators/__init__.py
|
||||
# Инициализация модуля validators, для проверки данных
|
||||
|
||||
# Экспортирование модулей во внешние слои проекта
|
||||
from .email_valid import *
|
||||
16
BotLibrary/validators/email_valid.py
Normal file
16
BotLibrary/validators/email_valid.py
Normal file
@@ -0,0 +1,16 @@
|
||||
# BotLibrary/validators/email_validators.py
|
||||
# Создание валидации почты для проекта
|
||||
|
||||
from email_validator import validate_email, EmailNotValidError
|
||||
|
||||
# Настройка экспорта из этого модуля
|
||||
__all__ = ("valid_email",)
|
||||
|
||||
|
||||
# Функция проверки почты на корректность
|
||||
def valid_email(text: str) -> str | None:
|
||||
try:
|
||||
email = validate_email(text)
|
||||
except EmailNotValidError:
|
||||
return None
|
||||
return email.normalized
|
||||
Reference in New Issue
Block a user