from pydantic import BaseModel


class NoteRequest(BaseModel):
    """
    Модель добавления заметки к просканированному устройству

    Атрибуты:
        note (str): Заметка устройства
    """

    note: str

class WifiConnectRequest(BaseModel):
    """
    Модель для подключения к точке доступа Wi-Fi

    Атрибуты:
        ssid (str): Название точки доступа Wi-Fi
        password (str): Пароль точки доступа Wi-Fi
    """
    ssid: str
    password: str

class WifiDisconnectRequest(BaseModel):
    """
    Модель для отключения от точки доступа Wi-Fi

    Атрибуты:
        ssid (str): Название точки доступа Wi-Fi
    """
    ssid: str