Изменение сканирования подсети и точек доступа wi-fi в связи с последними изменениями на сервере
This commit is contained in:
parent
990a7594fc
commit
f0d5481dfe
4 changed files with 188 additions and 501 deletions
|
|
@ -9,112 +9,79 @@ import pywifi
|
|||
from models.wifi import WifiAccessPoint, WifiScanResult
|
||||
from pathlib import Path
|
||||
import json
|
||||
from utils.i18n import get_translations, _
|
||||
from utils.i18n import get_translations
|
||||
from babel.support import Translations
|
||||
from fastapi import Depends
|
||||
|
||||
# Настройка логирования
|
||||
logging.basicConfig(
|
||||
level=logging.DEBUG,
|
||||
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
|
||||
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
||||
)
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class WifiStorage:
|
||||
"""Класс для хранения данных о Wi-Fi сетях и их заметках по SSID.
|
||||
|
||||
Предоставляет методы для загрузки и сохранения заметок, связанных с SSID сетей,
|
||||
в JSON-файле на диске.
|
||||
"""
|
||||
|
||||
"""Класс для хранения данных о Wi-Fi сетях и их заметках по SSID."""
|
||||
def __init__(self, storage_path: str = "storage/wifi/wifi.json", translations: Translations = Depends(get_translations("services"))):
|
||||
"""Инициализирует объект хранилища Wi-Fi с указанным путём к файлу.
|
||||
|
||||
Args:
|
||||
storage_path (str, optional): Путь к файлу для хранения заметок.
|
||||
По умолчанию "storage/wifi/wifi.json".
|
||||
|
||||
Создаёт родительские директории, если они не существуют.
|
||||
"""
|
||||
self.storage_file = Path(storage_path)
|
||||
self.storage_file.parent.mkdir(parents=True, exist_ok=True)
|
||||
self.translations = translations
|
||||
logger.info(f"Wi-Fi storage initialized at: {self.storage_file.absolute()}")
|
||||
|
||||
def load_notes(self) -> Dict[str, Optional[str]]:
|
||||
"""Загружает заметки по SSID из файла хранения.
|
||||
|
||||
Returns:
|
||||
Dict[str, Optional[str]]: Словарь, где ключ — SSID, значение — заметка
|
||||
или None, если заметки нет.
|
||||
|
||||
Raises:
|
||||
Exception: Если произошла ошибка при чтении или парсинге файла.
|
||||
"""
|
||||
"""Загружает заметки по SSID из файла хранения."""
|
||||
try:
|
||||
if self.storage_file.exists():
|
||||
with open(self.storage_file, "r", encoding="utf-8") as f:
|
||||
data = json.load(f)
|
||||
if not isinstance(data, dict) or "networks" not in data:
|
||||
logger.warning(f"Invalid JSON structure in {self.storage_file}, expected {'networks': [...]}, got {data}")
|
||||
return {}
|
||||
return {item["ssid"]: item.get("note") for item in data.get("networks", [])}
|
||||
return {}
|
||||
except json.JSONDecodeError as e:
|
||||
logger.error(f"JSON decode error in {self.storage_file}: {str(e)}")
|
||||
return {}
|
||||
except Exception as e:
|
||||
logger.error(f"Error loading notes: {str(e)}")
|
||||
logger.error(f"Error loading notes from {self.storage_file}: {str(e)}")
|
||||
return {}
|
||||
|
||||
def save_notes(self, notes: Dict[str, Optional[str]]) -> None:
|
||||
"""Сохраняет заметки по SSID в файл.
|
||||
|
||||
Args:
|
||||
notes (Dict[str, Optional[str]]): Словарь с SSID как ключами и заметками
|
||||
как значениями.
|
||||
|
||||
Raises:
|
||||
Exception: Если произошла ошибка при записи в файл.
|
||||
"""
|
||||
"""Сохраняет заметки по SSID в файл."""
|
||||
try:
|
||||
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"Saved {len(notes)} notes to {self.storage_file}")
|
||||
logger.info(f"Saved {len(notes)} notes to {self.storage_file.absolute()}")
|
||||
except PermissionError as e:
|
||||
logger.error(f"Permission denied writing to {self.storage_file}: {str(e)}")
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f"Error saving notes: {str(e)}")
|
||||
logger.error(f"Error saving notes to {self.storage_file}: {str(e)}")
|
||||
raise Exception(self.translations.gettext("Ошибка при сохранении заметок: {error}").format(error=str(e)))
|
||||
|
||||
class WifiService:
|
||||
"""Сервис для сканирования текущих Wi-Fi точек доступа.
|
||||
|
||||
Предоставляет функционал для обнаружения Wi-Fi сетей с использованием pywifi и iw,
|
||||
определения характеристик точек доступа (сигнал, шифрование, канал и т.д.),
|
||||
а также управления заметками.
|
||||
"""
|
||||
|
||||
"""Сервис для сканирования текущих Wi-Fi точек доступа."""
|
||||
def __init__(self, translations: Translations = Depends(get_translations("services"))):
|
||||
"""Инициализирует сервис Wi-Fi с настройкой интерфейса и хранилища.
|
||||
|
||||
Инициализирует объект MacLookup для определения вендоров по MAC-адресам,
|
||||
PyWiFi для работы с Wi-Fi-интерфейсами и объект хранилища заметок.
|
||||
"""
|
||||
self.mac_lookup = MacLookup()
|
||||
self.wifi = pywifi.PyWiFi()
|
||||
self.interface = self._get_wifi_interface()
|
||||
if not hasattr(translations, 'gettext'):
|
||||
logger.warning("Translations object does not support gettext, using fallback")
|
||||
self.translations = Translations.load("utils/locales/en/LC_MESSAGES/services.mo")
|
||||
else:
|
||||
self.translations = translations
|
||||
self.storage = WifiStorage()
|
||||
self.translations = translations
|
||||
|
||||
def _get_wifi_interface(self) -> Optional[object]:
|
||||
"""Автоматически определяет активный Wi-Fi-интерфейс.
|
||||
|
||||
Returns:
|
||||
Optional[object]: Объект интерфейса PyWiFi или None, если интерфейс не найден.
|
||||
|
||||
Проверяет доступные интерфейсы, выполняет тестовое сканирование и возвращает
|
||||
первый активный интерфейс, исключая P2P-интерфейсы.
|
||||
"""
|
||||
"""Автоматически определяет активный Wi-Fi-интерфейс."""
|
||||
try:
|
||||
interfaces = self.wifi.interfaces()
|
||||
if not interfaces:
|
||||
logger.error(self.translations.gettext("Не найдено сетевых интерфейсов"))
|
||||
return None
|
||||
|
||||
logger.debug(
|
||||
f"Available interfaces: {[iface.name() for iface in interfaces]}"
|
||||
)
|
||||
logger.debug(f"Available interfaces: {[iface.name() for iface in interfaces]}")
|
||||
|
||||
for iface in interfaces:
|
||||
iface_name = iface.name()
|
||||
|
|
@ -128,9 +95,7 @@ class WifiService:
|
|||
if results:
|
||||
logger.info(f"Selected Wi-Fi interface: {iface_name}")
|
||||
return iface
|
||||
logger.debug(
|
||||
f"Interface {iface_name} returned no scan results"
|
||||
)
|
||||
logger.debug(f"Interface {iface_name} returned no scan results")
|
||||
except Exception as e:
|
||||
logger.debug(f"Interface {iface_name} inactive: {str(e)}")
|
||||
|
||||
|
|
@ -142,36 +107,17 @@ class WifiService:
|
|||
|
||||
@lru_cache(maxsize=1000)
|
||||
def get_vendor(self, mac: str) -> str:
|
||||
"""Определяет вендора по MAC-адресу с использованием кэширования.
|
||||
|
||||
Args:
|
||||
mac (str): MAC-адрес устройства в формате XX:XX:XX:XX:XX:XX.
|
||||
|
||||
Returns:
|
||||
str: Название вендора или "Unknown", если вендор не определён.
|
||||
|
||||
Raises:
|
||||
Exception: Если произошла ошибка при запросе к базе вендоров.
|
||||
"""
|
||||
"""Определяет вендора по MAC-адресу с использованием кэширования."""
|
||||
try:
|
||||
return self.mac_lookup.lookup(mac)
|
||||
except Exception as e:
|
||||
logger.warning(
|
||||
f"Failed to determine vendor for MAC {mac}: {str(e)}")
|
||||
logger.warning(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.
|
||||
|
||||
Args:
|
||||
ap_info (Dict): Словарь с данными точки доступа, включая поле "security".
|
||||
|
||||
Returns:
|
||||
str: Тип шифрования ("WPA3", "WPA2", "WPA", "WEP" или "Open").
|
||||
"""
|
||||
"""Определяет тип шифрования из данных, полученных через iw."""
|
||||
security = ap_info.get("security", "")
|
||||
logger.debug(
|
||||
f"Encryption for BSSID {ap_info.get('bssid')}: {security}")
|
||||
logger.debug(f"Encryption for BSSID {ap_info.get('bssid')}: {security}")
|
||||
if "WPA3" in security:
|
||||
return self.translations.gettext("WPA3")
|
||||
elif "WPA2" in security or "RSN" in security:
|
||||
|
|
@ -183,14 +129,7 @@ class WifiService:
|
|||
return self.translations.gettext("Открытая")
|
||||
|
||||
def _get_channel(self, frequency: Optional[int]) -> Optional[int]:
|
||||
"""Преобразует частоту в номер канала.
|
||||
|
||||
Args:
|
||||
frequency (Optional[int]): Частота в МГц или None.
|
||||
|
||||
Returns:
|
||||
Optional[int]: Номер канала или None, если частота вне диапазона.
|
||||
"""
|
||||
"""Преобразует частоту в номер канала."""
|
||||
if frequency is None:
|
||||
return None
|
||||
try:
|
||||
|
|
@ -201,20 +140,11 @@ class WifiService:
|
|||
logger.debug(f"Frequency {frequency} outside 2.4/5 GHz range")
|
||||
return None
|
||||
except Exception as e:
|
||||
logger.debug(
|
||||
f"Error determining channel for frequency {frequency}: {str(e)}"
|
||||
)
|
||||
logger.debug(f"Error determining channel for frequency {frequency}: {str(e)}")
|
||||
return None
|
||||
|
||||
def _get_frequency(self, frequency: Optional[int]) -> Optional[str]:
|
||||
"""Определяет частотный диапазон на основе частоты.
|
||||
|
||||
Args:
|
||||
frequency (Optional[int]): Частота в МГц или None.
|
||||
|
||||
Returns:
|
||||
Optional[str]: Диапазон ("2.4 GHz", "5 GHz" или None).
|
||||
"""
|
||||
"""Определяет частотный диапазон на основе частоты."""
|
||||
if frequency is None:
|
||||
return None
|
||||
try:
|
||||
|
|
@ -225,39 +155,23 @@ class WifiService:
|
|||
logger.debug(f"Frequency {frequency} outside 2.4/5 GHz range")
|
||||
return None
|
||||
except Exception as e:
|
||||
logger.debug(
|
||||
f"Error determining band for frequency {frequency}: {str(e)}"
|
||||
)
|
||||
logger.debug(f"Error determining band for frequency {frequency}: {str(e)}")
|
||||
return None
|
||||
|
||||
def _get_iw_scan_data(self) -> List[Dict]:
|
||||
"""Получает данные сканирования Wi-Fi через команду iw.
|
||||
|
||||
Returns:
|
||||
List[Dict]: Список словарей с данными точек доступа (BSSID, SSID, частота,
|
||||
сигнал, шифрование).
|
||||
|
||||
Raises:
|
||||
subprocess.CalledProcessError: Если команда iw завершилась с ошибкой.
|
||||
subprocess.TimeoutExpired: Если превышен таймаут выполнения.
|
||||
Exception: Если произошла ошибка при парсинге вывода.
|
||||
"""
|
||||
"""Получает данные сканирования Wi-Fi через команду iw."""
|
||||
try:
|
||||
if not self.interface:
|
||||
logger.error(self.translations.gettext("Интерфейс не определён"))
|
||||
return []
|
||||
cmd = ["sudo", "iw", "dev", self.interface.name(), "scan"]
|
||||
logger.debug(f"Executing command: {' '.join(cmd)}")
|
||||
result = subprocess.run(cmd,
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=10)
|
||||
result = subprocess.run(cmd, capture_output=True, text=True, timeout=10)
|
||||
if result.returncode != 0:
|
||||
logger.error(f"iw scan error: {result.stderr}")
|
||||
return []
|
||||
output = result.stdout
|
||||
logger.debug(
|
||||
f"iw scan output (first 500 chars): {output[:500]}...")
|
||||
logger.debug(f"iw scan output (first 500 chars): {output[:500]}...")
|
||||
|
||||
aps = []
|
||||
current_ap = None
|
||||
|
|
@ -274,15 +188,10 @@ class WifiService:
|
|||
if bssid_match := bssid_pattern.match(line):
|
||||
if current_ap:
|
||||
aps.append(current_ap)
|
||||
current_ap = {
|
||||
"bssid": bssid_match.group(1).lower(),
|
||||
"security": ""
|
||||
}
|
||||
current_ap = {"bssid": bssid_match.group(1).lower(), "security": ""}
|
||||
elif current_ap:
|
||||
if ssid_match := ssid_pattern.match(line):
|
||||
current_ap["ssid"] = (ssid_match.group(1)
|
||||
if ssid_match.group(1).strip()
|
||||
else "")
|
||||
current_ap["ssid"] = ssid_match.group(1).strip() if ssid_match.group(1).strip() else ""
|
||||
elif freq_match := freq_pattern.match(line):
|
||||
current_ap["frequency"] = int(freq_match.group(1))
|
||||
elif signal_match := signal_pattern.match(line):
|
||||
|
|
@ -291,19 +200,13 @@ class WifiService:
|
|||
current_ap["security"] += "RSN;"
|
||||
elif wpa_pattern.match(line):
|
||||
current_ap["security"] += "WPA;"
|
||||
elif security_type_match := security_type_pattern.search(
|
||||
line):
|
||||
current_ap["security"] += security_type_match.group(
|
||||
1) + ";"
|
||||
elif security_type_match := security_type_pattern.search(line):
|
||||
current_ap["security"] += security_type_match.group(1) + ";"
|
||||
|
||||
if current_ap:
|
||||
aps.append(current_ap)
|
||||
|
||||
logger.debug(
|
||||
f"Found {len(aps)} access points via iw: {[ap['bssid'] for ap in aps]}"
|
||||
)
|
||||
for ap in aps:
|
||||
logger.debug(f"Access point {ap['bssid']}: {ap}")
|
||||
logger.debug(f"Found {len(aps)} access points via iw: {[ap['bssid'] for ap in aps]}")
|
||||
return aps
|
||||
except subprocess.CalledProcessError as e:
|
||||
logger.error(f"Error executing iw scan: {e.stderr}")
|
||||
|
|
@ -316,18 +219,7 @@ class WifiService:
|
|||
return []
|
||||
|
||||
def scan_wifi(self, timeout: int = 30) -> WifiScanResult:
|
||||
"""Сканирует текущие Wi-Fi точки доступа.
|
||||
|
||||
Args:
|
||||
timeout (int, optional): Таймаут ожидания в секундах. По умолчанию 30.
|
||||
|
||||
Returns:
|
||||
WifiScanResult: Объект с результатами сканирования, содержащий список точек
|
||||
доступа.
|
||||
|
||||
Сканирует Wi-Fi с использованием pywifi и iw, объединяет данные, определяет
|
||||
лучшие точки по сигналу и сохраняет заметки.
|
||||
"""
|
||||
"""Сканирует текущие Wi-Fi точки доступа."""
|
||||
start_time = time.time()
|
||||
aps = []
|
||||
|
||||
|
|
@ -336,13 +228,20 @@ class WifiService:
|
|||
return WifiScanResult(access_points=aps)
|
||||
|
||||
try:
|
||||
logger.info(
|
||||
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"Found {len(py_results)} access points via pywifi")
|
||||
logger.info(f"Starting Wi-Fi scan on interface {self.interface.name()}")
|
||||
max_attempts = 3
|
||||
for attempt in range(max_attempts):
|
||||
self.interface.scan()
|
||||
time.sleep(min(timeout, 5))
|
||||
py_results = self.interface.scan_results()
|
||||
if py_results:
|
||||
logger.info(f"Found {len(py_results)} access points via pywifi after attempt {attempt + 1}")
|
||||
break
|
||||
else:
|
||||
logger.warning(f"Scan attempt {attempt + 1} failed, trying again (FAIL-BUSY?)")
|
||||
else:
|
||||
logger.error("All scan attempts failed")
|
||||
return WifiScanResult(access_points=aps)
|
||||
|
||||
iw_results = self._get_iw_scan_data()
|
||||
iw_by_bssid = {ap["bssid"]: ap for ap in iw_results if ap.get("bssid")}
|
||||
|
|
@ -355,8 +254,8 @@ class WifiService:
|
|||
if not bssid:
|
||||
logger.debug("Skipped access point without BSSID")
|
||||
continue
|
||||
ssid = ap.ssid if ap.ssid else ""
|
||||
hidden = not ssid
|
||||
ssid = ap.ssid if ap.ssid else f"Hidden_{bssid}"
|
||||
hidden = not ap.ssid
|
||||
signal = ap.signal if ap.signal else None
|
||||
iw_data = iw_by_bssid.get(bssid, {})
|
||||
|
||||
|
|
@ -366,10 +265,6 @@ class WifiService:
|
|||
encryption = self._get_encryption_from_iw(iw_data)
|
||||
vendor = self.get_vendor(bssid)
|
||||
|
||||
logger.debug(
|
||||
f"Access point {bssid}: freq={frequency}, channel={channel}, encryption={encryption}, signal={signal}"
|
||||
)
|
||||
|
||||
current_ap = WifiAccessPoint(
|
||||
ssid=ssid,
|
||||
bssid=bssid,
|
||||
|
|
@ -379,22 +274,27 @@ class WifiService:
|
|||
frequency=freq_band,
|
||||
vendor=vendor,
|
||||
hidden=hidden,
|
||||
note=saved_notes.get(ssid) if ap.ssid else None,
|
||||
)
|
||||
logger.debug(f"Processed AP: {current_ap.model_dump()}")
|
||||
|
||||
if ssid not in best_aps_by_ssid or (
|
||||
current_ap.signal and best_aps_by_ssid[ssid].signal
|
||||
and current_ap.signal > best_aps_by_ssid[ssid].signal):
|
||||
current_ap.note = saved_notes.get(ssid)
|
||||
current_ap.note = saved_notes.get(ssid) if ap.ssid else None
|
||||
best_aps_by_ssid[ssid] = current_ap
|
||||
|
||||
aps = list(best_aps_by_ssid.values())
|
||||
# Сохраняем хотя бы SSID, если заметок нет
|
||||
notes_to_save = {ap.ssid: ap.note for ap in aps if ap.ssid}
|
||||
self.storage.save_notes(notes_to_save)
|
||||
if not notes_to_save and aps:
|
||||
notes_to_save = {ap.ssid: None for ap in aps if ap.ssid} # Сохраняем SSID без заметок
|
||||
if notes_to_save:
|
||||
self.storage.save_notes(notes_to_save)
|
||||
else:
|
||||
logger.warning("No notes or SSIDs to save, skipping file write")
|
||||
|
||||
logger.info(
|
||||
f"Wi-Fi scan completed in {time.time() - start_time:.2f} seconds, "
|
||||
f"found {len(aps)} unique networks"
|
||||
)
|
||||
logger.info(f"Wi-Fi scan completed, found {len(aps)} unique networks")
|
||||
return WifiScanResult(access_points=aps)
|
||||
|
||||
except Exception as e:
|
||||
|
|
@ -402,20 +302,7 @@ class WifiService:
|
|||
return WifiScanResult(access_points=aps)
|
||||
|
||||
def add_note(self, ssid: str, note: str) -> Tuple[bool, Optional[str]]:
|
||||
"""Добавляет заметку к сети Wi-Fi по SSID.
|
||||
|
||||
Args:
|
||||
ssid (str): SSID сети Wi-Fi.
|
||||
note (str): Текст заметки.
|
||||
|
||||
Returns:
|
||||
Tuple[bool, Optional[str]]: Кортеж (успех, сообщение об ошибке).
|
||||
- Успех: (True, None).
|
||||
- Ошибка: (False, сообщение об ошибке).
|
||||
|
||||
Raises:
|
||||
Exception: Если произошла ошибка при сохранении заметки.
|
||||
"""
|
||||
"""Добавляет заметку к сети Wi-Fi по SSID."""
|
||||
try:
|
||||
saved_notes = self.storage.load_notes()
|
||||
if ssid not in saved_notes or saved_notes[ssid] is None:
|
||||
|
|
@ -432,20 +319,7 @@ class WifiService:
|
|||
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.
|
||||
|
||||
Args:
|
||||
ssid (str): SSID сети Wi-Fi.
|
||||
note (str): Новый текст заметки.
|
||||
|
||||
Returns:
|
||||
Tuple[bool, Optional[str]]: Кортеж (успех, сообщение об ошибке).
|
||||
- Успех: (True, None).
|
||||
- Ошибка: (False, сообщение об ошибке).
|
||||
|
||||
Raises:
|
||||
Exception: Если произошла ошибка при сохранении заметки.
|
||||
"""
|
||||
"""Обновляет заметку сети Wi-Fi по SSID."""
|
||||
try:
|
||||
saved_notes = self.storage.load_notes()
|
||||
if ssid not in saved_notes:
|
||||
|
|
@ -463,19 +337,7 @@ class WifiService:
|
|||
return False, self.translations.gettext("Внутренняя ошибка сервера: {error}").format(error=str(e))
|
||||
|
||||
def delete_note(self, ssid: str) -> Tuple[bool, Optional[str]]:
|
||||
"""Удаляет заметку сети Wi-Fi по SSID.
|
||||
|
||||
Args:
|
||||
ssid (str): SSID сети Wi-Fi.
|
||||
|
||||
Returns:
|
||||
Tuple[bool, Optional[str]]: Кортеж (успех, сообщение об ошибке).
|
||||
- Успех: (True, None).
|
||||
- Ошибка: (False, сообщение об ошибке).
|
||||
|
||||
Raises:
|
||||
Exception: Если произошла ошибка при сохранении изменений.
|
||||
"""
|
||||
"""Удаляет заметку сети Wi-Fi по SSID."""
|
||||
try:
|
||||
saved_notes = self.storage.load_notes()
|
||||
if ssid not in saved_notes:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue