src.db.models.wg_config

 18class WgConfig(Base):
 19    """Модель конфигурации WireGuard.
 20
 21    Эта модель представляет собой структуру данных для хранения конфигураций
 22    WireGuard, связанных с пользователями.
 23    """
 24
 25    __tablename__ = "wg_config"
 26    __table_args__ = {"extend_existing": True}
 27
 28    id: Mapped[int] = mapped_column(primary_key=True)
 29    """int: Уникальный идентификатор конфигурации WireGuard."""
 30
 31    user_id: Mapped[int] = mapped_column(
 32        ForeignKey("userdata.telegram_id", ondelete="CASCADE"),
 33        type_=BigInteger,
 34    )
 35    """int: Идентификатор пользователя в Telegram, связанный с конфигурацией."""
 36
 37    name: Mapped[str] = mapped_column(default=name_gen.get_random_word, unique=True)
 38    """str: Название конфигурации (по умолчанию генерируется случайным образом)."""
 39
 40    freeze: Mapped[FreezeSteps] = mapped_column(
 41        Enum(FreezeSteps, values_callable=lambda obj: [e.value for e in obj]),
 42        default=FreezeSteps.no,
 43    )
 44    """FreezeSteps: Статус заморозки конфигурации (по умолчанию 'no')."""
 45
 46    user_private_key: Mapped[str] = mapped_column(String(44))
 47    """str: Приватный ключ пользователя для WireGuard."""
 48
 49    address: Mapped[IPv4Interface] = mapped_column(type_=CIDR)
 50    """IPv4Interface: IP-адрес конфигурации WireGuard."""
 51
 52    dns: Mapped[str] = mapped_column(default="10.0.0.1,9.9.9.9")
 53    """str: DNS-серверы для конфигурации (по умолчанию '10.0.0.1,9.9.9.9')."""
 54
 55    server_public_key: Mapped[str] = mapped_column(String(44))
 56    """str: Публичный ключ сервера для WireGuard."""
 57
 58    allowed_ips: Mapped[IPv4Interface] = mapped_column(type_=CIDR, default="0.0.0.0/0")
 59    """IPv4Interface: Разрешенные IP-адреса для конфигурации (по умолчанию '0.0.0.0/0')."""
 60
 61    endpoint_ip: Mapped[IPv4Address] = mapped_column(
 62        type_=INET, default=settings.WG_HOST
 63    )
 64    """IPv4Address: IP-адрес конечной точки WireGuard (по умолчанию из настроек)."""
 65
 66    endpoint_port: Mapped[int] = mapped_column(default=settings.WG_PORT)
 67    """int: Порт конечной точки WireGuard (по умолчанию из настроек)."""
 68
 69    conf_connect: Mapped["UserData"] = relationship(  # noqa: F821 # type: ignore
 70        back_populates="configs", lazy="subquery"
 71    )
 72    """UserData: Связанные данные пользователя для конфигурации WireGuard."""
 73
 74    class ValidationSchema(BaseModel):
 75        """Схема валидации для модели конфигурации WireGuard.
 76
 77        Эта схема используется для валидации данных, связанных с конфигурациями.
 78        """
 79
 80        id: int | None = Field(default=None, title="ID")
 81        """Уникальный идентификатор конфигурации WireGuard."""
 82
 83        user_id: int = Field(title="Telegram ID")
 84        """Идентификатор пользователя в Telegram."""
 85
 86        name: str = Field(title="Title")
 87        """Название конфигурации."""
 88
 89        freeze: FreezeSteps = Field(default=FreezeSteps.no, title="Freeze")
 90        """Статус заморозки конфигурации."""
 91
 92        user_private_key: str = Field(title="User Priv Key")
 93        """Приватный ключ пользователя для WireGuard."""
 94
 95        address: IPv4Interface = Field(title="Address")
 96        """IP-адрес конфигурации WireGuard."""
 97
 98        dns: str = Field(default="10.0.0.1,9.9.9.9", title="DNS")
 99        """DNS-серверы для конфигурации."""
