mirror of
https://github.com/BEDOLAGA-DEV/remnawave-bedolaga-telegram-bot.git
synced 2026-02-07 20:50:34 +00:00
3aa5e304c3
- Updated ban notification messages to provide detailed reasons for account bans, including node information. - Refactored the BanNotificationService to safely format messages with optional node details. - Modified API routes and schemas to support the inclusion of node names in ban notifications.
63 lines
2.8 KiB
Python
63 lines
2.8 KiB
Python
from __future__ import annotations
|
||
|
||
from typing import Literal, Optional
|
||
from pydantic import BaseModel, Field
|
||
|
||
|
||
class BanNotificationRequest(BaseModel):
|
||
"""Запрос на отправку уведомления о бане пользователю"""
|
||
|
||
notification_type: Literal["punishment", "enabled", "warning", "network_wifi", "network_mobile"] = Field(
|
||
description="Тип уведомления: punishment (бан за устройства), enabled (разбан), warning (предупреждение), network_wifi (бан за WiFi), network_mobile (бан за мобильную сеть)"
|
||
)
|
||
user_identifier: str = Field(
|
||
description="Email или user_id пользователя из Remnawave Panel"
|
||
)
|
||
username: str = Field(
|
||
description="Имя пользователя для отображения"
|
||
)
|
||
|
||
# Данные для punishment
|
||
ip_count: Optional[int] = Field(None, description="Количество устройств")
|
||
limit: Optional[int] = Field(None, description="Лимит устройств")
|
||
ban_minutes: Optional[int] = Field(None, description="Длительность бана в минутах")
|
||
|
||
# Данные для warning
|
||
warning_message: Optional[str] = Field(None, description="Текст предупреждения")
|
||
|
||
# Данные для network_wifi/network_mobile и punishment
|
||
network_type: Optional[str] = Field(None, description="Тип сети (WiFi/Mobile)")
|
||
node_name: Optional[str] = Field(None, description="Название ноды/сервера с которой пришел бан")
|
||
|
||
class Config:
|
||
json_schema_extra = {
|
||
"example": {
|
||
"notification_type": "punishment",
|
||
"user_identifier": "user@example.com",
|
||
"username": "john_doe",
|
||
"ip_count": 5,
|
||
"limit": 3,
|
||
"ban_minutes": 30,
|
||
"node_name": "DE-Server-1"
|
||
}
|
||
}
|
||
|
||
|
||
class BanNotificationResponse(BaseModel):
|
||
"""Ответ на запрос отправки уведомления"""
|
||
|
||
success: bool = Field(description="Успешно ли отправлено уведомление")
|
||
message: str = Field(description="Сообщение о результате")
|
||
telegram_id: Optional[int] = Field(None, description="Telegram ID получателя")
|
||
sent: bool = Field(False, description="Было ли фактически отправлено сообщение")
|
||
|
||
class Config:
|
||
json_schema_extra = {
|
||
"example": {
|
||
"success": True,
|
||
"message": "Уведомление отправлено",
|
||
"telegram_id": 123456789,
|
||
"sent": True
|
||
}
|
||
}
|