gonets/SRV/api/v1/settings.py

import os
import logging
from fastapi import APIRouter, HTTPException, Depends, Response, Request
from starlette.responses import FileResponse
from services.settings_service import SettingsService
from utils.i18n import get_translations
from babel.support import Translations

logging.basicConfig(level=logging.DEBUG)
logger = logging.getLogger(__name__)

router = APIRouter()
settings_service = SettingsService()

@router.post("/clear_subnet")
def clear_subnet(translations: Translations = Depends(get_translations())):
    """Очищает данные узлов подсети, сохранённые в devices.json.

    Удаляет все записи об узлах подсети, оставляя файл пустым.

    Args:
        translations: Объект переводов для текущего языка.

    Returns:
        dict: Словарь с сообщением об успешной очистке.

    Raises:
        HTTPException: Если произошла ошибка очистки (status_code=500).
    """
    try:
        success = settings_service.clear_subnet()
        if success:
            return {"message": translations.gettext("Данные узлов подсети успешно очищены")}
        else:
            raise HTTPException(status_code=500, detail=translations.gettext("Не удалось очистить данные узлов подсети"))
    except Exception as 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(translations: Translations = Depends(get_translations())):
    """Очищает данные точек доступа Wi-Fi, сохранённые в wifi.json.

    Удаляет все записи о точках доступа, оставляя файл пустым.

    Args:
        translations: Объект переводов для текущего языка.

    Returns:
        dict: Словарь с сообщением об успешной очистке.

    Raises:
        HTTPException: Если произошла ошибка очистки (status_code=500).
    """
    try:
        success = settings_service.clear_wifi()
        if success:
            return {"message": translations.gettext("Данные точек доступа Wi-Fi успешно очищены")}
        else:
            raise HTTPException(status_code=500, detail=translations.gettext("Не удалось очистить данные точек доступа Wi-Fi"))
    except Exception as 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(translations: Translations = Depends(get_translations())):
    """Очищает данные узлов подсети и точек доступа Wi-Fi в обоих файлах.

    Удаляет все записи из devices.json и wifi.json.

    Args:
        translations: Объект переводов для текущего языка.

    Returns:
        dict: Словарь с сообщением об успешной очистке.

    Raises:
        HTTPException: Если произошла ошибка очистки (status_code=500).
    """
    try:
        success = settings_service.clear_all_data()
        if success:
            return {"message": translations.gettext("Все сетевые данные успешно очищены")}
        else:
            raise HTTPException(status_code=500, detail=translations.gettext("Не удалось очистить все сетевые данные"))
    except Exception as 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(translations: Translations = Depends(get_translations())):
    """
    Экспортирует файл с данными устройств в формате JSON.

    Args:
        translations: Объект переводов для текущего языка.

    Returns:
        FileResponse: Файл devices.json для скачивания.

    Raises:
        HTTPException: Если файл не найден (status_code=404).
    """
    if not os.path.exists(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",
        media_type="application/json"
    )

@router.get("/export_wifi", response_class=FileResponse)
async def export_wifi(translations: Translations = Depends(get_translations())):
    """
    Экспортирует файл с данными устройств в формате JSON.

    Args:
        translations: Объект переводов для текущего языка.

    Returns:
        FileResponse: Файл wifi.json для скачивания.

    Raises:
        HTTPException: Если файл не найден (status_code=404).
    """
    if not os.path.exists(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",
        media_type="application/json"
    )

@router.post("/set_language")
def set_language(locale: str, response: Response, translations: Translations = Depends(get_translations())):
    """
    Устанавливает куку locale для смены языка пользователя.

    Args:
        locale (str): Используемый язык.
        response (Response): Запрос.
        translations: Объект переводов текущего языка.

    Returns:
        dict: Сообщение об успешной установке языка.
    """
    lc_messages_path = os.path.join(os.path.dirname(__file__), '..', '..', 'locales', locale, 'LC_MESSAGES')
    if not os.path.isdir(lc_messages_path):
        raise HTTPException(
            status_code=400,
            detail=translations.gettext("Язык '{locale}' не поддерживается.").format(locale=locale)
        )
    has_mo_or_po = any(f.endswith('.mo') or f.endswith('.po') for f in os.listdir(lc_messages_path))
    if not has_mo_or_po:
        raise HTTPException(
            status_code=400,
            detail=translations.gettext("Язык '{locale}' не поддерживается (нет файлов переводов).").format(locale=locale)
        )
    response.set_cookie(key="locale", value=locale, path="/", httponly=True)
    return {"message": translations.gettext("Язык успешно установлен: {locale}").format(locale=locale)}