Интернационализация для всех сервисов сервера
This commit is contained in:
parent
7491ce455d
commit
ba8cea01e7
13 changed files with 1011 additions and 206 deletions
|
|
@ -9,6 +9,9 @@ import pywifi
|
|||
from models.wifi import WifiAccessPoint, WifiScanResult
|
||||
from pathlib import Path
|
||||
import json
|
||||
from utils.i18n import get_translations, _
|
||||
from babel.support import Translations
|
||||
from fastapi import Depends
|
||||
|
||||
logging.basicConfig(
|
||||
level=logging.DEBUG,
|
||||
|
|
@ -22,7 +25,7 @@ class WifiStorage:
|
|||
в JSON-файле на диске.
|
||||
"""
|
||||
|
||||
def __init__(self, storage_path: str = "storage/wifi/wifi.json"):
|
||||
def __init__(self, storage_path: str = "storage/wifi/wifi.json", translations: Translations = Depends(get_translations("services"))):
|
||||
"""Инициализирует объект хранилища Wi-Fi с указанным путём к файлу.
|
||||
|
||||
Args:
|
||||
|
|
@ -33,6 +36,7 @@ class WifiStorage:
|
|||
"""
|
||||
self.storage_file = Path(storage_path)
|
||||
self.storage_file.parent.mkdir(parents=True, exist_ok=True)
|
||||
self.translations = translations
|
||||
|
||||
def load_notes(self) -> Dict[str, Optional[str]]:
|
||||
"""Загружает заметки по SSID из файла хранения.
|
||||
|
|
@ -51,7 +55,7 @@ class WifiStorage:
|
|||
return {item["ssid"]: item.get("note") for item in data.get("networks", [])}
|
||||
return {}
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка при загрузке заметок: {str(e)}")
|
||||
logger.error(f"Error loading notes: {str(e)}")
|
||||
return {}
|
||||
|
||||
def save_notes(self, notes: Dict[str, Optional[str]]) -> None:
|
||||
|
|
@ -68,9 +72,10 @@ class WifiStorage:
|
|||
data = {"networks": [{"ssid": ssid, "note": note} for ssid, note in notes.items()]}
|
||||
with open(self.storage_file, "w", encoding="utf-8") as f:
|
||||
json.dump(data, f, indent=2, ensure_ascii=False)
|
||||
logger.info(f"Сохранено {len(notes)} заметок в {self.storage_file}")
|
||||
logger.info(f"Saved {len(notes)} notes to {self.storage_file}")
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка при сохранении заметок: {str(e)}")
|
||||
logger.error(f"Error saving notes: {str(e)}")
|
||||
raise Exception(self.translations.gettext("Ошибка при сохранении заметок: {error}").format(error=str(e)))
|
||||
|
||||
class WifiService:
|
||||
"""Сервис для сканирования текущих Wi-Fi точек доступа.
|
||||
|
|
@ -80,7 +85,7 @@ class WifiService:
|
|||
а также управления заметками.
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
def __init__(self, translations: Translations = Depends(get_translations("services"))):
|
||||
"""Инициализирует сервис Wi-Fi с настройкой интерфейса и хранилища.
|
||||
|
||||
Инициализирует объект MacLookup для определения вендоров по MAC-адресам,
|
||||
|
|
@ -90,6 +95,7 @@ class WifiService:
|
|||
self.wifi = pywifi.PyWiFi()
|
||||
self.interface = self._get_wifi_interface()
|
||||
self.storage = WifiStorage()
|
||||
self.translations = translations
|
||||
|
||||
def _get_wifi_interface(self) -> Optional[object]:
|
||||
"""Автоматически определяет активный Wi-Fi-интерфейс.
|
||||
|
|
@ -103,35 +109,35 @@ class WifiService:
|
|||
try:
|
||||
interfaces = self.wifi.interfaces()
|
||||
if not interfaces:
|
||||
logger.error("Не найдено сетевых интерфейсов")
|
||||
logger.error(self.translations.gettext("Не найдено сетевых интерфейсов"))
|
||||
return None
|
||||
|
||||
logger.debug(
|
||||
f"Доступные интерфейсы: {[iface.name() for iface in interfaces]}"
|
||||
f"Available interfaces: {[iface.name() for iface in interfaces]}"
|
||||
)
|
||||
|
||||
for iface in interfaces:
|
||||
iface_name = iface.name()
|
||||
if "p2p" in iface_name.lower():
|
||||
logger.debug(f"Пропущен интерфейс {iface_name} (P2P)")
|
||||
logger.debug(f"Skipped interface {iface_name} (P2P)")
|
||||
continue
|
||||
try:
|
||||
iface.scan()
|
||||
time.sleep(1)
|
||||
results = iface.scan_results()
|
||||
if results:
|
||||
logger.info(f"Выбран Wi-Fi-интерфейс: {iface_name}")
|
||||
logger.info(f"Selected Wi-Fi interface: {iface_name}")
|
||||
return iface
|
||||
logger.debug(
|
||||
f"Интерфейс {iface_name} не вернул результатов сканирования"
|
||||
f"Interface {iface_name} returned no scan results"
|
||||
)
|
||||
except Exception as e:
|
||||
logger.debug(f"Интерфейс {iface_name} неактивен: {str(e)}")
|
||||
logger.debug(f"Interface {iface_name} inactive: {str(e)}")
|
||||
|
||||
logger.error("Не найден активный Wi-Fi-интерфейс")
|
||||
logger.error(self.translations.gettext("Не найден активный Wi-Fi-интерфейс"))
|
||||
return None
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка при определении Wi-Fi-интерфейса: {str(e)}")
|
||||
logger.error(f"Error determining Wi-Fi interface: {str(e)}")
|
||||
return None
|
||||
|
||||
@lru_cache(maxsize=1000)
|
||||
|
|
@ -151,8 +157,8 @@ class WifiService:
|
|||
return self.mac_lookup.lookup(mac)
|
||||
except Exception as e:
|
||||
logger.warning(
|
||||
f"Не удалось определить производителя для MAC {mac}: {str(e)}")
|
||||
return "Unknown"
|
||||
f"Failed to determine vendor for MAC {mac}: {str(e)}")
|
||||
return self.translations.gettext("Неизвестный")
|
||||
|
||||
def _get_encryption_from_iw(self, ap_info: Dict) -> str:
|
||||
"""Определяет тип шифрования из данных, полученных через iw.
|
||||
|
|
@ -165,16 +171,16 @@ class WifiService:
|
|||
"""
|
||||
security = ap_info.get("security", "")
|
||||
logger.debug(
|
||||
f"Шифрование для BSSID {ap_info.get('bssid')}: {security}")
|
||||
f"Encryption for BSSID {ap_info.get('bssid')}: {security}")
|
||||
if "WPA3" in security:
|
||||
return "WPA3"
|
||||
return self.translations.gettext("WPA3")
|
||||
elif "WPA2" in security or "RSN" in security:
|
||||
return "WPA2"
|
||||
return self.translations.gettext("WPA2")
|
||||
elif "WPA" in security:
|
||||
return "WPA"
|
||||
return self.translations.gettext("WPA")
|
||||
elif "WEP" in security:
|
||||
return "WEP"
|
||||
return "Open"
|
||||
return self.translations.gettext("WEP")
|
||||
return self.translations.gettext("Открытая")
|
||||
|
||||
def _get_channel(self, frequency: Optional[int]) -> Optional[int]:
|
||||
"""Преобразует частоту в номер канала.
|
||||
|
|
@ -192,11 +198,11 @@ class WifiService:
|
|||
return (frequency - 2412) // 5 + 1
|
||||
elif 5180 <= frequency <= 5825:
|
||||
return (frequency - 5180) // 5 + 36
|
||||
logger.debug(f"Частота {frequency} вне диапазона 2.4/5 GHz")
|
||||
logger.debug(f"Frequency {frequency} outside 2.4/5 GHz range")
|
||||
return None
|
||||
except Exception as e:
|
||||
logger.debug(
|
||||
f"Ошибка при определении канала для частоты {frequency}: {str(e)}"
|
||||
f"Error determining channel for frequency {frequency}: {str(e)}"
|
||||
)
|
||||
return None
|
||||
|
||||
|
|
@ -213,14 +219,14 @@ class WifiService:
|
|||
return None
|
||||
try:
|
||||
if 2412 <= frequency <= 2472:
|
||||
return "2.4 GHz"
|
||||
return self.translations.gettext("2.4 ГГц")
|
||||
elif 5180 <= frequency <= 5825:
|
||||
return "5 GHz"
|
||||
logger.debug(f"Частота {frequency} вне диапазона 2.4/5 GHz")
|
||||
return self.translations.gettext("5 ГГц")
|
||||
logger.debug(f"Frequency {frequency} outside 2.4/5 GHz range")
|
||||
return None
|
||||
except Exception as e:
|
||||
logger.debug(
|
||||
f"Ошибка при определении диапазона для частоты {frequency}: {str(e)}"
|
||||
f"Error determining band for frequency {frequency}: {str(e)}"
|
||||
)
|
||||
return None
|
||||
|
||||
|
|
@ -238,20 +244,20 @@ class WifiService:
|
|||
"""
|
||||
try:
|
||||
if not self.interface:
|
||||
logger.error("Интерфейс не определён")
|
||||
logger.error(self.translations.gettext("Интерфейс не определён"))
|
||||
return []
|
||||
cmd = ["sudo", "iw", "dev", self.interface.name(), "scan"]
|
||||
logger.debug(f"Выполняем команду: {' '.join(cmd)}")
|
||||
logger.debug(f"Executing command: {' '.join(cmd)}")
|
||||
result = subprocess.run(cmd,
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=10)
|
||||
if result.returncode != 0:
|
||||
logger.error(f"Ошибка iw scan: {result.stderr}")
|
||||
logger.error(f"iw scan error: {result.stderr}")
|
||||
return []
|
||||
output = result.stdout
|
||||
logger.debug(
|
||||
f"Вывод iw scan (первые 500 символов): {output[:500]}...")
|
||||
f"iw scan output (first 500 chars): {output[:500]}...")
|
||||
|
||||
aps = []
|
||||
current_ap = None
|
||||
|
|
@ -294,19 +300,19 @@ class WifiService:
|
|||
aps.append(current_ap)
|
||||
|
||||
logger.debug(
|
||||
f"Найдено {len(aps)} точек через iw: {[ap['bssid'] for ap in aps]}"
|
||||
f"Found {len(aps)} access points via iw: {[ap['bssid'] for ap in aps]}"
|
||||
)
|
||||
for ap in aps:
|
||||
logger.debug(f"Данные точки {ap['bssid']}: {ap}")
|
||||
logger.debug(f"Access point {ap['bssid']}: {ap}")
|
||||
return aps
|
||||
except subprocess.CalledProcessError as e:
|
||||
logger.error(f"Ошибка при выполнении iw scan: {e.stderr}")
|
||||
logger.error(f"Error executing iw scan: {e.stderr}")
|
||||
return []
|
||||
except subprocess.TimeoutExpired:
|
||||
logger.error("Таймаут при выполнении iw scan")
|
||||
logger.error(self.translations.gettext("Таймаут при выполнении iw scan"))
|
||||
return []
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка при парсинге iw scan: {str(e)}")
|
||||
logger.error(f"Error parsing iw scan: {str(e)}")
|
||||
return []
|
||||
|
||||
def scan_wifi(self, timeout: int = 30) -> WifiScanResult:
|
||||
|
|
@ -326,17 +332,17 @@ class WifiService:
|
|||
aps = []
|
||||
|
||||
if not self.interface:
|
||||
logger.error("Wi-Fi-интерфейс не доступен")
|
||||
logger.error(self.translations.gettext("Wi-Fi-интерфейс не доступен"))
|
||||
return WifiScanResult(access_points=aps)
|
||||
|
||||
try:
|
||||
logger.info(
|
||||
f"Начало сканирования Wi-Fi на интерфейсе {self.interface.name()}"
|
||||
f"Starting Wi-Fi scan on interface {self.interface.name()}"
|
||||
)
|
||||
self.interface.scan()
|
||||
time.sleep(10)
|
||||
py_results = self.interface.scan_results()
|
||||
logger.debug(f"Найдено {len(py_results)} точек через pywifi")
|
||||
logger.debug(f"Found {len(py_results)} access points via pywifi")
|
||||
|
||||
iw_results = self._get_iw_scan_data()
|
||||
iw_by_bssid = {ap["bssid"]: ap for ap in iw_results if ap.get("bssid")}
|
||||
|
|
@ -347,7 +353,7 @@ class WifiService:
|
|||
for ap in py_results:
|
||||
bssid = ap.bssid.lower() if ap.bssid else None
|
||||
if not bssid:
|
||||
logger.debug("Пропущена точка без BSSID")
|
||||
logger.debug("Skipped access point without BSSID")
|
||||
continue
|
||||
ssid = ap.ssid if ap.ssid else ""
|
||||
hidden = not ssid
|
||||
|
|
@ -361,7 +367,7 @@ class WifiService:
|
|||
vendor = self.get_vendor(bssid)
|
||||
|
||||
logger.debug(
|
||||
f"Точка {bssid}: freq={frequency}, channel={channel}, encryption={encryption}, signal={signal}"
|
||||
f"Access point {bssid}: freq={frequency}, channel={channel}, encryption={encryption}, signal={signal}"
|
||||
)
|
||||
|
||||
current_ap = WifiAccessPoint(
|
||||
|
|
@ -386,12 +392,13 @@ class WifiService:
|
|||
self.storage.save_notes(notes_to_save)
|
||||
|
||||
logger.info(
|
||||
f"Сканирование Wi-Fi завершено за {time.time() - start_time:.2f} секунд, "
|
||||
f"найдено {len(aps)} уникальных сетей")
|
||||
f"Wi-Fi scan completed in {time.time() - start_time:.2f} seconds, "
|
||||
f"found {len(aps)} unique networks"
|
||||
)
|
||||
return WifiScanResult(access_points=aps)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка при сканировании Wi-Fi: {str(e)}")
|
||||
logger.error(f"Error scanning Wi-Fi: {str(e)}")
|
||||
return WifiScanResult(access_points=aps)
|
||||
|
||||
def add_note(self, ssid: str, note: str) -> Tuple[bool, Optional[str]]:
|
||||
|
|
@ -412,17 +419,17 @@ class WifiService:
|
|||
try:
|
||||
saved_notes = self.storage.load_notes()
|
||||
if ssid not in saved_notes or saved_notes[ssid] is None:
|
||||
logger.info(f"Добавлена новая заметка для сети с SSID {ssid}")
|
||||
logger.info(f"Added new note for network with SSID {ssid}")
|
||||
elif saved_notes[ssid] is not None:
|
||||
logger.error(f"Заметка для сети с SSID {ssid} уже существует")
|
||||
return False, f"Заметка для сети с SSID {ssid} уже существует"
|
||||
logger.error(f"Note for network with SSID {ssid} already exists")
|
||||
return False, self.translations.gettext("Заметка для сети с SSID {ssid} уже существует").format(ssid=ssid)
|
||||
saved_notes[ssid] = note
|
||||
self.storage.save_notes(saved_notes)
|
||||
logger.info(f"Добавлена заметка '{note}' для сети с SSID {ssid}")
|
||||
logger.info(f"Added note '{note}' for network with SSID {ssid}")
|
||||
return True, None
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка при добавлении заметки для SSID {ssid}: {str(e)}")
|
||||
return False, f"Внутренняя ошибка сервера: {str(e)}"
|
||||
logger.error(f"Error adding note for SSID {ssid}: {str(e)}")
|
||||
return False, self.translations.gettext("Внутренняя ошибка сервера: {error}").format(error=str(e))
|
||||
|
||||
def update_note(self, ssid: str, note: str) -> Tuple[bool, Optional[str]]:
|
||||
"""Обновляет заметку сети Wi-Fi по SSID.
|
||||
|
|
@ -442,18 +449,18 @@ class WifiService:
|
|||
try:
|
||||
saved_notes = self.storage.load_notes()
|
||||
if ssid not in saved_notes:
|
||||
logger.warning(f"Сеть с SSID {ssid} не найдена")
|
||||
return False, f"Сеть с SSID {ssid} не найдена"
|
||||
logger.warning(f"Network with SSID {ssid} not found")
|
||||
return False, self.translations.gettext("Сеть с SSID {ssid} не найдена").format(ssid=ssid)
|
||||
if saved_notes[ssid] is None:
|
||||
logger.error(f"Заметка для сети с SSID {ssid} не существует")
|
||||
return False, f"Заметка для сети с SSID {ssid} не существует"
|
||||
logger.error(f"Note for network with SSID {ssid} does not exist")
|
||||
return False, self.translations.gettext("Заметка для сети с SSID {ssid} не существует").format(ssid=ssid)
|
||||
saved_notes[ssid] = note
|
||||
self.storage.save_notes(saved_notes)
|
||||
logger.info(f"Обновлена заметка '{note}' для сети с SSID {ssid}")
|
||||
logger.info(f"Updated note '{note}' for network with SSID {ssid}")
|
||||
return True, None
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка при обновлении заметки для SSID {ssid}: {str(e)}")
|
||||
return False, f"Внутренняя ошибка сервера: {str(e)}"
|
||||
logger.error(f"Error updating note for SSID {ssid}: {str(e)}")
|
||||
return False, self.translations.gettext("Внутренняя ошибка сервера: {error}").format(error=str(e))
|
||||
|
||||
def delete_note(self, ssid: str) -> Tuple[bool, Optional[str]]:
|
||||
"""Удаляет заметку сети Wi-Fi по SSID.
|
||||
|
|
@ -472,15 +479,15 @@ class WifiService:
|
|||
try:
|
||||
saved_notes = self.storage.load_notes()
|
||||
if ssid not in saved_notes:
|
||||
logger.warning(f"Сеть с SSID {ssid} не найдена")
|
||||
return False, f"Сеть с SSID {ssid} не найдена"
|
||||
logger.warning(f"Network with SSID {ssid} not found")
|
||||
return False, self.translations.gettext("Сеть с SSID {ssid} не найдена").format(ssid=ssid)
|
||||
if saved_notes[ssid] is None:
|
||||
logger.error(f"Заметка для сети с SSID {ssid} не существует")
|
||||
return False, f"Заметка для сети с SSID {ssid} не существует"
|
||||
logger.error(f"Note for network with SSID {ssid} does not exist")
|
||||
return False, self.translations.gettext("Заметка для сети с SSID {ssid} не существует").format(ssid=ssid)
|
||||
saved_notes[ssid] = None
|
||||
self.storage.save_notes(saved_notes)
|
||||
logger.info(f"Удалена заметка для сети с SSID {ssid}")
|
||||
logger.info(f"Deleted note for network with SSID {ssid}")
|
||||
return True, None
|
||||
except Exception as e:
|
||||
logger.error(f"Ошибка при удалении заметки для SSID {ssid}: {str(e)}")
|
||||
return False, f"Внутренняя ошибка сервера: {str(e)}"
|
||||
logger.error(f"Error deleting note for SSID {ssid}: {str(e)}")
|
||||
return False, self.translations.gettext("Внутренняя ошибка сервера: {error}").format(error=str(e))
|
||||
Loading…
Add table
Add a link
Reference in a new issue