Инициализация интернационализации. Перевод на en, ru для эндпоинтов api/v1
This commit is contained in:
parent
48e8551b0e
commit
9ac52c8a8b
16 changed files with 990 additions and 181 deletions
|
|
@ -3,6 +3,8 @@ from fastapi.security import OAuth2PasswordBearer
|
|||
from services.auth_service import AuthService
|
||||
from models.auth import UserCreate, UserLogin, UserResponse, Token
|
||||
from storage.json_storage import JsonStorage
|
||||
from utils.i18n import get_translations, _
|
||||
from babel.support import Translations
|
||||
import logging
|
||||
import os
|
||||
|
||||
|
|
@ -13,9 +15,8 @@ router = APIRouter()
|
|||
auth_service = AuthService()
|
||||
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="/auth/login")
|
||||
|
||||
|
||||
@router.post("/register", response_model=UserResponse)
|
||||
async def register(user: UserCreate):
|
||||
async def register(user: UserCreate, translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Регистрирует нового пользователя.
|
||||
|
||||
Создаёт нового пользователя на основе предоставленных данных и возвращает
|
||||
|
|
@ -23,6 +24,7 @@ async def register(user: UserCreate):
|
|||
|
||||
Args:
|
||||
user (UserCreate): Объект с данными нового пользователя (username, password, is_admin).
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
UserResponse: Объект с данными зарегистрированного пользователя (id, username, is_admin).
|
||||
|
|
@ -38,12 +40,11 @@ async def register(user: UserCreate):
|
|||
except HTTPException as e:
|
||||
raise e
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка в register: {type(e).__name__} - {str(e)}")
|
||||
raise HTTPException(status_code=500, detail="Не удалось зарегистрировать пользователя")
|
||||
|
||||
logger.error(f"Error in register: {type(e).__name__} - {str(e)}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Не удалось зарегистрировать пользователя"))
|
||||
|
||||
@router.post("/login", response_model=Token)
|
||||
async def login(user: UserLogin, request: Request):
|
||||
async def login(user: UserLogin, request: Request, translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Авторизует пользователя и создаёт сессию.
|
||||
|
||||
Проверяет учетные данные пользователя и возвращает токен и session_id для
|
||||
|
|
@ -52,6 +53,7 @@ async def login(user: UserLogin, request: Request):
|
|||
Args:
|
||||
user (UserLogin): Объект с данными для входа (username, password).
|
||||
request (Request): Объект запроса для получения client_id (IP-адрес клиента).
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
Token: Объект с access_token и session_id для авторизации.
|
||||
|
|
@ -62,21 +64,18 @@ async def login(user: UserLogin, request: Request):
|
|||
"""
|
||||
try:
|
||||
client_id = request.client.host
|
||||
logger.debug(f"Попытка авторизации для client_id {client_id}")
|
||||
logger.debug(f"Попытка входа в систему клиента {client_id}")
|
||||
result = auth_service.login(user, client_id)
|
||||
logger.info(
|
||||
f"Успешная авторизация для client_id {client_id}, session_id: {result['session_id']}"
|
||||
)
|
||||
logger.info(f"Успешная авторизация для клиента {client_id}, Идентификатор сессии: {result['session_id']}")
|
||||
return result
|
||||
except HTTPException as e:
|
||||
raise e
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка в login: {type(e).__name__} - {str(e)}")
|
||||
raise HTTPException(status_code=500, detail="Не удалось авторизовать пользователя")
|
||||
|
||||
logger.error(f"Ошибка при авторизации: {type(e).__name__} - {str(e)}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Не удалось авторизовать пользователя"))
|
||||
|
||||
@router.get("/me")
|
||||
async def get_current_user(request: Request):
|
||||
async def get_current_user(request: Request, translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Возвращает данные текущего пользователя и токен.
|
||||
|
||||
Проверяет активную сессию по client_id и возвращает информацию о пользователе
|
||||
|
|
@ -84,6 +83,7 @@ async def get_current_user(request: Request):
|
|||
|
||||
Args:
|
||||
request (Request): Объект запроса для получения client_id (IP-адрес клиента).
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с ключами "user" (UserResponse), "access_token" (str) и
|
||||
|
|
@ -95,29 +95,28 @@ async def get_current_user(request: Request):
|
|||
"""
|
||||
try:
|
||||
client_id = request.client.host
|
||||
logger.debug(f"Запрос /me для client_id {client_id}")
|
||||
logger.debug(f"Запрос /me для клиента {client_id}")
|
||||
result = auth_service.get_user_by_session(client_id)
|
||||
return {
|
||||
"user": result["user"],
|
||||
"access_token": result["access_token"],
|
||||
"token_type": "bearer",
|
||||
"token_type": translations.gettext("bearer"),
|
||||
}
|
||||
except HTTPException as e:
|
||||
raise e
|
||||
except Exception as e:
|
||||
logger.error(
|
||||
f"Ошибка в get_current_user: {type(e).__name__} - {str(e)}")
|
||||
raise HTTPException(status_code=500, detail="Не удалось получить данные пользователя")
|
||||
|
||||
logger.error(f"Ошибка при get_current_user: {type(e).__name__} - {str(e)}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Не удалось получить данные пользователя"))
|
||||
|
||||
@router.get("/logout")
|
||||
async def logout(request: Request):
|
||||
async def logout(request: Request, translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Выход из системы, удаление сессии.
|
||||
|
||||
Удаляет активную сессию пользователя по client_id.
|
||||
|
||||
Args:
|
||||
request (Request): Объект запроса для получения client_id (IP-адрес клиента).
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Пустой словарь или сообщение об успешном выходе.
|
||||
|
|
@ -128,17 +127,16 @@ async def logout(request: Request):
|
|||
"""
|
||||
try:
|
||||
client_id = request.client.host
|
||||
logger.debug(f"Запрос logout для client_id {client_id}")
|
||||
logger.debug(f"Запрос на выход из системы для клиента {client_id}")
|
||||
return auth_service.logout(client_id)
|
||||
except HTTPException as e:
|
||||
raise e
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка в logout: {type(e).__name__} - {str(e)}")
|
||||
raise HTTPException(status_code=500, detail="Не удалось выйти из системы")
|
||||
|
||||
logger.error(f"Не удалось выйти из системы: {type(e).__name__} - {str(e)}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Не удалось выйти из системы"))
|
||||
|
||||
@router.get("/check_json")
|
||||
async def check_users_file():
|
||||
async def check_users_file(translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Проверяет, существует ли файл users.json и содержит ли он пользователей.
|
||||
|
||||
Возвращает информацию о состоянии файла и наличии пользователей.
|
||||
|
|
@ -161,31 +159,25 @@ async def check_users_file():
|
|||
return {
|
||||
"exists": False,
|
||||
"has_users": False,
|
||||
"message": "Файл users.json не существует",
|
||||
"message": translations.gettext("Файл users.json не существует"),
|
||||
}
|
||||
|
||||
data = storage._read_data()
|
||||
has_users = len(data.get("users", [])) > 0
|
||||
if has_users:
|
||||
logger.info(
|
||||
f"Файл {file_path} существует и содержит пользователей"
|
||||
)
|
||||
logger.info(f"Файл {file_path} существует и содержит пользователей")
|
||||
return {
|
||||
"exists": True,
|
||||
"has_users": True,
|
||||
"message": "Файл users.json существует и содержит пользователей",
|
||||
"message": translations.gettext("Файл users.json существует и содержит пользователей"),
|
||||
}
|
||||
else:
|
||||
logger.info(
|
||||
f"Файл {file_path} существует, но не содержит пользователей"
|
||||
)
|
||||
logger.info(f"Файл {file_path} существует, но не содержит пользователей")
|
||||
return {
|
||||
"exists": True,
|
||||
"has_users": False,
|
||||
"message": "Файл users.json существует, но не содержит пользователей",
|
||||
"message": translations.gettext("Файл users.json существует, но не содержит пользователей"),
|
||||
}
|
||||
except Exception as e:
|
||||
logger.error(
|
||||
f"Ошибка при проверке файла users.json: {type(e).__name__} - {str(e)}"
|
||||
)
|
||||
raise HTTPException(status_code=500, detail="Ошибка при проверке файла")
|
||||
logger.error(f"Ошибка при проверке файла: {type(e).__name__} - {str(e)}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Ошибка при проверке файла"))
|
||||
|
|
@ -1,13 +1,14 @@
|
|||
from typing import Dict, Any
|
||||
from fastapi import APIRouter, Body, HTTPException
|
||||
from fastapi import APIRouter, Body, HTTPException, Depends
|
||||
from services.ip_service import IPService
|
||||
from utils.i18n import get_translations, _
|
||||
from babel.support import Translations
|
||||
|
||||
router = APIRouter()
|
||||
ip_service = IPService()
|
||||
|
||||
|
||||
@router.post("/check_ip")
|
||||
async def check_ip_config(data: Dict[str, Any] = Body(...)):
|
||||
async def check_ip_config(data: Dict[str, Any] = Body(...), translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Проверяет доступность IP-адреса перед его применением.
|
||||
|
||||
Выполняет ARP-сканирование для проверки, занят ли указанный IP-адрес,
|
||||
|
|
@ -16,6 +17,7 @@ async def check_ip_config(data: Dict[str, Any] = Body(...)):
|
|||
Args:
|
||||
data (Dict[str, Any]): Словарь с данными конфигурации, включая
|
||||
"ipv4_updates" с полем "ipv4.address-data" для нового IP-адреса.
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
Dict[str, Any]: Словарь с ключами:
|
||||
|
|
@ -31,11 +33,11 @@ async def check_ip_config(data: Dict[str, Any] = Body(...)):
|
|||
except Exception as e:
|
||||
raise HTTPException(
|
||||
status_code=500,
|
||||
detail=f"Ошибка проверки IP конфигурации: {str(e)}")
|
||||
|
||||
detail=translations.gettext("Ошибка проверки IP конфигурации: {error}").format(error=str(e))
|
||||
)
|
||||
|
||||
@router.post("/apply_ip")
|
||||
async def apply_ip_config(data: Dict[str, Any] = Body(...)):
|
||||
async def apply_ip_config(data: Dict[str, Any] = Body(...), translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Применяет проверенные IP-настройки к указанному сетевому профилю.
|
||||
|
||||
Обновляет настройки соединения через NetworkManager, отключает и
|
||||
|
|
@ -44,6 +46,7 @@ async def apply_ip_config(data: Dict[str, Any] = Body(...)):
|
|||
Args:
|
||||
data (Dict[str, Any]): Словарь с данными конфигурации, включая
|
||||
"connection_id" и "ipv4_updates" с обновлениями IP-настроек.
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с ключами:
|
||||
|
|
@ -58,16 +61,20 @@ async def apply_ip_config(data: Dict[str, Any] = Body(...)):
|
|||
await ip_service.apply_ip_config(data)
|
||||
return {
|
||||
"success": True,
|
||||
"message": f"Настройки сетевого профиля для {data['connection_id']} применены",
|
||||
"message": translations.gettext("Настройки сетевого профиля для {connection_id} применены").format(
|
||||
connection_id=data["connection_id"]
|
||||
),
|
||||
}
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
raise HTTPException(status_code=404, detail=translations.gettext(str(e)))
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=f"Ошибка проверки IP конфигурации: {str(e)}")
|
||||
|
||||
raise HTTPException(
|
||||
status_code=500,
|
||||
detail=translations.gettext("Ошибка проверки IP конфигурации: {error}").format(error=str(e))
|
||||
)
|
||||
|
||||
@router.post("/save_ip")
|
||||
async def save_ip_config(data: Dict[str, Any] = Body(...)):
|
||||
async def save_ip_config(data: Dict[str, Any] = Body(...), translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Применяет проверенные IP-настройки для отката сетевого профиля.
|
||||
|
||||
Используется для сохранения и отката настроек сетевого профиля к предыдущему
|
||||
|
|
@ -76,6 +83,7 @@ async def save_ip_config(data: Dict[str, Any] = Body(...)):
|
|||
Args:
|
||||
data (Dict[str, Any]): Словарь с данными конфигурации, включая
|
||||
"connection_id" и "ipv4_updates" с обновлениями IP-настроек.
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с ключами:
|
||||
|
|
@ -90,9 +98,14 @@ async def save_ip_config(data: Dict[str, Any] = Body(...)):
|
|||
await ip_service.apply_ip_config(data)
|
||||
return {
|
||||
"success": True,
|
||||
"message": f"Настройки сетевого профиля для {data['connection_id']} применены",
|
||||
"message": translations.gettext("Настройки сетевого профиля для {connection_id} применены").format(
|
||||
connection_id=data["connection_id"]
|
||||
),
|
||||
}
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
raise HTTPException(status_code=404, detail=translations.gettext(str(e)))
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=f"Ошибка проверки IP конфигурации: {str(e)}")
|
||||
raise HTTPException(
|
||||
status_code=500,
|
||||
detail=translations.gettext("Ошибка проверки IP конфигурации: {error}").format(error=str(e))
|
||||
)
|
||||
|
|
@ -1,7 +1,6 @@
|
|||
from fastapi import APIRouter, Body, HTTPException
|
||||
from fastapi import APIRouter, Body, HTTPException, Depends
|
||||
import logging
|
||||
from typing import Dict, Tuple, List, Union
|
||||
|
||||
from services.network_service import NetworkService
|
||||
from models.network import (
|
||||
NetworkInterfacesContainer,
|
||||
|
|
@ -10,6 +9,8 @@ from models.network import (
|
|||
NetworkInterface,
|
||||
WiFiDevice,
|
||||
)
|
||||
from utils.i18n import get_translations, _
|
||||
from babel.support import Translations
|
||||
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
logger = logging.getLogger(__name__)
|
||||
|
|
@ -17,7 +18,6 @@ logger = logging.getLogger(__name__)
|
|||
router = APIRouter()
|
||||
network_service = NetworkService()
|
||||
|
||||
|
||||
class ContainerPositionManager:
|
||||
"""Класс для управления позициями и состоянием контейнеров.
|
||||
|
||||
|
|
@ -44,11 +44,12 @@ class ContainerPositionManager:
|
|||
},
|
||||
}
|
||||
|
||||
def get_position_and_state(self, container_name: str) -> Tuple[int, int, bool]:
|
||||
def get_position_and_state(self, container_name: str, translations: Translations = Depends(get_translations("api_v1"))) -> Tuple[int, int, bool]:
|
||||
"""Получает текущие координаты и состояние указанного контейнера.
|
||||
|
||||
Args:
|
||||
container_name (str): Имя контейнера ("activation" или "deactivation").
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
Tuple[int, int, bool]: Кортеж с координатами (x, y) и состоянием (isOpen).
|
||||
|
|
@ -58,7 +59,7 @@ class ContainerPositionManager:
|
|||
"""
|
||||
container = self.positions.get(container_name)
|
||||
if not container:
|
||||
raise ValueError(f"Неверное имя контейнера: {container_name}")
|
||||
raise ValueError(translations.gettext("Неверное имя контейнера: {container_name}").format(container_name=container_name))
|
||||
return container["x"], container["y"], container["isOpen"]
|
||||
|
||||
def update_position(self, container_id: int, x: int, y: int, isOpen: bool) -> None:
|
||||
|
|
@ -76,17 +77,18 @@ class ContainerPositionManager:
|
|||
container_name = "activation" if container_id == 1 else "deactivation"
|
||||
self.positions[container_name] = {"x": x, "y": y, "isOpen": isOpen}
|
||||
|
||||
|
||||
container_manager = ContainerPositionManager()
|
||||
|
||||
|
||||
async def get_network_data() -> Dict[str, Dict]:
|
||||
async def get_network_data(translations: Translations = Depends(get_translations("api_v1"))) -> Dict[str, Dict]:
|
||||
"""Возвращает текущие сетевые данные в виде словаря.
|
||||
|
||||
Получает данные о сетевых интерфейсах, разделяет их на активные и неактивные,
|
||||
и возвращает их в формате Activation и Deactivation с учётом позиций и состояний
|
||||
контейнеров.
|
||||
|
||||
Args:
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
Dict[str, Dict]: Словарь с ключами "activation" и "deactivation", содержащими
|
||||
данные о соответствующих контейнерах.
|
||||
|
|
@ -94,7 +96,7 @@ async def get_network_data() -> Dict[str, Dict]:
|
|||
Raises:
|
||||
HTTPException: Если произошла ошибка при получении данных (status_code=500).
|
||||
"""
|
||||
logger.debug("Вызов get_network_data()")
|
||||
logger.debug("Calling get_network_data()")
|
||||
try:
|
||||
network_data = await network_service.get_network_interfaces_container()
|
||||
|
||||
|
|
@ -123,9 +125,8 @@ async def get_network_data() -> Dict[str, Dict]:
|
|||
"deactivation": deactivation.model_dump(),
|
||||
}
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка в get_network_data: {e}")
|
||||
raise HTTPException(status_code=500, detail="Не удалось получить сетевые данные")
|
||||
|
||||
logger.error(f"Error in get_network_data: {e}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Не удалось получить сетевые данные"))
|
||||
|
||||
def _split_interfaces(
|
||||
interfaces: List[Union[NetworkInterface, WiFiDevice]],
|
||||
|
|
@ -148,9 +149,8 @@ def _split_interfaces(
|
|||
deactivation_interfaces.append(interface)
|
||||
return activation_interfaces, deactivation_interfaces
|
||||
|
||||
|
||||
@router.post("/update_container_position")
|
||||
async def update_container_position(data: Dict[str, Union[int, bool]] = Body(...)):
|
||||
async def update_container_position(data: Dict[str, Union[int, bool]] = Body(...), translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Обновляет позицию и состояние контейнера.
|
||||
|
||||
Принимает данные о новом положении и состоянии контейнера и обновляет их
|
||||
|
|
@ -162,6 +162,7 @@ async def update_container_position(data: Dict[str, Union[int, bool]] = Body(...
|
|||
- "container_x" (int): Новая координата X.
|
||||
- "container_y" (int): Новая координата Y.
|
||||
- "isOpen" (bool): Новое состояние контейнера (открыт/закрыт).
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с ключами:
|
||||
|
|
@ -178,7 +179,7 @@ async def update_container_position(data: Dict[str, Union[int, bool]] = Body(...
|
|||
]
|
||||
if not all(key in data for key in required_keys):
|
||||
raise ValueError(
|
||||
"Отсутствуют обязательные поля: container_id, container_x, container_y, isOpen"
|
||||
translations.gettext("Отсутствуют обязательные поля: container_id, container_x, container_y, isOpen")
|
||||
)
|
||||
|
||||
container_id = data["container_id"]
|
||||
|
|
@ -190,24 +191,26 @@ async def update_container_position(data: Dict[str, Union[int, bool]] = Body(...
|
|||
or not isinstance(container_x, int)
|
||||
or not isinstance(container_y, int)
|
||||
or not isinstance(is_open, bool)):
|
||||
raise ValueError("Неверный тип данных в запросе")
|
||||
raise ValueError(translations.gettext("Неверный тип данных в запросе"))
|
||||
|
||||
container_manager.update_position(container_id, container_x, container_y, is_open)
|
||||
return {
|
||||
"success": True,
|
||||
"message": "Позиция и состояние контейнера обновлены"
|
||||
"message": translations.gettext("Позиция и состояние контейнера обновлены")
|
||||
}
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка обновления позиции контейнера: {e}")
|
||||
raise HTTPException(status_code=400, detail=f"Не удалось обновить положение контейнера: {str(e)}")
|
||||
|
||||
logger.error(f"Error updating container position: {e}")
|
||||
raise HTTPException(status_code=400, detail=translations.gettext("Не удалось обновить положение контейнера: {error}").format(error=str(e)))
|
||||
|
||||
@router.get("/get_hostname")
|
||||
async def get_hostname():
|
||||
async def get_hostname(translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Возвращает имя хоста системы.
|
||||
|
||||
Выполняет запрос к сервису для получения имени хоста.
|
||||
|
||||
Args:
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с ключом "hostname" и значением имени хоста.
|
||||
|
||||
|
|
@ -218,5 +221,5 @@ async def get_hostname():
|
|||
hostname_info = await network_service.get_hostname()
|
||||
return hostname_info
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка в get_hostname: {e}")
|
||||
raise HTTPException(status_code=500, detail="Не удалось получить имя хоста")
|
||||
logger.error(f"Error in get_hostname: {e}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Не удалось получить имя хоста"))
|
||||
|
|
@ -1,11 +1,12 @@
|
|||
# api/v1/settings.py
|
||||
import os
|
||||
import json
|
||||
import logging
|
||||
import tempfile
|
||||
from fastapi import APIRouter, HTTPException, Body
|
||||
from fastapi import APIRouter, HTTPException, Body, Depends
|
||||
from starlette.responses import FileResponse
|
||||
from services.settings_service import SettingsService
|
||||
from utils.i18n import get_translations, _
|
||||
from babel.support import Translations
|
||||
import shutil
|
||||
|
||||
# Настройка логирования
|
||||
|
|
@ -16,11 +17,14 @@ router = APIRouter()
|
|||
settings_service = SettingsService()
|
||||
|
||||
@router.post("/clear_subnet")
|
||||
def clear_subnet():
|
||||
def clear_subnet(translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Очищает данные узлов подсети, сохранённые в devices.json.
|
||||
|
||||
Удаляет все записи об узлах подсети, оставляя файл пустым.
|
||||
|
||||
Args:
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с сообщением об успешной очистке.
|
||||
|
||||
|
|
@ -30,16 +34,22 @@ def clear_subnet():
|
|||
try:
|
||||
success = settings_service.clear_subnet()
|
||||
if success:
|
||||
return {"message": "Данные узлов подсети успешно очищены"}
|
||||
return {"message": translations.gettext("Данные узлов подсети успешно очищены")}
|
||||
else:
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Не удалось очистить данные узлов подсети"))
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=f"Ошибка очистки: {str(e)}")
|
||||
logger.error(f"Error clearing subnet data: {str(e)}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Ошибка очистки: {error}").format(error=str(e)))
|
||||
|
||||
@router.post("/clear_wifi")
|
||||
def clear_wifi():
|
||||
def clear_wifi(translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Очищает данные точек доступа Wi-Fi, сохранённые в wifi.json.
|
||||
|
||||
Удаляет все записи о точках доступа, оставляя файл пустым.
|
||||
|
||||
Args:
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с сообщением об успешной очистке.
|
||||
|
||||
|
|
@ -49,16 +59,22 @@ def clear_wifi():
|
|||
try:
|
||||
success = settings_service.clear_wifi()
|
||||
if success:
|
||||
return {"message": "Данные точек доступа Wi-Fi успешно очищены"}
|
||||
return {"message": translations.gettext("Данные точек доступа Wi-Fi успешно очищены")}
|
||||
else:
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Не удалось очистить данные точек доступа Wi-Fi"))
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=f"Ошибка очистки: {str(e)}")
|
||||
logger.error(f"Error clearing wifi data: {str(e)}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Ошибка очистки: {error}").format(error=str(e)))
|
||||
|
||||
@router.post("/clear_all_data")
|
||||
def clear_all_data():
|
||||
def clear_all_data(translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Очищает данные узлов подсети и точек доступа Wi-Fi в обоих файлах.
|
||||
|
||||
Удаляет все записи из devices.json и wifi.json.
|
||||
|
||||
Args:
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с сообщением об успешной очистке.
|
||||
|
||||
|
|
@ -68,18 +84,24 @@ def clear_all_data():
|
|||
try:
|
||||
success = settings_service.clear_all_data()
|
||||
if success:
|
||||
return {"message": "Все сетевые данные успешно очищены"}
|
||||
return {"message": translations.gettext("Все сетевые данные успешно очищены")}
|
||||
else:
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Не удалось очистить все сетевые данные"))
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=f"Ошибка очистки: {str(e)}")
|
||||
logger.error(f"Error clearing all data: {str(e)}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Ошибка очистки: {error}").format(error=str(e)))
|
||||
|
||||
DEVICE_PATH = "storage/devices/devices.json"
|
||||
WIFI_PATH = "storage/wifi/wifi.json"
|
||||
|
||||
@router.get("/export_devices", response_class=FileResponse)
|
||||
async def export_devices():
|
||||
async def export_devices(translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""
|
||||
Экспортирует файл с данными устройств в формате JSON.
|
||||
|
||||
Args:
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
FileResponse: Файл devices.json для скачивания.
|
||||
|
||||
|
|
@ -87,9 +109,9 @@ async def export_devices():
|
|||
HTTPException: Если файл не найден (status_code=404).
|
||||
"""
|
||||
if not os.path.exists(DEVICE_PATH):
|
||||
logger.warning(f"Файл для экспорта не найден: {DEVICE_PATH}")
|
||||
raise HTTPException(status_code=404, detail="Файл не найден")
|
||||
logger.info(f"Экспорт файла: {DEVICE_PATH}")
|
||||
logger.warning(f"Export file not found: {DEVICE_PATH}")
|
||||
raise HTTPException(status_code=404, detail=translations.gettext("Файл не найден"))
|
||||
logger.info(f"Exporting file: {DEVICE_PATH}")
|
||||
return FileResponse(
|
||||
path=DEVICE_PATH,
|
||||
filename="devices.json",
|
||||
|
|
@ -97,20 +119,23 @@ async def export_devices():
|
|||
)
|
||||
|
||||
@router.get("/export_wifi", response_class=FileResponse)
|
||||
async def export_wifi():
|
||||
async def export_wifi(translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""
|
||||
Экспортирует файл с данными устройств в формате JSON.
|
||||
|
||||
Args:
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
FileResponse: Файл devices.json для скачивания.
|
||||
FileResponse: Файл wifi.json для скачивания.
|
||||
|
||||
Raises:
|
||||
HTTPException: Если файл не найден (status_code=404).
|
||||
"""
|
||||
if not os.path.exists(WIFI_PATH):
|
||||
logger.warning(f"Файл для экспорта не найден: {WIFI_PATH}")
|
||||
raise HTTPException(status_code=404, detail="Файл не найден")
|
||||
logger.info(f"Экспорт файла: {WIFI_PATH}")
|
||||
logger.warning(f"Export file not found: {WIFI_PATH}")
|
||||
raise HTTPException(status_code=404, detail=translations.gettext("Файл не найден"))
|
||||
logger.info(f"Exporting file: {WIFI_PATH}")
|
||||
return FileResponse(
|
||||
path=WIFI_PATH,
|
||||
filename="wifi.json",
|
||||
|
|
|
|||
|
|
@ -1,24 +1,25 @@
|
|||
from fastapi import APIRouter, HTTPException, Body
|
||||
from fastapi import APIRouter, HTTPException, Body, Depends
|
||||
from models.subnet import SubnetScanResult, SubnetScanRequest
|
||||
from models.wifi import WifiScanResult
|
||||
from models.request import NoteRequest, WifiConnectRequest, WifiDisconnectRequest
|
||||
from services.subnet_service import SubnetService
|
||||
from services.wifi_service import WifiService
|
||||
from services.connect_service import WifiConnectionService
|
||||
from utils.i18n import get_translations, _
|
||||
from babel.support import Translations
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
@router.post("/devices", response_model=SubnetScanResult)
|
||||
def scan_subnet(request: SubnetScanRequest = Body(...)):
|
||||
def scan_subnet(request: SubnetScanRequest = Body(...), translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Сканирует локальную подсеть и возвращает данные об устройствах.
|
||||
|
||||
Выполняет сканирование указанной подсети для обнаружения активных устройств
|
||||
с возможностью проверки указанных портов.
|
||||
|
||||
Args:
|
||||
subnet (str, optional): Адрес подсети в формате CIDR (например, "192.168.1.0/24").
|
||||
По умолчанию "192.168.1.0/24".
|
||||
ports (List[int], optional): Список портов для сканирования. По умолчанию None.
|
||||
request (SubnetScanRequest): Объект с полями subnet и ports.
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
SubnetScanResult: Объект с результатами сканирования (устройства и их данные).
|
||||
|
|
@ -32,16 +33,17 @@ def scan_subnet(request: SubnetScanRequest = Body(...)):
|
|||
result = subnet_service.scan_subnet(subnet)
|
||||
return result
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=f"Ошибка сканирования: {str(e)}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Ошибка сканирования: {error}").format(error=str(e)))
|
||||
|
||||
@router.get("/wifi", response_model=WifiScanResult)
|
||||
def scan_wifi(timeout: int = 30):
|
||||
def scan_wifi(timeout: int = 30, translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Сканирует Wi-Fi точки доступа и возвращает данные о них.
|
||||
|
||||
Выполняет сканирование доступных Wi-Fi сетей с указанным таймаутом.
|
||||
|
||||
Args:
|
||||
timeout (int, optional): Таймаут сканирования в секундах. По умолчанию 30.
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
WifiScanResult: Объект с результатами сканирования (список Wi-Fi сетей).
|
||||
|
|
@ -54,10 +56,10 @@ def scan_wifi(timeout: int = 30):
|
|||
result = wifi_service.scan_wifi(timeout=timeout)
|
||||
return result
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=f"Ошибка сканирования Wi-Fi: {str(e)}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Ошибка сканирования Wi-Fi: {error}").format(error=str(e)))
|
||||
|
||||
@router.post("/devices/{mac}/note")
|
||||
def add_note(mac: str, request: NoteRequest):
|
||||
def add_note(mac: str, request: NoteRequest, translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Добавляет заметку для устройства по MAC-адресу.
|
||||
|
||||
Привязывает заметку к устройству, идентифицируемому по MAC-адресу.
|
||||
|
|
@ -65,6 +67,7 @@ def add_note(mac: str, request: NoteRequest):
|
|||
Args:
|
||||
mac (str): MAC-адрес устройства.
|
||||
request (NoteRequest): Объект с полем "note" для текста заметки.
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с сообщением об успешном добавлении заметки.
|
||||
|
|
@ -77,18 +80,18 @@ def add_note(mac: str, request: NoteRequest):
|
|||
subnet_service = SubnetService()
|
||||
success, error_message = subnet_service.add_note(mac, request.note)
|
||||
if not success:
|
||||
if "не найдено" in error_message:
|
||||
raise HTTPException(status_code=404, detail=error_message)
|
||||
elif "уже существует" in error_message:
|
||||
raise HTTPException(status_code=400, detail=error_message)
|
||||
return {"message": f"Заметка '{request.note}' добавлена для устройства {mac}"}
|
||||
if "не найдено" in error_message.lower():
|
||||
raise HTTPException(status_code=404, detail=translations.gettext(error_message))
|
||||
elif "уже существует" in error_message.lower():
|
||||
raise HTTPException(status_code=400, detail=translations.gettext(error_message))
|
||||
return {"message": translations.gettext("Заметка '{note}' добавлена для устройства {mac}").format(note=request.note, mac=mac)}
|
||||
except HTTPException as e:
|
||||
raise e
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=f"Ошибка добавления заметки: {str(e)}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Ошибка добавления заметки: {error}").format(error=str(e)))
|
||||
|
||||
@router.put("/devices/{mac}/note")
|
||||
def update_note(mac: str, request: NoteRequest):
|
||||
def update_note(mac: str, request: NoteRequest, translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Обновляет заметку для устройства по MAC-адресу.
|
||||
|
||||
Изменяет существующую заметку, привязанную к устройству.
|
||||
|
|
@ -96,6 +99,7 @@ def update_note(mac: str, request: NoteRequest):
|
|||
Args:
|
||||
mac (str): MAC-адрес устройства.
|
||||
request (NoteRequest): Объект с полем "note" для нового текста заметки.
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с сообщением об успешном обновлении заметки.
|
||||
|
|
@ -108,24 +112,25 @@ def update_note(mac: str, request: NoteRequest):
|
|||
subnet_service = SubnetService()
|
||||
success, error_message = subnet_service.update_note(mac, request.note)
|
||||
if not success:
|
||||
if "не найдено" in error_message:
|
||||
raise HTTPException(status_code=404, detail=error_message)
|
||||
elif "не существует" in error_message:
|
||||
raise HTTPException(status_code=400, detail=error_message)
|
||||
return {"message": f"Заметка '{request.note}' обновлена для устройства {mac}"}
|
||||
if "не найдено" in error_message.lower():
|
||||
raise HTTPException(status_code=404, detail=translations.gettext(error_message))
|
||||
elif "не существует" in error_message.lower():
|
||||
raise HTTPException(status_code=400, detail=translations.gettext(error_message))
|
||||
return {"message": translations.gettext("Заметка '{note}' обновлена для устройства {mac}").format(note=request.note, mac=mac)}
|
||||
except HTTPException as e:
|
||||
raise e
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=f"Ошибка обновления заметки: {str(e)}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Ошибка обновления заметки: {error}").format(error=str(e)))
|
||||
|
||||
@router.delete("/devices/{mac}/note")
|
||||
def delete_note(mac: str):
|
||||
def delete_note(mac: str, translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Удаляет заметку для устройства по MAC-адресу.
|
||||
|
||||
Удаляет привязанную заметку для указанного устройства.
|
||||
|
||||
Args:
|
||||
mac (str): MAC-адрес устройства.
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с сообщением об успешном удалении заметки.
|
||||
|
|
@ -138,18 +143,18 @@ def delete_note(mac: str):
|
|||
subnet_service = SubnetService()
|
||||
success, error_message = subnet_service.delete_note(mac)
|
||||
if not success:
|
||||
if "не найдено" in error_message:
|
||||
raise HTTPException(status_code=404, detail=error_message)
|
||||
elif "не существует" in error_message:
|
||||
raise HTTPException(status_code=400, detail=error_message)
|
||||
return {"message": f"Заметка удалена для устройства {mac}"}
|
||||
if "не найдено" in error_message.lower():
|
||||
raise HTTPException(status_code=404, detail=translations.gettext(error_message))
|
||||
elif "не существует" in error_message.lower():
|
||||
raise HTTPException(status_code=400, detail=translations.gettext(error_message))
|
||||
return {"message": translations.gettext("Заметка удалена для устройства {mac}").format(mac=mac)}
|
||||
except HTTPException as e:
|
||||
raise e
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=f"Ошибка удаления заметки: {str(e)}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Ошибка удаления заметки: {error}").format(error=str(e)))
|
||||
|
||||
@router.post("/wifi/{ssid}/note")
|
||||
def add_wifi_note(ssid: str, request: NoteRequest):
|
||||
def add_wifi_note(ssid: str, request: NoteRequest, translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Добавляет заметку для сети Wi-Fi по SSID.
|
||||
|
||||
Привязывает заметку к Wi-Fi сети, идентифицируемой по SSID.
|
||||
|
|
@ -157,6 +162,7 @@ def add_wifi_note(ssid: str, request: NoteRequest):
|
|||
Args:
|
||||
ssid (str): Идентификатор Wi-Fi сети.
|
||||
request (NoteRequest): Объект с полем "note" для текста заметки.
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с сообщением об успешном добавлении заметки.
|
||||
|
|
@ -169,18 +175,18 @@ def add_wifi_note(ssid: str, request: NoteRequest):
|
|||
wifi_service = WifiService()
|
||||
success, error_message = wifi_service.add_note(ssid, request.note)
|
||||
if not success:
|
||||
if "не найдена" in error_message:
|
||||
raise HTTPException(status_code=404, detail=error_message)
|
||||
elif "уже существует" in error_message:
|
||||
raise HTTPException(status_code=400, detail=error_message)
|
||||
return {"message": f"Заметка '{request.note}' добавлена для сети с SSID {ssid}"}
|
||||
if "не найдена" in error_message.lower():
|
||||
raise HTTPException(status_code=404, detail=translations.gettext(error_message))
|
||||
elif "уже существует" in error_message.lower():
|
||||
raise HTTPException(status_code=400, detail=translations.gettext(error_message))
|
||||
return {"message": translations.gettext("Заметка '{note}' добавлена для сети с SSID {ssid}").format(note=request.note, ssid=ssid)}
|
||||
except HTTPException as e:
|
||||
raise e
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=f"Ошибка добавления заметки: {str(e)}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Ошибка добавления заметки: {error}").format(error=str(e)))
|
||||
|
||||
@router.put("/wifi/{ssid}/note")
|
||||
def update_wifi_note(ssid: str, request: NoteRequest):
|
||||
def update_wifi_note(ssid: str, request: NoteRequest, translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Обновляет заметку для сети Wi-Fi по SSID.
|
||||
|
||||
Изменяет существующую заметку, привязанную к Wi-Fi сети.
|
||||
|
|
@ -188,6 +194,7 @@ def update_wifi_note(ssid: str, request: NoteRequest):
|
|||
Args:
|
||||
ssid (str): Идентификатор Wi-Fi сети.
|
||||
request (NoteRequest): Объект с полем "note" для нового текста заметки.
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с сообщением об успешном обновлении заметки.
|
||||
|
|
@ -200,24 +207,25 @@ def update_wifi_note(ssid: str, request: NoteRequest):
|
|||
wifi_service = WifiService()
|
||||
success, error_message = wifi_service.update_note(ssid, request.note)
|
||||
if not success:
|
||||
if "не найдена" in error_message:
|
||||
raise HTTPException(status_code=404, detail=error_message)
|
||||
elif "не существует" in error_message:
|
||||
raise HTTPException(status_code=400, detail=error_message)
|
||||
return {"message": f"Заметка '{request.note}' обновлена для сети с SSID {ssid}"}
|
||||
if "не найдена" in error_message.lower():
|
||||
raise HTTPException(status_code=404, detail=translations.gettext(error_message))
|
||||
elif "не существует" in error_message.lower():
|
||||
raise HTTPException(status_code=400, detail=translations.gettext(error_message))
|
||||
return {"message": translations.gettext("Заметка '{note}' обновлена для сети с SSID {ssid}").format(note=request.note, ssid=ssid)}
|
||||
except HTTPException as e:
|
||||
raise e
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=f"Ошибка обновления заметки: {str(e)}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Ошибка обновления заметки: {error}").format(error=str(e)))
|
||||
|
||||
@router.delete("/wifi/{ssid}/note")
|
||||
def delete_wifi_note(ssid: str):
|
||||
def delete_wifi_note(ssid: str, translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Удаляет заметку для сети Wi-Fi по SSID.
|
||||
|
||||
Удаляет привязанную заметку для указанной Wi-Fi сети.
|
||||
|
||||
Args:
|
||||
ssid (str): Идентификатор Wi-Fi сети.
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с сообщением об успешном удалении заметки.
|
||||
|
|
@ -230,18 +238,18 @@ def delete_wifi_note(ssid: str):
|
|||
wifi_service = WifiService()
|
||||
success, error_message = wifi_service.delete_note(ssid)
|
||||
if not success:
|
||||
if "не найдена" in error_message:
|
||||
raise HTTPException(status_code=404, detail=error_message)
|
||||
elif "не существует" in error_message:
|
||||
raise HTTPException(status_code=400, detail=error_message)
|
||||
return {"message": f"Заметка удалена для сети с SSID {ssid}"}
|
||||
if "не найдена" in error_message.lower():
|
||||
raise HTTPException(status_code=404, detail=translations.gettext(error_message))
|
||||
elif "не существует" in error_message.lower():
|
||||
raise HTTPException(status_code=400, detail=translations.gettext(error_message))
|
||||
return {"message": translations.gettext("Заметка удалена для сети с SSID {ssid}").format(ssid=ssid)}
|
||||
except HTTPException as e:
|
||||
raise e
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=f"Ошибка удаления заметки: {str(e)}")
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Ошибка удаления заметки: {error}").format(error=str(e)))
|
||||
|
||||
@router.post("/wifi/connect")
|
||||
def connect_to_wifi(request: WifiConnectRequest):
|
||||
def connect_to_wifi(request: WifiConnectRequest, translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Подключается к Wi-Fi сети по SSID и паролю.
|
||||
|
||||
Выполняет подключение к указанной Wi-Fi сети с использованием предоставленных
|
||||
|
|
@ -249,6 +257,7 @@ def connect_to_wifi(request: WifiConnectRequest):
|
|||
|
||||
Args:
|
||||
request (WifiConnectRequest): Объект с полями "ssid" и "password".
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с сообщением об успешном подключении.
|
||||
|
|
@ -256,20 +265,26 @@ def connect_to_wifi(request: WifiConnectRequest):
|
|||
Raises:
|
||||
HTTPException: Если подключение не удалось (status_code=400).
|
||||
"""
|
||||
wifi_service = WifiConnectionService()
|
||||
success, message = wifi_service.connect(request.ssid, request.password)
|
||||
if not success:
|
||||
raise HTTPException(status_code=400, detail=message)
|
||||
return {"message": f"Успешно подключено к {request.ssid}"}
|
||||
try:
|
||||
wifi_service = WifiConnectionService()
|
||||
success, message = wifi_service.connect(request.ssid, request.password)
|
||||
if not success:
|
||||
raise HTTPException(status_code=400, detail=translations.gettext(message))
|
||||
return {"message": translations.gettext("Успешно подключено к {ssid}").format(ssid=request.ssid)}
|
||||
except HTTPException as e:
|
||||
raise e
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Ошибка подключения к Wi-Fi: {error}").format(error=str(e)))
|
||||
|
||||
@router.post("/wifi/disconnect")
|
||||
def disconnect_from_wifi(request: WifiDisconnectRequest):
|
||||
def disconnect_from_wifi(request: WifiDisconnectRequest, translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Отключается от Wi-Fi сети по указанному SSID.
|
||||
|
||||
Выполняет отключение от указанной Wi-Fi сети.
|
||||
|
||||
Args:
|
||||
request (WifiDisconnectRequest): Объект с полем "ssid".
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с сообщением об успешном отключении.
|
||||
|
|
@ -277,18 +292,26 @@ def disconnect_from_wifi(request: WifiDisconnectRequest):
|
|||
Raises:
|
||||
HTTPException: Если отключение не удалось (status_code=400).
|
||||
"""
|
||||
wifi_service = WifiConnectionService()
|
||||
success, message = wifi_service.disconnect(request.ssid)
|
||||
if not success:
|
||||
raise HTTPException(status_code=400, detail=message)
|
||||
return {"message": message or f"Успешно отключено от {request.ssid}"}
|
||||
try:
|
||||
wifi_service = WifiConnectionService()
|
||||
success, message = wifi_service.disconnect(request.ssid)
|
||||
if not success:
|
||||
raise HTTPException(status_code=400, detail=translations.gettext(message))
|
||||
return {"message": translations.gettext(message or "Успешно отключено от {ssid}").format(ssid=request.ssid)}
|
||||
except HTTPException as e:
|
||||
raise e
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Ошибка отключения от Wi-Fi: {error}").format(error=str(e)))
|
||||
|
||||
@router.get("/wifi/status")
|
||||
def get_wifi_status():
|
||||
def get_wifi_status(translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Проверяет статус текущего подключения к Wi-Fi.
|
||||
|
||||
Возвращает информацию о текущем состоянии подключения и SSID, если подключение активно.
|
||||
|
||||
Args:
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Returns:
|
||||
dict: Словарь с ключами:
|
||||
- "status" (str): Состояние ("connected" или другое).
|
||||
|
|
@ -297,8 +320,13 @@ def get_wifi_status():
|
|||
Raises:
|
||||
HTTPException: Если проверка статуса не удалась (status_code=400).
|
||||
"""
|
||||
wifi_service = WifiConnectionService()
|
||||
success, message = wifi_service.get_status()
|
||||
if not success:
|
||||
raise HTTPException(status_code=400, detail=message)
|
||||
return {"status": "connected", "ssid": message}
|
||||
try:
|
||||
wifi_service = WifiConnectionService()
|
||||
success, message = wifi_service.get_status()
|
||||
if not success:
|
||||
raise HTTPException(status_code=400, detail=translations.gettext(message))
|
||||
return {"status": translations.gettext("connected"), "ssid": message}
|
||||
except HTTPException as e:
|
||||
raise e
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=translations.gettext("Ошибка проверки статуса Wi-Fi: {error}").format(error=str(e)))
|
||||
|
|
@ -2,8 +2,10 @@ import asyncio
|
|||
import logging
|
||||
from typing import Set
|
||||
|
||||
from fastapi import APIRouter, WebSocket, WebSocketDisconnect
|
||||
from fastapi import APIRouter, WebSocket, WebSocketDisconnect, Depends
|
||||
from starlette.websockets import WebSocketState
|
||||
from utils.i18n import get_translations, _
|
||||
from babel.support import Translations
|
||||
|
||||
from .network import update_container_position, get_network_data
|
||||
|
||||
|
|
@ -70,15 +72,13 @@ class WebSocketManager:
|
|||
except WebSocketDisconnect:
|
||||
self.disconnect(connection)
|
||||
except Exception as e:
|
||||
logger.error(f"Сообщение об ошибке трансляции: {e}")
|
||||
logger.error(f"Broadcast error: {e}")
|
||||
self.disconnect(connection)
|
||||
|
||||
|
||||
manager = WebSocketManager()
|
||||
|
||||
|
||||
@router.websocket("/api/v1")
|
||||
async def websocket_endpoint(websocket: WebSocket):
|
||||
async def websocket_endpoint(websocket: WebSocket, translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Обработчик WebSocket-соединения.
|
||||
|
||||
Управляет соединением клиента, принимает данные для обновления позиций
|
||||
|
|
@ -86,6 +86,7 @@ async def websocket_endpoint(websocket: WebSocket):
|
|||
|
||||
Args:
|
||||
websocket (WebSocket): Объект WebSocket для обработки соединения.
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Raises:
|
||||
WebSocketDisconnect: Если клиент отключился.
|
||||
|
|
@ -95,26 +96,25 @@ async def websocket_endpoint(websocket: WebSocket):
|
|||
try:
|
||||
while True:
|
||||
data = await websocket.receive_json()
|
||||
logger.info(f"Полученные данные: {data}")
|
||||
logger.info(f"Received data: {data}")
|
||||
|
||||
if all(key in data for key in ["container_id", "container_x", "container_y"]):
|
||||
await update_container_position(data)
|
||||
|
||||
network_data = await get_network_data()
|
||||
logger.info(f"Текущие сетевые данные: {network_data}")
|
||||
logger.info(f"Current network data: {network_data}")
|
||||
|
||||
await manager.broadcast(network_data)
|
||||
await asyncio.sleep(2)
|
||||
except WebSocketDisconnect:
|
||||
logger.info("Клиент отключен")
|
||||
logger.info("Client disconnected")
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка в websocket_endpoint:{e}")
|
||||
logger.error(f"Error in websocket_endpoint: {e}")
|
||||
finally:
|
||||
manager.disconnect(websocket)
|
||||
if websocket.client_state != WebSocketState.DISCONNECTED:
|
||||
await websocket.close()
|
||||
|
||||
|
||||
@router.on_event("startup")
|
||||
async def startup_event():
|
||||
"""Запуск таска для трансляции сетевых данных при старте приложения.
|
||||
|
|
@ -126,13 +126,15 @@ async def startup_event():
|
|||
"""
|
||||
asyncio.create_task(broadcast_data())
|
||||
|
||||
|
||||
async def broadcast_data():
|
||||
async def broadcast_data(translations: Translations = Depends(get_translations("api_v1"))):
|
||||
"""Таск для периодической трансляции сетевых данных.
|
||||
|
||||
Выполняет сканирование сети и отправляет данные всем подключённым клиентам
|
||||
с интервалом в 2 секунды.
|
||||
|
||||
Args:
|
||||
translations: Объект переводов для текущего языка.
|
||||
|
||||
Raises:
|
||||
Exception: Если произошла ошибка при получении данных или трансляции.
|
||||
"""
|
||||
|
|
@ -140,13 +142,11 @@ async def broadcast_data():
|
|||
try:
|
||||
network_data = await get_network_data()
|
||||
if "activation" not in network_data:
|
||||
logger.error(
|
||||
"Ошибка: Ключ 'activation' отсутствует в полученных данных сети"
|
||||
)
|
||||
logger.error(translations.gettext("Ошибка: Ключ 'activation' отсутствует в полученных данных сети"))
|
||||
await asyncio.sleep(2)
|
||||
continue
|
||||
|
||||
await manager.broadcast(network_data)
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка в broadcast_data:{e}")
|
||||
logger.error(f"Error in broadcast_data: {e}")
|
||||
await asyncio.sleep(2)
|
||||
Loading…
Add table
Add a link
Reference in a new issue