mirror of
https://gitflic.ru/project/alt-gnome/karkas.git
synced 2024-12-23 16:23:02 +03:00
удален блок standard.message_processing
This commit is contained in:
parent
8512b3300b
commit
5513481330
@ -1,14 +0,0 @@
|
|||||||
# Модуль Message Processing
|
|
||||||
|
|
||||||
Модуль `message_processing` обрабатывает все входящие сообщения.
|
|
||||||
|
|
||||||
## Функциональность
|
|
||||||
|
|
||||||
- Проверка чата и пользователя на наличие в базе данных.
|
|
||||||
- Обновление информации о чате и пользователе.
|
|
||||||
- Добавление статистики сообщений.
|
|
||||||
- Передача сообщения модулю `yandexgpt`, если оно соответствует условиям.
|
|
||||||
|
|
||||||
## Использование
|
|
||||||
|
|
||||||
Модуль автоматически обрабатывает все входящие сообщения.
|
|
@ -1 +0,0 @@
|
|||||||
from .message_api import module_init
|
|
@ -1,15 +0,0 @@
|
|||||||
{
|
|
||||||
"id": "standard.message_processing",
|
|
||||||
"name": "Info",
|
|
||||||
"description": "Модуль с информацией",
|
|
||||||
"author": "Karkas Team",
|
|
||||||
"version": "1.0.0",
|
|
||||||
"privileged": false,
|
|
||||||
"dependencies": {
|
|
||||||
"required": {
|
|
||||||
"standard.roles": "^1.0.0",
|
|
||||||
"standard.database": "^1.0.0",
|
|
||||||
"standard.command_helper": "^1.0.0"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
@ -1,155 +0,0 @@
|
|||||||
# flake8: noqa
|
|
||||||
|
|
||||||
from typing import TYPE_CHECKING
|
|
||||||
|
|
||||||
from aiogram import Bot, F, Router, types
|
|
||||||
|
|
||||||
from karkas_core.modules_system.public_api import get_module, log, register_router
|
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
|
||||||
from karkas_blocks.standard.config import IConfig
|
|
||||||
|
|
||||||
config: "IConfig" = get_module("standard.config", "config")
|
|
||||||
|
|
||||||
|
|
||||||
def get_yandexgpt_start_words():
|
|
||||||
return config.get("yandexgpt::startword").split(" | ")
|
|
||||||
|
|
||||||
|
|
||||||
def get_yandexgpt_in_words():
|
|
||||||
return config.get("yandexgpt::inword").split(" | ")
|
|
||||||
|
|
||||||
|
|
||||||
chat_not_in_approve = get_module("standard.filters", ["chat_not_in_approve"])
|
|
||||||
|
|
||||||
answer_to_message = get_module("external.yandexgpt", "answer_to_message")
|
|
||||||
|
|
||||||
(
|
|
||||||
get_chat,
|
|
||||||
add_chat,
|
|
||||||
get_user,
|
|
||||||
add_user,
|
|
||||||
get_user_name,
|
|
||||||
change_user_name,
|
|
||||||
get_user_tag,
|
|
||||||
change_user_tag,
|
|
||||||
update_chat_all_stat,
|
|
||||||
update_user_all_stat,
|
|
||||||
add_message,
|
|
||||||
) = get_module(
|
|
||||||
"standard.database",
|
|
||||||
[
|
|
||||||
"db_api.get_chat",
|
|
||||||
"db_api.add_chat",
|
|
||||||
"db_api.get_user",
|
|
||||||
"db_api.add_user",
|
|
||||||
"db_api.get_user_name",
|
|
||||||
"db_api.change_user_name",
|
|
||||||
"db_api.get_user_tag",
|
|
||||||
"db_api.change_user_tag",
|
|
||||||
"db_api.update_chat_all_stat",
|
|
||||||
"db_api.update_user_all_stat",
|
|
||||||
"db_api.add_message",
|
|
||||||
],
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
async def chat_check(message: types.Message):
|
|
||||||
# Проверка наличия id чата в базе данных чатов
|
|
||||||
# Если чата нет в базе данных, то проверяем его в наличии в конфиге и если он там есть то добавляем его в БД
|
|
||||||
# Если чат есть в базе данных, то pass
|
|
||||||
if get_chat(message.chat.id) is None:
|
|
||||||
if not chat_not_in_approve(message):
|
|
||||||
# print(f"Chat in approve list: {message.chat.id} {message.chat.title}")
|
|
||||||
log(f"Chat in approve list: {message.chat.id} {message.chat.title}")
|
|
||||||
add_chat(message.chat.id, message.chat.title)
|
|
||||||
# print(f"Chat added: {message.chat.id} {message.chat.title}")
|
|
||||||
log(f"Chat added: {message.chat.id} {message.chat.title}")
|
|
||||||
else:
|
|
||||||
# print(f"Chat not in approve list: {message.chat.id} {message.chat.title}")
|
|
||||||
log(f"Chat not in approve list: {message.chat.id} {message.chat.title}")
|
|
||||||
pass
|
|
||||||
else:
|
|
||||||
# Проверяем обновление названия чата
|
|
||||||
chat = get_chat(message.chat.id)
|
|
||||||
if chat.chat_name != message.chat.title:
|
|
||||||
chat.chat_name = message.chat.title
|
|
||||||
chat.save()
|
|
||||||
# print(f"Chat updated: {message.chat.id} {message.chat.title}")
|
|
||||||
log(f"Chat updated: {message.chat.id} {message.chat.title}")
|
|
||||||
else:
|
|
||||||
# print(f"Chat already exists: {message.chat.id} {message.chat.title}")
|
|
||||||
log(f"Chat already exists: {message.chat.id} {message.chat.title}")
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
async def user_check(message: types.Message):
|
|
||||||
# Проверка наличия id пользователя в базе данных пользователей
|
|
||||||
# Если пользователя нет в базе данных, то добавляем его
|
|
||||||
# Если пользователь есть в базе данных, то pass
|
|
||||||
current_user_name = ""
|
|
||||||
if message.from_user.last_name is None:
|
|
||||||
current_user_name = message.from_user.first_name
|
|
||||||
else:
|
|
||||||
current_user_name = (
|
|
||||||
message.from_user.first_name + " " + message.from_user.last_name
|
|
||||||
)
|
|
||||||
|
|
||||||
if get_user(message.from_user.id) is None:
|
|
||||||
add_user(
|
|
||||||
message.from_user.id,
|
|
||||||
message.from_user.first_name,
|
|
||||||
message.from_user.last_name,
|
|
||||||
message.from_user.username,
|
|
||||||
)
|
|
||||||
log(f"User added: {message.from_user.id} {current_user_name}")
|
|
||||||
else:
|
|
||||||
log(
|
|
||||||
f"User already exists: {message.from_user.id} {current_user_name} {message.from_user.username}"
|
|
||||||
)
|
|
||||||
# Проверяем обновление имени пользователя
|
|
||||||
if get_user_name(message.from_user.id) != current_user_name:
|
|
||||||
change_user_name(message.from_user.id, current_user_name)
|
|
||||||
log(f"User name updated: {message.from_user.id} {current_user_name}")
|
|
||||||
# Проверяем обновление username пользователя
|
|
||||||
if get_user_tag(message.from_user.id) != message.from_user.username:
|
|
||||||
change_user_tag(message.from_user.id, message.from_user.username)
|
|
||||||
log(
|
|
||||||
f"User tag updated: {message.from_user.id} {message.from_user.username}"
|
|
||||||
)
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
async def add_stats(message: types.Message):
|
|
||||||
# Добавляем пользователю и чату статистику
|
|
||||||
update_chat_all_stat(message.chat.id)
|
|
||||||
update_user_all_stat(message.from_user.id)
|
|
||||||
|
|
||||||
|
|
||||||
async def message_processing(message: types.Message, bot: Bot):
|
|
||||||
await chat_check(message)
|
|
||||||
await user_check(message)
|
|
||||||
await add_stats(message)
|
|
||||||
|
|
||||||
add_message(message)
|
|
||||||
# Если сообщение в начале содержит слово из списка или внутри сообщения содержится слово из списка или сообщение отвечает на сообщение бота
|
|
||||||
|
|
||||||
if (message.text.split(" ")[0] in get_yandexgpt_start_words()) or (
|
|
||||||
any(word in message.text for word in get_yandexgpt_in_words())
|
|
||||||
):
|
|
||||||
log("message_processing")
|
|
||||||
await answer_to_message(message, bot)
|
|
||||||
|
|
||||||
elif message.reply_to_message is not None:
|
|
||||||
if message.reply_to_message.from_user.is_bot:
|
|
||||||
log("message_processing")
|
|
||||||
await answer_to_message(message, bot)
|
|
||||||
|
|
||||||
|
|
||||||
router = Router()
|
|
||||||
# Если сообщение содержит текст то вызывается функция message_processing
|
|
||||||
router.message.register(message_processing, F.text)
|
|
||||||
|
|
||||||
|
|
||||||
async def module_init():
|
|
||||||
register_router(router)
|
|
Loading…
Reference in New Issue
Block a user