unifi-network-mcp 0.2.0__tar.gz → 0.3.1__tar.gz

{unifi_network_mcp-0.2.0 → unifi_network_mcp-0.3.1}/.well-known/mcp-server.json

@@ -46,10 +46,14 @@
       "tool": "unifi_tool_index",
       "description": "Returns machine-readable list of all available tools with schemas"
     },
-    "async_operations": {
-      "start_tool": "unifi_async_start",
-      "status_tool": "unifi_async_status",
-      "description": "Background job execution for long-running operations"
+    "execution": {
+      "tool": "unifi_execute",
+      "description": "Execute any discovered tool synchronously"
+    },
+    "batch_operations": {
+      "start_tool": "unifi_batch",
+      "status_tool": "unifi_batch_status",
+      "description": "Parallel batch execution for bulk operations"
     }
   }
 }

{unifi_network_mcp-0.2.0 → unifi_network_mcp-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unifi-network-mcp
-Version: 0.2.0
+Version: 0.3.1
 Summary: Unifi Network MCP Server
 License-File: LICENSE
 Requires-Python: >=3.13
@@ -45,7 +45,7 @@ A self-hosted [Model Context Protocol](https://github.com/modelcontextprotocol)
   * [Overview](#overview)
   * [Context Optimization](#context-optimization)
   * [Tool Index](#tool-index)
-  * [Async Operations](#async-operations)
+  * [Tool Execution](#tool-execution)
 * [Runtime Configuration](#runtime-configuration)
 * [Diagnostics (Advanced Logging)](#diagnostics-advanced-logging)
 * [Developer Console (Local Tool Tester)](#developer-console-local-tool-tester)
@@ -62,6 +62,7 @@ A self-hosted [Model Context Protocol](https://github.com/modelcontextprotocol)
 
 * Full catalog of UniFi controller operations – firewall, traffic-routes, port-forwards, QoS, VPN, WLANs, stats, devices, clients **and more**.
 * All mutating tools require `confirm=true` so nothing can change your network by accident.
+* **Workflow automation friendly** – set `UNIFI_AUTO_CONFIRM=true` to skip confirmation prompts (ideal for n8n, Make, Zapier).
 * Works over **stdio** (FastMCP). Optional SSE HTTP endpoint can be enabled via config.
 * **Code execution mode** with tool index, async operations, and TypeScript examples.
 * One-liner launch via the console-script **`unifi-network-mcp`**.
@@ -267,73 +268,73 @@ The server exposes a special `unifi_tool_index` tool that returns a complete lis
 - Dynamic client configuration
 - IDE autocomplete support
 
-### Async Operations
+### Tool Execution
 
-For long-running operations (device upgrades, bulk configuration changes), the server provides async job execution:
+The server provides two execution modes for discovered tools:
 
-**Start a background job:**
+**Single Tool Execution (synchronous):**
 ```json
 {
-  "name": "unifi_async_start",
+  "name": "unifi_execute",
   "arguments": {
-    "tool": "unifi_upgrade_device",
-    "arguments": {
-      "mac_address": "aa:bb:cc:dd:ee:ff",
-      "confirm": true
-    }
+    "tool": "unifi_list_clients",
+    "arguments": {}
   }
 }
 ```
 
-**Response:**
-```json
-{
-  "jobId": "af33b233cbdc860c"
-}
-```
+**Batch Execution (parallel, async):**
+
+For bulk operations or long-running tasks, use batch mode:
 
-**Check job status:**
 ```json
 {
-  "name": "unifi_async_status",
+  "name": "unifi_batch",
   "arguments": {
-    "jobId": "af33b233cbdc860c"
+    "operations": [
+      {"tool": "unifi_get_client_details", "arguments": {"mac": "aa:bb:cc:dd:ee:ff"}},
+      {"tool": "unifi_get_client_details", "arguments": {"mac": "11:22:33:44:55:66"}}
+    ]
   }
 }
 ```
 
-**Response (running):**
+**Response:**
 ```json
 {
-  "status": "running",
-  "started": 1234567890.123
+  "jobs": [
+    {"index": 0, "tool": "unifi_get_client_details", "jobId": "af33b233cbdc860c"},
+    {"index": 1, "tool": "unifi_get_client_details", "jobId": "bf44c344dcde971d"}
+  ],
+  "message": "Started 2 operation(s). Use unifi_batch_status to check progress."
 }
 ```
 
-**Response (completed):**
+**Check batch status:**
 ```json
 {
-  "status": "done",
-  "result": {...},
-  "started": 1234567890.123,
-  "finished": 1234567891.456
+  "name": "unifi_batch_status",
+  "arguments": {
+    "jobIds": ["af33b233cbdc860c", "bf44c344dcde971d"]
+  }
 }
 ```
 
-**Response (error):**
+**Response:**
 ```json
 {
-  "status": "error",
-  "error": "Device not found",
-  "started": 1234567890.123,
-  "finished": 1234567891.456
+  "jobs": [
+    {"jobId": "af33b233cbdc860c", "status": "done", "result": {...}},
+    {"jobId": "bf44c344dcde971d", "status": "done", "result": {...}}
+  ]
 }
 ```
 
 **Notes:**
+- Use `unifi_execute` for single operations (returns result directly)
+- Use `unifi_batch` + `unifi_batch_status` for parallel/bulk operations
 - Jobs are stored in-memory only (no persistence)
 - Job IDs are unique per server session
-- Failed jobs capture exception details in the `error` field
 
 ### Using with Claude Desktop
 
@@ -360,22 +361,32 @@ Practical examples showing programmatic usage:
 ```python
 from mcp import ClientSession, stdio_client
 
-# Query tool index
-result = await session.call_tool("unifi_tool_index", {})
+# Discover tools
+tools = await session.call_tool("unifi_tool_index", {})
+
+# Execute a single tool (returns result directly)
+result = await session.call_tool("unifi_execute", {
+    "tool": "unifi_list_clients",
+    "arguments": {}
+})
 
-# Use async jobs for long operations
-job = await session.call_tool("unifi_async_start", {
-    "tool": "unifi_upgrade_device",
-    "arguments": {"mac_address": "...", "confirm": True}
+# Batch execution for parallel operations
+batch = await session.call_tool("unifi_batch", {
+    "operations": [
+        {"tool": "unifi_get_client_details", "arguments": {"mac": "..."}},
+        {"tool": "unifi_get_device_details", "arguments": {"mac": "..."}}
+    ]
 })
 
-# Check job status
-status = await session.call_tool("unifi_async_status", {"jobId": job["jobId"]})
+# Check batch status
+status = await session.call_tool("unifi_batch_status", {
+    "jobIds": [j["jobId"] for j in batch["jobs"]]
+})
 ```
 
 **Three complete examples:**
 - `query_tool_index.py` - Discover available tools
-- `use_async_jobs.py` - Background job management
+- `use_async_jobs.py` - Batch operations and status checking
 - `programmatic_client.py` - Build custom Python clients
 
 See [`examples/python/README.md`](examples/python/README.md) for complete examples.
@@ -392,15 +403,18 @@ The server advertises its capabilities via an MCP identity file at [`.well-known
   "capabilities": {
     "tools": true,
     "tool_index": true,
-    "async_operations": true
+    "batch_operations": true
   },
   "features": {
     "tool_index": {
       "tool": "unifi_tool_index"
     },
-    "async_operations": {
-      "start_tool": "unifi_async_start",
-      "status_tool": "unifi_async_status"
+    "execution": {
+      "tool": "unifi_execute"
+    },
+    "batch_operations": {
+      "start_tool": "unifi_batch",
+      "status_tool": "unifi_batch_status"
     }
   }
 }
@@ -520,6 +534,44 @@ The server merges settings from **environment variables**, an optional `.env` fi
 | `UNIFI_VERIFY_SSL` | Set to `false` if using self-signed certs |
 | `UNIFI_CONTROLLER_TYPE` | Controller API path type: `auto` (detect), `proxy` (UniFi OS), `direct` (standalone). Default `auto` |
 | `UNIFI_MCP_HTTP_ENABLED` | Set `true` to enable optional HTTP SSE server (default `false`) |
+| `UNIFI_AUTO_CONFIRM` | Set `true` to auto-confirm all mutating operations (skips preview step). Ideal for workflow automation (n8n, Make, Zapier). Default `false` |
+| `UNIFI_TOOL_REGISTRATION_MODE` | Tool loading mode: `lazy` (default), `eager`, or `meta_only`. See [Context Optimization](#context-optimization) |
+| `UNIFI_ENABLED_CATEGORIES` | Comma-separated list of tool categories to load (eager mode). See table below |
+| `UNIFI_ENABLED_TOOLS` | Comma-separated list of specific tool names to register (eager mode) |
+
+### Tool Categories (for UNIFI_ENABLED_CATEGORIES)
+
+When using eager mode with category filtering, these are the valid category names:
+
+| Category | Description | Example Tools |
+|----------|-------------|---------------|
+| `clients` | Client listing, blocking, guest auth | `unifi_list_clients`, `unifi_block_client` |
+| `config` | Configuration management | - |
+| `devices` | Device listing, reboot, locate, upgrade | `unifi_list_devices`, `unifi_reboot_device` |
+| `events` | Events and alarms | `unifi_list_events`, `unifi_list_alarms` |
+| `firewall` | Firewall rules and groups | `unifi_list_firewall_rules`, `unifi_create_firewall_rule` |
+| `hotspot` | Vouchers for guest network | `unifi_list_vouchers`, `unifi_create_voucher` |
+| `network` | Network/VLAN management | `unifi_list_networks`, `unifi_create_network` |
+| `port_forwards` | Port forwarding rules | `unifi_list_port_forwards` |
+| `qos` | QoS/traffic shaping rules | `unifi_list_qos_rules`, `unifi_create_qos_rule` |
+| `routing` | Static routes (V1 API) | `unifi_list_routes`, `unifi_create_route` |
+| `stats` | Statistics and metrics | `unifi_get_client_stats`, `unifi_get_device_stats` |
+| `system` | System info, health, settings | `unifi_get_system_info`, `unifi_get_network_health` |
+| `traffic_routes` | Policy-based routing (V2 API) | `unifi_list_traffic_routes` |
+| `usergroups` | Bandwidth profiles/user groups | `unifi_list_usergroups`, `unifi_create_usergroup` |
+| `vpn` | VPN servers and clients | `unifi_list_vpn_servers`, `unifi_list_vpn_clients` |
+
+**Example usage:**
+```bash
+# Load only client and system tools
+export UNIFI_TOOL_REGISTRATION_MODE=eager
+export UNIFI_ENABLED_CATEGORIES=clients,system
+
+# Or load specific tools only
+export UNIFI_ENABLED_TOOLS=unifi_list_clients,unifi_list_devices,unifi_get_system_info
+```
+
+**Note:** Tools may also be filtered by the `permissions` section in config.yaml (e.g., `clients.update: false` blocks mutating client tools).
 
 ### Controller Type Detection
 
@@ -842,6 +894,37 @@ See [docs/permissions.md](docs/permissions.md) for complete documentation includ
 * `unifi_force_reconnect_client`
 * `unifi_authorize_guest`
 * `unifi_unauthorize_guest`
+* `unifi_set_client_ip_settings`
+
+### Events & Alarms
+
+* `unifi_list_events`
+* `unifi_list_alarms`
+* `unifi_archive_alarm`
+* `unifi_archive_all_alarms`
+* `unifi_get_event_types`
+
+### Routing (Static Routes)
+
+* `unifi_list_routes`
+* `unifi_get_route_details`
+* `unifi_create_route`
+* `unifi_update_route`
+* `unifi_list_active_routes`
+
+### Hotspot (Vouchers)
+
+* `unifi_list_vouchers`
+* `unifi_get_voucher_details`
+* `unifi_create_voucher`
+* `unifi_revoke_voucher`
+
+### User Groups
+
+* `unifi_list_usergroups`
+* `unifi_get_usergroup_details`
+* `unifi_create_usergroup`
+* `unifi_update_usergroup`
 
 ### Statistics & Alerts
 
@@ -1018,9 +1101,9 @@ uv run python devtools/dev_console.py
 
 # You'll see a menu of all tools including:
 # - unifi_tool_index (list all tools with schemas)
-# - unifi_async_start (start background jobs)
-# - unifi_async_status (check job status)
-# - All 60+ UniFi tools (clients, devices, networks, etc.)
+# - unifi_execute (run any discovered tool)
+# - unifi_batch / unifi_batch_status (parallel operations)
+# - All 80+ UniFi tools (clients, devices, networks, etc.)
 ```
 
 **4. Test with Python examples:**

{unifi_network_mcp-0.2.0 → unifi_network_mcp-0.3.1}/README.md

@@ -29,7 +29,7 @@ A self-hosted [Model Context Protocol](https://github.com/modelcontextprotocol)
   * [Overview](#overview)
   * [Context Optimization](#context-optimization)
   * [Tool Index](#tool-index)
-  * [Async Operations](#async-operations)
+  * [Tool Execution](#tool-execution)
 * [Runtime Configuration](#runtime-configuration)
 * [Diagnostics (Advanced Logging)](#diagnostics-advanced-logging)
 * [Developer Console (Local Tool Tester)](#developer-console-local-tool-tester)
@@ -46,6 +46,7 @@ A self-hosted [Model Context Protocol](https://github.com/modelcontextprotocol)
 
 * Full catalog of UniFi controller operations – firewall, traffic-routes, port-forwards, QoS, VPN, WLANs, stats, devices, clients **and more**.
 * All mutating tools require `confirm=true` so nothing can change your network by accident.
+* **Workflow automation friendly** – set `UNIFI_AUTO_CONFIRM=true` to skip confirmation prompts (ideal for n8n, Make, Zapier).
 * Works over **stdio** (FastMCP). Optional SSE HTTP endpoint can be enabled via config.
 * **Code execution mode** with tool index, async operations, and TypeScript examples.
 * One-liner launch via the console-script **`unifi-network-mcp`**.
@@ -251,73 +252,73 @@ The server exposes a special `unifi_tool_index` tool that returns a complete lis
 - Dynamic client configuration
 - IDE autocomplete support
 
-### Async Operations
+### Tool Execution
 
-For long-running operations (device upgrades, bulk configuration changes), the server provides async job execution:
+The server provides two execution modes for discovered tools:
 
-**Start a background job:**
+**Single Tool Execution (synchronous):**
 ```json
 {
-  "name": "unifi_async_start",
+  "name": "unifi_execute",
   "arguments": {
-    "tool": "unifi_upgrade_device",
-    "arguments": {
-      "mac_address": "aa:bb:cc:dd:ee:ff",
-      "confirm": true
-    }
+    "tool": "unifi_list_clients",
+    "arguments": {}
   }
 }
 ```
 
-**Response:**
-```json
-{
-  "jobId": "af33b233cbdc860c"
-}
-```
+**Batch Execution (parallel, async):**
+
+For bulk operations or long-running tasks, use batch mode:
 
-**Check job status:**
 ```json
 {
-  "name": "unifi_async_status",
+  "name": "unifi_batch",
   "arguments": {
-    "jobId": "af33b233cbdc860c"
+    "operations": [
+      {"tool": "unifi_get_client_details", "arguments": {"mac": "aa:bb:cc:dd:ee:ff"}},
+      {"tool": "unifi_get_client_details", "arguments": {"mac": "11:22:33:44:55:66"}}
+    ]
   }
 }
 ```
 
-**Response (running):**
+**Response:**
 ```json
 {
-  "status": "running",
-  "started": 1234567890.123
+  "jobs": [
+    {"index": 0, "tool": "unifi_get_client_details", "jobId": "af33b233cbdc860c"},
+    {"index": 1, "tool": "unifi_get_client_details", "jobId": "bf44c344dcde971d"}
+  ],
+  "message": "Started 2 operation(s). Use unifi_batch_status to check progress."
 }
 ```
 
-**Response (completed):**
+**Check batch status:**
 ```json
 {
-  "status": "done",
-  "result": {...},
-  "started": 1234567890.123,
-  "finished": 1234567891.456
+  "name": "unifi_batch_status",
+  "arguments": {
+    "jobIds": ["af33b233cbdc860c", "bf44c344dcde971d"]
+  }
 }
 ```
 
-**Response (error):**
+**Response:**
 ```json
 {
-  "status": "error",
-  "error": "Device not found",
-  "started": 1234567890.123,
-  "finished": 1234567891.456
+  "jobs": [
+    {"jobId": "af33b233cbdc860c", "status": "done", "result": {...}},
+    {"jobId": "bf44c344dcde971d", "status": "done", "result": {...}}
+  ]
 }
 ```
 
 **Notes:**
+- Use `unifi_execute` for single operations (returns result directly)
+- Use `unifi_batch` + `unifi_batch_status` for parallel/bulk operations
 - Jobs are stored in-memory only (no persistence)
 - Job IDs are unique per server session
-- Failed jobs capture exception details in the `error` field
 
 ### Using with Claude Desktop
 
@@ -344,22 +345,32 @@ Practical examples showing programmatic usage:
 ```python
 from mcp import ClientSession, stdio_client
 
-# Query tool index
-result = await session.call_tool("unifi_tool_index", {})
+# Discover tools
+tools = await session.call_tool("unifi_tool_index", {})
+
+# Execute a single tool (returns result directly)
+result = await session.call_tool("unifi_execute", {
+    "tool": "unifi_list_clients",
+    "arguments": {}
+})
 
-# Use async jobs for long operations
-job = await session.call_tool("unifi_async_start", {
-    "tool": "unifi_upgrade_device",
-    "arguments": {"mac_address": "...", "confirm": True}
+# Batch execution for parallel operations
+batch = await session.call_tool("unifi_batch", {
+    "operations": [
+        {"tool": "unifi_get_client_details", "arguments": {"mac": "..."}},
+        {"tool": "unifi_get_device_details", "arguments": {"mac": "..."}}
+    ]
 })
 
-# Check job status
-status = await session.call_tool("unifi_async_status", {"jobId": job["jobId"]})
+# Check batch status
+status = await session.call_tool("unifi_batch_status", {
+    "jobIds": [j["jobId"] for j in batch["jobs"]]
+})
 ```
 
 **Three complete examples:**
 - `query_tool_index.py` - Discover available tools
-- `use_async_jobs.py` - Background job management
+- `use_async_jobs.py` - Batch operations and status checking
 - `programmatic_client.py` - Build custom Python clients
 
 See [`examples/python/README.md`](examples/python/README.md) for complete examples.
@@ -376,15 +387,18 @@ The server advertises its capabilities via an MCP identity file at [`.well-known
   "capabilities": {
     "tools": true,
     "tool_index": true,
-    "async_operations": true
+    "batch_operations": true
   },
   "features": {
     "tool_index": {
       "tool": "unifi_tool_index"
     },
-    "async_operations": {
-      "start_tool": "unifi_async_start",
-      "status_tool": "unifi_async_status"
+    "execution": {
+      "tool": "unifi_execute"
+    },
+    "batch_operations": {
+      "start_tool": "unifi_batch",
+      "status_tool": "unifi_batch_status"
     }
   }
 }
@@ -504,6 +518,44 @@ The server merges settings from **environment variables**, an optional `.env` fi
 | `UNIFI_VERIFY_SSL` | Set to `false` if using self-signed certs |
 | `UNIFI_CONTROLLER_TYPE` | Controller API path type: `auto` (detect), `proxy` (UniFi OS), `direct` (standalone). Default `auto` |
 | `UNIFI_MCP_HTTP_ENABLED` | Set `true` to enable optional HTTP SSE server (default `false`) |
+| `UNIFI_AUTO_CONFIRM` | Set `true` to auto-confirm all mutating operations (skips preview step). Ideal for workflow automation (n8n, Make, Zapier). Default `false` |
+| `UNIFI_TOOL_REGISTRATION_MODE` | Tool loading mode: `lazy` (default), `eager`, or `meta_only`. See [Context Optimization](#context-optimization) |
+| `UNIFI_ENABLED_CATEGORIES` | Comma-separated list of tool categories to load (eager mode). See table below |
+| `UNIFI_ENABLED_TOOLS` | Comma-separated list of specific tool names to register (eager mode) |
+
+### Tool Categories (for UNIFI_ENABLED_CATEGORIES)
+
+When using eager mode with category filtering, these are the valid category names:
+
+| Category | Description | Example Tools |
+|----------|-------------|---------------|
+| `clients` | Client listing, blocking, guest auth | `unifi_list_clients`, `unifi_block_client` |
+| `config` | Configuration management | - |
+| `devices` | Device listing, reboot, locate, upgrade | `unifi_list_devices`, `unifi_reboot_device` |
+| `events` | Events and alarms | `unifi_list_events`, `unifi_list_alarms` |
+| `firewall` | Firewall rules and groups | `unifi_list_firewall_rules`, `unifi_create_firewall_rule` |
+| `hotspot` | Vouchers for guest network | `unifi_list_vouchers`, `unifi_create_voucher` |
+| `network` | Network/VLAN management | `unifi_list_networks`, `unifi_create_network` |
+| `port_forwards` | Port forwarding rules | `unifi_list_port_forwards` |
+| `qos` | QoS/traffic shaping rules | `unifi_list_qos_rules`, `unifi_create_qos_rule` |
+| `routing` | Static routes (V1 API) | `unifi_list_routes`, `unifi_create_route` |
+| `stats` | Statistics and metrics | `unifi_get_client_stats`, `unifi_get_device_stats` |
+| `system` | System info, health, settings | `unifi_get_system_info`, `unifi_get_network_health` |
+| `traffic_routes` | Policy-based routing (V2 API) | `unifi_list_traffic_routes` |
+| `usergroups` | Bandwidth profiles/user groups | `unifi_list_usergroups`, `unifi_create_usergroup` |
+| `vpn` | VPN servers and clients | `unifi_list_vpn_servers`, `unifi_list_vpn_clients` |
+
+**Example usage:**
+```bash
+# Load only client and system tools
+export UNIFI_TOOL_REGISTRATION_MODE=eager
+export UNIFI_ENABLED_CATEGORIES=clients,system
+
+# Or load specific tools only
+export UNIFI_ENABLED_TOOLS=unifi_list_clients,unifi_list_devices,unifi_get_system_info
+```
+
+**Note:** Tools may also be filtered by the `permissions` section in config.yaml (e.g., `clients.update: false` blocks mutating client tools).
 
 ### Controller Type Detection
 
@@ -826,6 +878,37 @@ See [docs/permissions.md](docs/permissions.md) for complete documentation includ
 * `unifi_force_reconnect_client`
 * `unifi_authorize_guest`
 * `unifi_unauthorize_guest`
+* `unifi_set_client_ip_settings`
+
+### Events & Alarms
+
+* `unifi_list_events`
+* `unifi_list_alarms`
+* `unifi_archive_alarm`
+* `unifi_archive_all_alarms`
+* `unifi_get_event_types`
+
+### Routing (Static Routes)
+
+* `unifi_list_routes`
+* `unifi_get_route_details`
+* `unifi_create_route`
+* `unifi_update_route`
+* `unifi_list_active_routes`
+
+### Hotspot (Vouchers)
+
+* `unifi_list_vouchers`
+* `unifi_get_voucher_details`
+* `unifi_create_voucher`
+* `unifi_revoke_voucher`
+
+### User Groups
+
+* `unifi_list_usergroups`
+* `unifi_get_usergroup_details`
+* `unifi_create_usergroup`
+* `unifi_update_usergroup`
 
 ### Statistics & Alerts
 
@@ -1002,9 +1085,9 @@ uv run python devtools/dev_console.py
 
 # You'll see a menu of all tools including:
 # - unifi_tool_index (list all tools with schemas)
-# - unifi_async_start (start background jobs)
-# - unifi_async_status (check job status)
-# - All 60+ UniFi tools (clients, devices, networks, etc.)
+# - unifi_execute (run any discovered tool)
+# - unifi_batch / unifi_batch_status (parallel operations)
+# - All 80+ UniFi tools (clients, devices, networks, etc.)
 ```
 
 **4. Test with Python examples:**

{unifi_network_mcp-0.2.0 → unifi_network_mcp-0.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unifi-network-mcp"
-version = "0.2.0"
+version = "0.3.1"
 description = "Unifi Network MCP Server"
 readme = "README.md"
 requires-python = ">=3.13"
@@ -58,3 +58,15 @@ dev-dependencies = [
 [project.scripts]
 # CLI entrypoint that launches the UniFi Network MCP server
 unifi-network-mcp = "src.main:main"
+
+[tool.ruff]
+line-length = 120
+
+[tool.ruff.lint]
+select = ["E", "F", "I"]
+ignore = [
+    "E501",  # Line too long - handled by line-length setting above
+]
+
+[tool.ruff.lint.isort]
+known-first-party = ["src"]

{unifi_network_mcp-0.2.0 → unifi_network_mcp-0.3.1}/src/bootstrap.py

@@ -12,13 +12,13 @@ Importing it early guarantees deterministic side‑effects (env + logging) and
 exposes a `load_config()` helper that the rest of the codebase can share.
 """
 
-from dataclasses import dataclass
+import importlib.resources
 import logging
 import os
 import sys
+from dataclasses import dataclass
 from pathlib import Path
 from typing import Any
-import importlib.resources
 
 from dotenv import load_dotenv
 from omegaconf import OmegaConf
@@ -118,32 +118,20 @@ def load_config(path_override: str | Path | None = None) -> OmegaConf:
         relative_path = Path("config/config.yaml")
         if relative_path.exists() and relative_path.is_file():
             resolved_path = relative_path
-            logger.info(
-                "Using configuration file from relative path: %s", relative_path
-            )
+            logger.info("Using configuration file from relative path: %s", relative_path)
         else:
             # 3. Use bundled default config
             try:
                 # Use importlib.resources to safely access package data
-                config_file_ref = importlib.resources.files("src.config").joinpath(
-                    "config.yaml"
-                )
+                config_file_ref = importlib.resources.files("src.config").joinpath("config.yaml")
                 if config_file_ref.is_file():
-                    resolved_path = Path(
-                        str(config_file_ref)
-                    )  # Convert Traversable to Path
-                    logger.info(
-                        "Using bundled default configuration: %s", resolved_path
-                    )
+                    resolved_path = Path(str(config_file_ref))  # Convert Traversable to Path
+                    logger.info("Using bundled default configuration: %s", resolved_path)
                 else:
-                    logger.error(
-                        "Bundled default configuration file could not be accessed (not a file)."
-                    )
+                    logger.error("Bundled default configuration file could not be accessed (not a file).")
                     raise SystemExit(3)  # Exit if bundled config isn't a file
             except (ModuleNotFoundError, FileNotFoundError, Exception) as e:
-                logger.error(
-                    "Could not find or access bundled default configuration: %s", e
-                )
+                logger.error("Could not find or access bundled default configuration: %s", e)
                 raise SystemExit(3)  # Exit if bundled config cannot be loaded
 
     if resolved_path is None: