159 lines
No EOL
9.5 KiB
Python
159 lines
No EOL
9.5 KiB
Python
import subprocess
|
||
import logging
|
||
from typing import Tuple, Optional
|
||
|
||
logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
|
||
logger = logging.getLogger(__name__)
|
||
|
||
class WifiConnectionService:
|
||
"""Сервис для управления подключением к Wi-Fi сетям с использованием nmcli.
|
||
|
||
Предоставляет методы для подключения, отключения и проверки статуса Wi-Fi соединений.
|
||
"""
|
||
|
||
def __init__(self):
|
||
"""Инициализирует сервис Wi-Fi.
|
||
|
||
Автоматически определяет доступный Wi-Fi-интерфейс и сохраняет его имя.
|
||
"""
|
||
self.interface = self._get_wifi_interface()
|
||
|
||
def _get_wifi_interface(self) -> str:
|
||
"""Автоматически определяет Wi-Fi-интерфейс.
|
||
|
||
Returns:
|
||
str: Название Wi-Fi-интерфейса (например, "wlan0") или пустая строка,
|
||
если интерфейс не найден или Wi-Fi отключён.
|
||
|
||
Raises:
|
||
Exception: Если произошла ошибка при выполнении команд nmcli.
|
||
"""
|
||
try:
|
||
result = subprocess.run(["nmcli", "radio", "wifi"], capture_output=True, text=True)
|
||
if "enabled" in result.stdout.lower():
|
||
interfaces = subprocess.run(["nmcli", "device", "status"], capture_output=True, text=True).stdout
|
||
for line in interfaces.splitlines():
|
||
if "wifi" in line.lower() and "connected" in line.lower():
|
||
iface = line.split()[0]
|
||
logger.info(f"Используется Wi-Fi-интерфейс: {iface}")
|
||
return iface
|
||
logger.warning("Wi-Fi-интерфейс не найден или не подключён")
|
||
return "wlan0"
|
||
logger.error("Wi-Fi радио отключено")
|
||
return ""
|
||
except Exception as e:
|
||
logger.error(f"Ошибка при определении Wi-Fi-интерфейса: {str(e)}")
|
||
return ""
|
||
|
||
def connect(self, ssid: str, password: str) -> Tuple[bool, Optional[str]]:
|
||
"""Подключается к Wi-Fi сети по SSID и паролю.
|
||
|
||
Args:
|
||
ssid (str): Идентификатор сети Wi-Fi.
|
||
password (str): Пароль для подключения.
|
||
|
||
Returns:
|
||
Tuple[bool, Optional[str]]: Кортеж (успех, сообщение об ошибке).
|
||
- Успех: (True, None).
|
||
- Ошибка: (False, сообщение об ошибке).
|
||
|
||
Raises:
|
||
subprocess.CalledProcessError: Если команда nmcli завершилась с ошибкой.
|
||
Exception: Если произошла неизвестная ошибка при выполнении.
|
||
"""
|
||
try:
|
||
if not self.interface:
|
||
return False, "Wi-Fi-интерфейс не доступен"
|
||
|
||
result = subprocess.run(["nmcli", "-f", "SSID", "device", "wifi", "list"], capture_output=True, text=True)
|
||
if ssid not in result.stdout:
|
||
return False, f"Сеть с SSID {ssid} не найдена в списке доступных сетей"
|
||
|
||
cmd = ["nmcli", "device", "wifi", "connect", ssid, "password", password]
|
||
process = subprocess.run(cmd, capture_output=True, text=True, check=False)
|
||
if process.returncode != 0:
|
||
error_msg = process.stderr.lower()
|
||
if "secrets were required" in error_msg or "authentication failed" in error_msg or "802-11-wireless-security.psk: недействительное свойство" in process.stderr:
|
||
return False, f"Ошибка подключения к {ssid}: Неверный пароль"
|
||
elif "no network with ssid" in error_msg:
|
||
return False, f"Ошибка подключения к {ssid}: Сеть с SSID {ssid} недоступна или неверно указана"
|
||
else:
|
||
return False, f"Ошибка подключения к {ssid}: {process.stderr}"
|
||
logger.info(f"Успешное подключение к сети {ssid}")
|
||
return True, None
|
||
except subprocess.CalledProcessError as e:
|
||
logger.error(f"Ошибка подключения к {ssid}: {e.stderr}")
|
||
error_msg = e.stderr.lower()
|
||
if "secrets were required" in error_msg or "authentication failed" in error_msg or "802-11-wireless-security.psk: недействительное свойство" in e.stderr:
|
||
return False, f"Ошибка подключения к {ssid}: Неверный пароль"
|
||
elif "no network with ssid" in error_msg:
|
||
return False, f"Ошибка подключения к {ssid}: Сеть с SSID {ssid} недоступна или неверно указана"
|
||
return False, f"Ошибка подключения к {ssid}: {e.stderr}"
|
||
except Exception as e:
|
||
logger.error(f"Неизвестная ошибка при подключении к {ssid}: {str(e)}")
|
||
return False, f"Ошибка подключения к {ssid}: Неизвестная ошибка - {str(e)}"
|
||
|
||
def disconnect(self, ssid: str) -> Tuple[bool, Optional[str]]:
|
||
"""Отключается от Wi-Fi сети по указанному SSID.
|
||
|
||
Args:
|
||
ssid (str): Идентификатор сети Wi-Fi для отключения.
|
||
|
||
Returns:
|
||
Tuple[bool, Optional[str]]: Кортеж (успех, сообщение об ошибке).
|
||
- Успех: (True, None или сообщение, если подключение не найдено).
|
||
- Ошибка: (False, сообщение об ошибке).
|
||
|
||
Raises:
|
||
subprocess.CalledProcessError: Если команда nmcli завершилась с ошибкой.
|
||
Exception: Если произошла неизвестная ошибка при выполнении.
|
||
"""
|
||
try:
|
||
if not self.interface:
|
||
return False, "Wi-Fi-интерфейс не доступен"
|
||
result = subprocess.run(["nmcli", "connection", "show", "--active"], capture_output=True, text=True)
|
||
active_connections = result.stdout.strip().splitlines()
|
||
if not active_connections:
|
||
logger.info(f"Нет активных подключений для SSID {ssid}")
|
||
return True, f"Нет активного подключения для SSID {ssid}"
|
||
for conn in active_connections:
|
||
conn_details = conn.split()
|
||
if len(conn_details) > 1:
|
||
conn_name = conn_details[0]
|
||
conn_info = subprocess.run(["nmcli", "connection", "show", conn_name], capture_output=True, text=True).stdout
|
||
if "802-11-wireless.ssid:" in conn_info and ssid in conn_info:
|
||
subprocess.run(["nmcli", "connection", "down", conn_name], capture_output=True, text=True, check=True)
|
||
logger.info(f"Отключено соединение с SSID {ssid}")
|
||
return True, None
|
||
logger.warning(f"Активное подключение с SSID {ssid} не найдено")
|
||
return True, f"Активное подключение с SSID {ssid} не найдено"
|
||
except subprocess.CalledProcessError as e:
|
||
logger.error(f"Ошибка при отключении от {ssid}: {e.stderr}")
|
||
return False, f"Ошибка отключения от {ssid}: {e.stderr}"
|
||
except Exception as e:
|
||
logger.error(f"Неизвестная ошибка при отключении от {ssid}: {str(e)}")
|
||
return False, f"Ошибка отключения от {ssid}: {str(e)}"
|
||
|
||
def get_status(self) -> Tuple[bool, Optional[str]]:
|
||
"""Проверяет статус текущего подключения к Wi-Fi.
|
||
|
||
Returns:
|
||
Tuple[bool, Optional[str]]: Кортеж (успех, SSID или сообщение об ошибке).
|
||
- Успех: (True, SSID текущей сети).
|
||
- Ошибка или отсутствие подключения: (False, сообщение об ошибке).
|
||
|
||
Raises:
|
||
Exception: Если произошла ошибка при выполнении команды nmcli.
|
||
"""
|
||
try:
|
||
if not self.interface:
|
||
return False, "Wi-Fi-интерфейс не доступен"
|
||
result = subprocess.run(["nmcli", "connection", "show", "--active"], capture_output=True, text=True)
|
||
if result.stdout.strip():
|
||
ssid = result.stdout.splitlines()[1].split()[0]
|
||
logger.info(f"Текущая сеть: {ssid}")
|
||
return True, ssid
|
||
return False, "Не подключено к Wi-Fi"
|
||
except Exception as e:
|
||
logger.error(f"Ошибка при проверке статуса: {str(e)}")
|
||
return False, f"Ошибка проверки статуса: {str(e)}" |