100
101        server_public_key: str = Field(title="Server Pub Key")
102        """Публичный ключ сервера для WireGuard."""
103
104        allowed_ips: IPv4Interface = Field(default="0.0.0.0/0", title="Allowed IPs")
105        """Разрешенные IP-адреса для конфигурации."""
106
107        endpoint_ip: IPv4Address = Field(default=settings.WG_HOST, title="Endpoint IP")
108        """IP-адрес конечной точки WireGuard."""
109
110        endpoint_port: int = Field(default=settings.WG_PORT, title="Endpoint Port")
111        """Порт конечной точки WireGuard."""
112
113        model_config = ConfigDict(extra="ignore")
114
115    # INTERFACE (fastui)
116    site_display = [
117        DisplayLookup(field="id"),
118        DisplayLookup(
119            field="user_id",
120            mode=DisplayMode.plain,
121            on_click=GoToEvent(url="/bot/tables/userdata/?telegram_id={user_id}"),
122        ),
123        DisplayLookup(
124            field="name",
125            mode=DisplayMode.as_title,
126            on_click=GoToEvent(url="/bot/tables/wg_config/?name={name}"),
127        ),
128        DisplayLookup(field="freeze"),
129        DisplayLookup(field="address"),
130    ]
131    """list[DisplayLookup]: Отображение конфигураций WireGuard на сайте."""
132
133    site_display_all = site_display + [
134        DisplayLookup(field="user_private_key", mode=DisplayMode.inline_code),
135        DisplayLookup(field="server_public_key", mode=DisplayMode.inline_code),
136        DisplayLookup(field="dns"),
137        DisplayLookup(field="allowed_ips"),
138        DisplayLookup(field="endpoint_ip"),
139        DisplayLookup(field="endpoint_port"),
140    ]
141    """list[DisplayLookup]: Полное отображение конфигураций WireGuard на сайте."""
142
143    def __init__(self, **kwargs):
144        """Инициализирует модель конфигурации WireGuard.
145
146        Args:
147            **kwargs: Дополнительные параметры для инициализации модели.
148        """
149        if kwargs:
150            validated_data = self.ValidationSchema(**kwargs).__dict__
151            super().__init__(**(validated_data))
152        else:
153            super().__init__(**(kwargs))

 74    class ValidationSchema(BaseModel):
 75        """Схема валидации для модели конфигурации WireGuard.
 76
 77        Эта схема используется для валидации данных, связанных с конфигурациями.
 78        """
 79
 80        id: int | None = Field(default=None, title="ID")
 81        """Уникальный идентификатор конфигурации WireGuard."""
 82
 83        user_id: int = Field(title="Telegram ID")
 84        """Идентификатор пользователя в Telegram."""
 85
 86        name: str = Field(title="Title")
 87        """Название конфигурации."""
 88
 89        freeze: FreezeSteps = Field(default=FreezeSteps.no, title="Freeze")
 90        """Статус заморозки конфигурации."""
 91
 92        user_private_key: str = Field(title="User Priv Key")
 93        """Приватный ключ пользователя для WireGuard."""
 94
 95        address: IPv4Interface = Field(title="Address")
 96        """IP-адрес конфигурации WireGuard."""
 97
 98        dns: str = Field(default="10.0.0.1,9.9.9.9", title="DNS")
 99        """DNS-серверы для конфигурации."""
100
101        server_public_key: str = Field(title="Server Pub Key")
102        """Публичный ключ сервера для WireGuard."""
103
104        allowed_ips: IPv4Interface = Field(default="0.0.0.0/0", title="Allowed IPs")
105        """Разрешенные IP-адреса для конфигурации."""
106
107        endpoint_ip: IPv4Address = Field(default=settings.WG_HOST, title="Endpoint IP")
108        """IP-адрес конечной точки WireGuard."""
109
110        endpoint_port: int = Field(default=settings.WG_PORT, title="Endpoint Port")
111        """Порт конечной точки WireGuard."""
112
113        model_config = ConfigDict(extra="ignore")