diff --git a/gns3server/__init__.py b/gns3server/__init__.py index 66dd93d3f..b338903ce 100644 --- a/gns3server/__init__.py +++ b/gns3server/__init__.py @@ -14,4 +14,4 @@ # You should have received a copy of the GNU General Public License # along with this program. If not, see . -from .version import __version__ +from .version import __version__ as __version__ diff --git a/gns3server/agent/gns3_copilot/skills/__init__.py b/gns3server/agent/gns3_copilot/skills/__init__.py index 891389d12..40dbd4476 100644 --- a/gns3server/agent/gns3_copilot/skills/__init__.py +++ b/gns3server/agent/gns3_copilot/skills/__init__.py @@ -66,6 +66,7 @@ __all__ = [ "get_skill", "get_skills_manager", "get_skills_repository_info", + "reload_forbidden_commands", "reload_injection_skills", "reload_skills_repository", "set_skills_manager", diff --git a/gns3server/agent/gns3_copilot/utils/custom_netmiko/__init__.py b/gns3server/agent/gns3_copilot/utils/custom_netmiko/__init__.py index 9cad247ca..30eb84bc4 100644 --- a/gns3server/agent/gns3_copilot/utils/custom_netmiko/__init__.py +++ b/gns3server/agent/gns3_copilot/utils/custom_netmiko/__init__.py @@ -66,7 +66,7 @@ except Exception as e: logger.warning(f"Failed to import Huawei CE driver: {e}", exc_info=True) try: - from .ruijie_telnet import RuijieTelnetEnhanced + from .ruijie_telnet import RuijieTelnetEnhanced as RuijieTelnetEnhanced except Exception as e: logger.warning(f"Failed to import Ruijie driver: {e}", exc_info=True) diff --git a/gns3server/agent/mcp/__init__.py b/gns3server/agent/mcp/__init__.py index f55cdfc6e..f52b05552 100644 --- a/gns3server/agent/mcp/__init__.py +++ b/gns3server/agent/mcp/__init__.py @@ -41,10 +41,8 @@ from fastapi.responses import Response from mcp.server.fastmcp import FastMCP from mcp.server.transport_security import TransportSecuritySettings from pydantic import Field -from sqlalchemy import select from sqlalchemy.ext.asyncio import AsyncSession -import gns3server.db.models as models from gns3server.agent.gns3_copilot.gns3_client.api_handlers import ( available_filters_handler, create_link_handler, @@ -80,10 +78,11 @@ from gns3server.agent.gns3_copilot.gns3_client.api_handlers import ( write_node_file_handler, ) from gns3server.config import Config +from gns3server.db import models as models from gns3server.db.repositories.api_keys import ApiKeysRepository from gns3server.db.repositories.users import UsersRepository from gns3server.services import auth_service -from gns3server.services.authentication import AuthService +from gns3server.services.authentication import AuthService as AuthService from gns3server.utils.request_utils import extract_client_info # Symbol tools are disabled for now: they require a vision-capable model to diff --git a/gns3server/compute/iou/__init__.py b/gns3server/compute/iou/__init__.py index b89381f06..85ffb495a 100644 --- a/gns3server/compute/iou/__init__.py +++ b/gns3server/compute/iou/__init__.py @@ -23,7 +23,7 @@ import logging import os from ..base_manager import BaseManager -from .iou_error import IOUError +from .iou_error import IOUError as IOUError from .iou_vm import IOUVM log = logging.getLogger(__name__) diff --git a/gns3server/compute/vpcs/__init__.py b/gns3server/compute/vpcs/__init__.py index 15ab54f3e..a17e5638a 100644 --- a/gns3server/compute/vpcs/__init__.py +++ b/gns3server/compute/vpcs/__init__.py @@ -18,7 +18,6 @@ VPCS server module. """ -import asyncio import os from ..base_manager import BaseManager diff --git a/gns3server/controller/__init__.py b/gns3server/controller/__init__.py index f2629065c..67cef57f8 100644 --- a/gns3server/controller/__init__.py +++ b/gns3server/controller/__init__.py @@ -40,7 +40,7 @@ from ..utils import md5sum, parse_version from ..utils.asyncio import wait_run_in_executor from ..utils.images import default_images_directory from ..version import __version__ -from .appliance import Appliance +from .appliance import Appliance as Appliance from .appliance_manager import ApplianceManager from .compute import Compute, ComputeError from .controller_error import ControllerError, ControllerNotFoundError diff --git a/gns3server/db/models/__init__.py b/gns3server/db/models/__init__.py index cf6acc92f..ee4a11122 100644 --- a/gns3server/db/models/__init__.py +++ b/gns3server/db/models/__init__.py @@ -38,3 +38,29 @@ from .templates import ( VPCSTemplate, ) from .users import User, UserGroup + +__all__ = [ + "ACE", + "ApiKey", + "Base", + "CloudTemplate", + "Compute", + "DockerTemplate", + "DynamipsTemplate", + "EthernetHubTemplate", + "EthernetSwitchTemplate", + "IOUTemplate", + "Image", + "LLMModelConfig", + "Privilege", + "QemuTemplate", + "Resource", + "ResourcePool", + "Role", + "Template", + "User", + "UserGroup", + "VMwareTemplate", + "VPCSTemplate", + "VirtualBoxTemplate", +] diff --git a/gns3server/schemas/__init__.py b/gns3server/schemas/__init__.py index 34939c954..e88623cde 100644 --- a/gns3server/schemas/__init__.py +++ b/gns3server/schemas/__init__.py @@ -145,3 +145,177 @@ from .controller.tokens import ApiKeyCreate, RefreshTokenRequest, Token # Schemas for both controller and compute from .qemu_disk_image import QemuDiskImageCreate, QemuDiskImageFormat, QemuDiskImageUpdate + +__all__ = [ + "ACE", + "GNS3VM", + "IOU", + "NAT", + "TAPNIO", + "UDPNIO", + "VPCS", + "ACECreate", + "ACEUpdate", + "ATMSwitch", + "ATMSwitchCreate", + "ATMSwitchUpdate", + "ApiKeyCreate", + "Appliance", + "ApplianceVersion", + "ApplianceVersionV8", + "AutoIdlePC", + "BatchNIOCreate", + "BatchNIOEntry", + "C1700DynamipsTemplate", + "C1700DynamipsTemplateUpdate", + "C2600DynamipsTemplate", + "C2600DynamipsTemplateUpdate", + "C2691DynamipsTemplate", + "C2691DynamipsTemplateUpdate", + "C3600DynamipsTemplate", + "C3600DynamipsTemplateUpdate", + "C3725DynamipsTemplate", + "C3725DynamipsTemplateUpdate", + "C3745DynamipsTemplate", + "C3745DynamipsTemplateUpdate", + "C7200DynamipsTemplate", + "C7200DynamipsTemplateUpdate", + "Capabilities", + "ChatRequest", + "ChatResponse", + "ChatSession", + "Cloud", + "CloudCreate", + "CloudTemplate", + "CloudTemplateUpdate", + "CloudUpdate", + "Compute", + "ComputeCreate", + "ComputeDockerImage", + "ComputeUpdate", + "ComputeVMwareVM", + "ComputeVirtualBoxVM", + "ConversationHistory", + "Credentials", + "Docker", + "DockerCreate", + "DockerTemplate", + "DockerTemplateUpdate", + "DockerUpdate", + "Drawing", + "Dynamips", + "DynamipsCreate", + "DynamipsTemplate", + "DynamipsUpdate", + "ErrorMessage", + "EthernetHub", + "EthernetHubCreate", + "EthernetHubTemplate", + "EthernetHubTemplateUpdate", + "EthernetHubUpdate", + "EthernetNIO", + "EthernetPortInfo", + "EthernetSwitch", + "EthernetSwitchCreate", + "EthernetSwitchTemplate", + "EthernetSwitchTemplateUpdate", + "EthernetSwitchUpdate", + "FrameRelaySwitch", + "FrameRelaySwitchCreate", + "FrameRelaySwitchUpdate", + "IOUCreate", + "IOULicense", + "IOUStart", + "IOUTemplate", + "IOUTemplateUpdate", + "IOUUpdate", + "Image", + "ImageType", + "LLMModelConfigCreate", + "LLMModelConfigData", + "LLMModelConfigInheritedResponse", + "LLMModelConfigListResponse", + "LLMModelConfigResponse", + "LLMModelConfigUpdate", + "LLMModelConfigWithSource", + "Link", + "LinkCapture", + "LinkCreate", + "LinkUpdate", + "LoggedInUserUpdate", + "MarkerCreate", + "MarkerDefinitionCreate", + "MarkerRebuild", + "MarkerToggle", + "MarkerUpdate", + "NATCreate", + "NATUpdate", + "NetmikoDeviceType", + "NetmikoDeviceTypeList", + "Node", + "NodeCapture", + "NodeCreate", + "NodeDuplicate", + "NodeFile", + "NodeUpdate", + "OpenAIMessage", + "OpenAIToolCall", + "Privilege", + "Project", + "ProjectCompression", + "ProjectCreate", + "ProjectDuplicate", + "ProjectFile", + "ProjectUpdate", + "Qemu", + "QemuCreate", + "QemuDiskImageCreate", + "QemuDiskImageFormat", + "QemuDiskImageUpdate", + "QemuTemplate", + "QemuTemplateUpdate", + "QemuUpdate", + "RefreshTokenRequest", + "RenameSession", + "Resource", + "ResourceCreate", + "ResourcePool", + "ResourcePoolCreate", + "ResourcePoolUpdate", + "Role", + "RoleCreate", + "RoleUpdate", + "ServerConfig", + "SettingsResponse", + "SettingsUpdate", + "SettingsUpdateResponse", + "Snapshot", + "SnapshotCreate", + "Template", + "TemplateCreate", + "TemplateUpdate", + "TemplateUsage", + "Token", + "UDPPortInfo", + "User", + "UserCreate", + "UserGroup", + "UserGroupCreate", + "UserGroupUpdate", + "UserUpdate", + "VMware", + "VMwareCreate", + "VMwareTemplate", + "VMwareTemplateUpdate", + "VMwareUpdate", + "VPCSCreate", + "VPCSTemplate", + "VPCSTemplateUpdate", + "VPCSUpdate", + "Version", + "VirtualBox", + "VirtualBoxCreate", + "VirtualBoxTemplate", + "VirtualBoxTemplateUpdate", + "VirtualBoxUpdate", +] diff --git a/gns3server/utils/__init__.py b/gns3server/utils/__init__.py index 44c224ff7..286a76757 100644 --- a/gns3server/utils/__init__.py +++ b/gns3server/utils/__init__.py @@ -20,9 +20,7 @@ import errno import hashlib import posixpath import re -import shlex import socket -import sys import textwrap diff --git a/gns3server/utils/zipfile_zstd/__init__.py b/gns3server/utils/zipfile_zstd/__init__.py index 687a808b0..633266150 100644 --- a/gns3server/utils/zipfile_zstd/__init__.py +++ b/gns3server/utils/zipfile_zstd/__init__.py @@ -1,8 +1,5 @@ # NOTE: this patches the standard zipfile module from zipfile import * -from . import _zipfile -from ._zipfile import ( - ZIP_ZSTANDARD, - ZSTANDARD_VERSION, -) +from . import _zipfile # noqa: F401 +from ._zipfile import ZIP_ZSTANDARD, ZSTANDARD_VERSION # noqa: F401