c0mrade
9a2aea038a
- Add pyproject.toml with uv and ruff configuration - Pin Python version to 3.13 via .python-version - Add Makefile commands: lint, format, fix - Apply ruff formatting to entire codebase - Remove unused imports (base64 in yookassa/simple_subscription) - Update .gitignore for new config files
55 lines
2.6 KiB
Python
55 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Literal
|
|
|
|
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: int | None = Field(None, description='Количество устройств')
|
|
limit: int | None = Field(None, description='Лимит устройств')
|
|
ban_minutes: int | None = Field(None, description='Длительность бана в минутах')
|
|
|
|
# Данные для warning
|
|
warning_message: str | None = Field(None, description='Текст предупреждения')
|
|
|
|
# Данные для network_wifi/network_mobile и punishment
|
|
network_type: str | None = Field(None, description='Тип сети (WiFi/Mobile)')
|
|
node_name: str | None = 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: int | None = Field(None, description='Telegram ID получателя')
|
|
sent: bool = Field(False, description='Было ли фактически отправлено сообщение')
|
|
|
|
class Config:
|
|
json_schema_extra = {
|
|
'example': {'success': True, 'message': 'Уведомление отправлено', 'telegram_id': 123456789, 'sent': True}
|
|
}
|