HybrydMain/gonets
3
2
Fork 0
Code Issues 5 Pull requests Projects 1 Releases 3 Packages Wiki Activity Actions

Форматирование всего серверного кода по стандартам (black)

Browse source
This commit is contained in:
Галанский Сергей Александрович 2025-06-12 16:32:05 +05:00
parent e61ffd793c
commit 06a5a758a3
28 changed files with 611 additions and 305 deletions
Download patch file Download diff file Expand all files Collapse all files

2
SRV/api/router.py
View file

@ -9,4 +9,4 @@ router.include_router(ip.router, prefix="/ip", tags=["IP"])
router.include_router(network.router, prefix="/network", tags=["Network"])
router.include_router(auth.router, prefix="/auth", tags=["Auth"])
router.include_router(subnet.router, prefix="/subnet", tags=["Subnet"])
router.include_router(subnet.router, prefix="/subnet", tags=["Subnet"])

30
SRV/api/v1/auth.py
View file

@ -26,7 +26,9 @@ async def register(user: UserCreate):
raise e
except Exception as e:
logger.error(f"Ошибка в register: {type(e).__name__} - {str(e)}")
raise HTTPException(status_code=500, detail="Не удалось зарегистрировать пользователя")
raise HTTPException(
status_code=500, detail="Не удалось зарегистрировать пользователя"
)
@router.post("/login", response_model=Token)
@ -36,13 +38,17 @@ async def login(user: UserLogin, request: Request):
client_id = request.client.host
logger.debug(f"Попытка авторизации для client_id {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 {client_id}, session_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="Не удалось авторизовать пользователя")
raise HTTPException(
status_code=500, detail="Не удалось авторизовать пользователя"
)
@router.get("/me")
@ -55,13 +61,15 @@ async def get_current_user(request: Request):
return {
"user": result["user"],
"access_token": result["access_token"],
"token_type": "bearer"
"token_type": "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="Не удалось получить данные пользователя")
raise HTTPException(
status_code=500, detail="Не удалось получить данные пользователя"
)
@router.get("/logout")
@ -90,7 +98,7 @@ async def check_users_file():
return {
"exists": False,
"has_users": False,
"message": "Файл users.json не существует"
"message": "Файл users.json не существует",
}
data = storage._read_data()
@ -100,15 +108,17 @@ async def check_users_file():
return {
"exists": True,
"has_users": True,
"message": "Файл users.json существует и содержит пользователей"
"message": "Файл users.json существует и содержит пользователей",
}
else:
logger.info(f"Файл {file_path} существует, но не содержит пользователей")
return {
"exists": True,
"has_users": False,
"message": "Файл users.json существует, но не содержит пользователей"
"message": "Файл 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"Ошибка при проверке файла users.json: {type(e).__name__} - {str(e)}"
)
raise HTTPException(status_code=500, detail="Ошибка при проверке файла")

5
SRV/api/v1/firewalld.py
View file

@ -123,10 +123,7 @@ async def get_zone(zone: str = Path(..., description="Zone name")) -> FirewallZo
try:
zone_info = firewalld.get_zone_settings(zone)
return FirewallZone(
name=zone,
target=FirewallTarget.DEFAULT,
interfaces=[],
**zone_info
name=zone, target=FirewallTarget.DEFAULT, interfaces=[], **zone_info
)
except ValueError:
raise HTTPException(

31
SRV/api/v1/ip.py
View file

@ -5,33 +5,48 @@ from services.ip_service import IPService
router = APIRouter()
ip_service = IPService()
@router.post("/check_ip")
async def check_ip_config(data: Dict[str, Any] = Body(...)):
""" Проверяет доступность IP-адреса """
"""Проверяет доступность IP-адреса"""
try:
result = await ip_service.check_ip_config(data)
return result
except Exception as e:
raise HTTPException(status_code=500, detail=f"Ошибка проверки IP конфигурации: {str(e)}")
raise HTTPException(
status_code=500, detail=f"Ошибка проверки IP конфигурации: {str(e)}"
)
@router.post("/apply_ip")
async def apply_ip_config(data: Dict[str, Any] = Body(...)):
""" Применяет проверенные IP-настройки """
"""Применяет проверенные IP-настройки"""
try:
await ip_service.apply_ip_config(data)
return {"success": True, "message": f"Настройки сетевого профиля для {data['connection_id']} применены"}
return {
"success": True,
"message": f"Настройки сетевого профиля для {data['connection_id']} применены",
}
except ValueError as e:
raise HTTPException(status_code=404, detail=str(e))
except Exception as e:
raise HTTPException(status_code=500, detail=f"Ошибка проверки IP конфигурации: {str(e)}")
raise HTTPException(
status_code=500, detail=f"Ошибка проверки IP конфигурации: {str(e)}"
)
@router.post("/save_ip")
async def save_ip_config(data: Dict[str, Any] = Body(...)):
""" Применяет проверенные IP-настройки (Нужен для того, чтобы откатить настройки сетевого профиля обратно) """
"""Применяет проверенные IP-настройки (Нужен для того, чтобы откатить настройки сетевого профиля обратно)"""
try:
await ip_service.apply_ip_config(data)
return {"success": True, "message": f"Настройки сетевого профиля для {data['connection_id']} применены"}
return {
"success": True,
"message": f"Настройки сетевого профиля для {data['connection_id']} применены",
}
except ValueError as e:
raise HTTPException(status_code=404, detail=str(e))
except Exception as e:
raise HTTPException(status_code=500, detail=f"Ошибка проверки IP конфигурации: {str(e)}")
raise HTTPException(
status_code=500, detail=f"Ошибка проверки IP конфигурации: {str(e)}"
)

73
SRV/api/v1/network.py
View file

@ -3,7 +3,13 @@ import logging
from typing import Dict, Tuple, List, Union
from services.network_service import NetworkService
from models.network import NetworkInterfacesContainer, Activation, Deactivation, NetworkInterface, WiFiDevice
from models.network import (
NetworkInterfacesContainer,
Activation,
Deactivation,
NetworkInterface,
WiFiDevice,
)
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
@ -19,19 +25,19 @@ class ContainerPositionManager:
def __init__(self):
self.positions: Dict[str, Dict[str, Union[int, bool]]] = {
'activation': {'x': 0, 'y': 0, 'isOpen': False},
'deactivation': {'x': 0, 'y': 0, 'isOpen': False}
"activation": {"x": 0, "y": 0, "isOpen": False},
"deactivation": {"x": 0, "y": 0, "isOpen": False},
}
def get_position_and_state(self, container_name: str) -> Tuple[int, int, bool]:
container = self.positions.get(container_name)
if not container:
raise ValueError(f"Неверное имя контейнера: {container_name}")
return container['x'], container['y'], container['isOpen']
return container["x"], container["y"], container["isOpen"]
def update_position(self, container_id: int, x: int, y: int, isOpen: bool) -> None:
container_name = 'activation' if container_id == 1 else 'deactivation'
self.positions[container_name] = {'x': x, 'y': y, 'isOpen': isOpen}
container_name = "activation" if container_id == 1 else "deactivation"
self.positions[container_name] = {"x": x, "y": y, "isOpen": isOpen}
container_manager = ContainerPositionManager()
@ -45,44 +51,55 @@ async def get_network_data() -> Dict[str, Dict]:
try:
network_data = await network_service.get_network_interfaces_container()
activation_interfaces, deactivation_interfaces = _split_interfaces(network_data.interfaces)
activation_interfaces, deactivation_interfaces = _split_interfaces(
network_data.interfaces
)
activation_x, activation_y, activation_is_open = container_manager.get_position_and_state('activation')
deactivation_x, deactivation_y, deactivation_is_open = container_manager.get_position_and_state('deactivation')
activation_x, activation_y, activation_is_open = (
container_manager.get_position_and_state("activation")
)
deactivation_x, deactivation_y, deactivation_is_open = (
container_manager.get_position_and_state("deactivation")
)
activation = Activation(
id=1,
interfaces=activation_interfaces,
container_x=activation_x,
container_y=activation_y,
isOpen=activation_is_open
isOpen=activation_is_open,
)
deactivation = Deactivation(
id=2,
interfaces=deactivation_interfaces,
container_x=deactivation_x,
container_y=deactivation_y,
isOpen=deactivation_is_open
isOpen=deactivation_is_open,
)
return {
"activation": activation.model_dump(),
"deactivation": deactivation.model_dump()
"deactivation": deactivation.model_dump(),
}
except Exception as e:
logger.error(f"Ошибка в get_network_data: {e}")
raise HTTPException(status_code=500, detail="Не удалось получить сетевые данные")
raise HTTPException(
status_code=500, detail="Не удалось получить сетевые данные"
)
def _split_interfaces(interfaces: List[Union[NetworkInterface, WiFiDevice]]) -> Tuple[
List[Union[NetworkInterface, WiFiDevice]], List[Union[NetworkInterface, WiFiDevice]]]:
def _split_interfaces(
interfaces: List[Union[NetworkInterface, WiFiDevice]],
) -> Tuple[
List[Union[NetworkInterface, WiFiDevice]], List[Union[NetworkInterface, WiFiDevice]]
]:
"""
Разделяет интерфейсы на активные и неактивные
"""
activation_interfaces = []
deactivation_interfaces = []
for interface in interfaces:
if interface.device_state == '100':
if interface.device_state == "100":
activation_interfaces.append(interface)
else:
deactivation_interfaces.append(interface)
@ -98,7 +115,9 @@ async def update_container_position(data: Dict[str, Union[int, bool]] = Body(...
required_keys = ["container_id", "container_x", "container_y", "isOpen"]
if not all(key in data for key in required_keys):
raise ValueError("Отсутствуют обязательные поля: container_id, container_x, container_y, isOpen")
raise ValueError(
"Отсутствуют обязательные поля: container_id, container_x, container_y, isOpen"
)
container_id = data["container_id"]
container_x = data["container_x"]
@ -106,16 +125,24 @@ async def update_container_position(data: Dict[str, Union[int, bool]] = Body(...
is_open = data["isOpen"]
# Проверка типов
if not isinstance(container_id, int) or not isinstance(container_x, int) or not isinstance(container_y,
int) or not isinstance(
is_open, bool):
if (
not isinstance(container_id, int)
or not isinstance(container_x, int)
or not isinstance(container_y, int)
or not isinstance(is_open, bool)
):
raise ValueError("Неверный тип данных в запросе")
container_manager.update_position(container_id, container_x, container_y, is_open)
container_manager.update_position(
container_id, container_x, container_y, is_open
)
return {"success": True, "message": "Позиция и состояние контейнера обновлены"}
except Exception as e:
logger.error(f"Ошибка обновления позиции контейнера: {e}")
raise HTTPException(status_code=400, detail=f"Не удалось обновить положение контейнера: {str(e)}")
raise HTTPException(
status_code=400,
detail=f"Не удалось обновить положение контейнера: {str(e)}",
)
@router.get("/get_hostname")
@ -125,4 +152,4 @@ async def get_hostname():
return hostname_info
except Exception as e:
logger.error(f"Ошибка в get_hostname: {e}")
raise HTTPException(status_code=500, detail="Не удалось получить имя хоста")
raise HTTPException(status_code=500, detail="Не удалось получить имя хоста")

21
SRV/api/v1/subnet.py
View file

@ -8,6 +8,7 @@ from services.wifi_service import WifiService
router = APIRouter()
@router.get("/scan", response_model=SubnetScanResult)
def scan_subnet(subnet: str = "192.168.1.0/24", ports: List[int] = None):
"""Сканирует локальную подсеть и возвращает данные об устройствах"""
@ -18,6 +19,7 @@ def scan_subnet(subnet: str = "192.168.1.0/24", ports: List[int] = None):
except Exception as e:
raise HTTPException(status_code=500, detail=f"Ошибка сканирования: {str(e)}")
@router.get("/wifi", response_model=WifiScanResult)
def scan_wifi(timeout: int = 30):
"""Сканирует Wi-Fi точки доступа и возвращает данные о них"""
@ -26,7 +28,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=f"Ошибка сканирования Wi-Fi: {str(e)}"
)
@router.post("/devices/{mac}/note")
def add_note(mac: str, request: NoteRequest):
@ -43,7 +48,10 @@ def add_note(mac: str, request: NoteRequest):
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=f"Ошибка добавления заметки: {str(e)}"
)
@router.put("/devices/{mac}/note")
def update_note(mac: str, request: NoteRequest):
@ -60,7 +68,10 @@ def update_note(mac: str, request: NoteRequest):
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=f"Ошибка обновления заметки: {str(e)}"
)
@router.delete("/devices/{mac}/note")
def delete_note(mac: str):
@ -77,4 +88,6 @@ def delete_note(mac: str):
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=f"Ошибка удаления заметки: {str(e)}"
)

34
SRV/api/v1/ufw.py
View file

@ -7,9 +7,11 @@ from services.ufw_service import UFWService
router = APIRouter()
class RuleNumbers(BaseModel):
numbers: List[int] = Field(..., description="Список номеров правил для удаления")
@router.get("/rules", response_model=UFWRulesResponse)
async def get_ufw_rules():
"""
@ -21,6 +23,7 @@ async def get_ufw_rules():
except Exception as e:
raise HTTPException(status_code=500, detail=f"Error: {str(e)}")
@router.get("/status", response_model=UFWStatus)
async def get_ufw_status():
"""
@ -32,6 +35,7 @@ async def get_ufw_status():
except Exception as e:
raise HTTPException(status_code=500, detail=f"Error: {str(e)}")
@router.post("/enable")
async def enable_ufw():
"""
@ -47,6 +51,7 @@ async def enable_ufw():
except Exception as e:
raise HTTPException(status_code=500, detail=f"Error: {str(e)}")
@router.post("/disable")
async def disable_ufw():
"""
@ -62,6 +67,7 @@ async def disable_ufw():
except Exception as e:
raise HTTPException(status_code=500, detail=f"Error: {str(e)}")
@router.post("/rules/in")
async def add_ufw_in_rules(rules: List[UFWRule]):
"""
@ -74,9 +80,9 @@ async def add_ufw_in_rules(rules: List[UFWRule]):
if not status.enabled:
raise HTTPException(
status_code=400,
detail="UFW отключен. Включите UFW перед добавлением правил."
detail="UFW отключен. Включите UFW перед добавлением правил.",
)
result = ufw_service.add_rules(rules)
if result["success"]:
successful_rules = len([r for r in result["results"] if r["success"]])
@ -84,13 +90,17 @@ async def add_ufw_in_rules(rules: List[UFWRule]):
return {
"status": "success",
"message": f"Добавлено {successful_rules} из {total_rules} правил",
"details": result
"details": result,
}
else:
raise HTTPException(status_code=500, detail=result.get("error", "Не удалось добавить правила"))
raise HTTPException(
status_code=500,
detail=result.get("error", "Не удалось добавить правила"),
)
except Exception as e:
raise HTTPException(status_code=500, detail=f"Error: {str(e)}")
@router.post("/rules/out")
async def add_ufw_out_rules(rules: List[UFWRule]):
"""
@ -103,17 +113,23 @@ async def add_ufw_out_rules(rules: List[UFWRule]):
if not status.enabled:
raise HTTPException(
status_code=400,
detail="UFW отключен. Включите UFW перед добавлением правил."
detail="UFW отключен. Включите UFW перед добавлением правил.",
)
success = ufw_service.add_ufw_out_rules(rules)
if success:
return {"status": "success", "message": "Исходящие правила UFW успешно добавлены"}
return {
"status": "success",
"message": "Исходящие правила UFW успешно добавлены",
}
else:
raise HTTPException(status_code=500, detail="Не удалось добавить исходящие правила UFW")
raise HTTPException(
status_code=500, detail="Не удалось добавить исходящие правила UFW"
)
except Exception as e:
raise HTTPException(status_code=500, detail=f"Error: {str(e)}")
@router.delete("/rules")
async def delete_ufw_rules(rule_numbers: RuleNumbers):
"""
@ -126,7 +142,7 @@ async def delete_ufw_rules(rule_numbers: RuleNumbers):
return {
"status": "success",
"message": f"Удалено {result['successful']} из {result['total']} правил",
"details": result
"details": result,
}
else:
raise HTTPException(status_code=500, detail=result["error"])

28
SRV/api/v1/websocket.py
View file

@ -12,25 +12,27 @@ router = APIRouter()
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
class WebSocketManager:
"""
Класс для управления подключениями к WebSocket-серверу
"""
def __init__(self):
""" Инициализация менеджера подключений к WebSocket-серверу """
"""Инициализация менеджера подключений к WebSocket-серверу"""
self.active_connections: Set[WebSocket] = set()
async def connect(self, websocket: WebSocket):
""" Подключение клиента к WebSocket-серверу """
"""Подключение клиента к WebSocket-серверу"""
await websocket.accept()
self.active_connections.add(websocket)
def disconnect(self, websocket: WebSocket):
""" Отключение клиента от WebSocket-серверу """
"""Отключение клиента от WebSocket-серверу"""
self.active_connections.remove(websocket)
async def broadcast(self, message: dict):
""" Трансляция сообщений всем подключенным клиентам """
"""Трансляция сообщений всем подключенным клиентам"""
for connection in list(self.active_connections):
try:
if connection.client_state == WebSocketState.CONNECTED:
@ -41,18 +43,22 @@ class WebSocketManager:
logger.error(f"Сообщение об ошибке трансляции: {e}")
self.disconnect(connection)
manager = WebSocketManager()
@router.websocket("/api/v1")
async def websocket_endpoint(websocket: WebSocket):
""" Обработчик WebSocket-соединения """
"""Обработчик WebSocket-соединения"""
await manager.connect(websocket)
try:
while True:
data = await websocket.receive_json()
logger.info(f"Полученные данные: {data}")
if all(key in data for key in ["container_id", "container_x", "container_y"]):
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()
@ -69,18 +75,22 @@ async def websocket_endpoint(websocket: WebSocket):
if websocket.client_state != WebSocketState.DISCONNECTED:
await websocket.close()
@router.on_event("startup")
async def startup_event():
""" Запуск таска для трансляции сетевых данных """
"""Запуск таска для трансляции сетевых данных"""
asyncio.create_task(broadcast_data())
async def broadcast_data():
""" Таск для трансляции сетевых данных """
"""Таск для трансляции сетевых данных"""
while True:
try:
network_data = await get_network_data()
if "activation" not in network_data:
logger.error("Ошибка: Ключ 'activation' отсутствует в полученных данных сети")
logger.error(
"Ошибка: Ключ 'activation' отсутствует в полученных данных сети"
)
await asyncio.sleep(2)
continue