annet 3.1.4__py3-none-any.whl → 3.3.0__py3-none-any.whl

annet/adapters/netbox/v42/models.py

@@ -3,6 +3,7 @@ from datetime import datetime, timezone
 from typing import Optional
 from annet.adapters.netbox.common.models import InterfaceType, IpFamily, Label, Prefix, Entity
 from annet.adapters.netbox.v41.models import InterfaceV41, IpAddressV41, NetboxDeviceV41
+import annetbox.v42.models
 
 
 @dataclass
@@ -61,3 +62,8 @@ class NetboxDeviceV42(NetboxDeviceV41):
             connected_endpoints=[],
             mode=None,
         )
+
+
+# should be unified in case of netbox update
+Vrf = annetbox.v42.models.Vrf
+Vlan = annetbox.v42.models.Vlan

annet/adapters/netbox/v42/storage.py

@@ -1,4 +1,6 @@
 import ssl
+from typing import Optional
+
 from adaptix import P
 from adaptix.conversion import get_converter, link, link_constant, link_function
 from annetbox.v42 import client_sync
@@ -6,7 +8,8 @@ from annetbox.v42 import models as api_models
 
 from annet.adapters.netbox.common.adapter import NetboxAdapter, get_device_breed, get_device_hw
 from annet.adapters.netbox.common.storage_base import BaseNetboxStorage
-from annet.adapters.netbox.v42.models import InterfaceV42, NetboxDeviceV42, PrefixV42, IpAddressV42
+from annet.adapters.netbox.common.storage_opts import NetboxStorageOpts
+from annet.adapters.netbox.v42.models import InterfaceV42, NetboxDeviceV42, PrefixV42, IpAddressV42, Vlan, Vrf
 from annet.storage import Storage
 
 
@@ -79,8 +82,21 @@ class NetboxV42Adapter(NetboxAdapter[NetboxDeviceV42, InterfaceV42, IpAddressV42
     def list_ipprefixes(self, prefixes: list[str]) -> list[PrefixV42]:
         return self.convert_ip_prefixes(self.netbox.ipam_all_prefixes(prefix=prefixes).results)
 
+    def list_all_vrfs(self) -> list[Vrf]:
+        return self.netbox.ipam_all_vrfs().results
+
+    def list_all_vlans(self) -> list[Vlan]:
+        return self.netbox.ipam_all_vlans().results
+
 
 class NetboxStorageV42(BaseNetboxStorage[NetboxDeviceV42, InterfaceV42, IpAddressV42, PrefixV42]):
+    netbox: NetboxV42Adapter
+
+    def __init__(self, opts: Optional[NetboxStorageOpts] = None):
+        super().__init__(opts)
+        self._all_vlans: list[Vlan] | None = None
+        self._all_vrfs: list[Vrf] | None = None
+
     def _init_adapter(
             self,
             url: str,
@@ -89,3 +105,13 @@ class NetboxStorageV42(BaseNetboxStorage[NetboxDeviceV42, InterfaceV42, IpAddres
             threads: int,
     ) -> NetboxAdapter[NetboxDeviceV42, InterfaceV42, IpAddressV42, PrefixV42]:
         return NetboxV42Adapter(self, url, token, ssl_context, threads)
+
+    def resolve_all_vlans(self) -> list[Vlan]:
+        if self._all_vlans is None:
+            self._all_vlans = self.netbox.list_all_vlans()
+        return self._all_vlans
+
+    def resolve_all_vrfs(self) -> list[Vrf]:
+        if self._all_vrfs is None:
+            self._all_vrfs = self.netbox.list_all_vrfs()
+        return self._all_vrfs

annet/annlib/rulebook/common.py

@@ -2,9 +2,9 @@ import os
 import typing
 from collections import OrderedDict as odict
 
-from annet.annlib.command import CommandList, Command
-
+from annet.annlib.command import Command, CommandList
 from annet.types import Op
+from annet.vendors import registry_connector
 
 
 # =====
@@ -287,87 +287,5 @@ class ApplyItem(typing.NamedTuple):
     after: CommandList
 
 
-def apply(hw, do_commit, do_finalize, **_):
-    before, after = CommandList(), CommandList()
-    if hw.Huawei:
-        before.add_cmd(Command("system-view"))
-        if do_commit and (hw.Huawei.CE or hw.Huawei.NE):
-            after.add_cmd(Command("commit"))
-        after.add_cmd(Command("q"))
-        if do_finalize:
-            after.add_cmd(Command("save", timeout=20))
-    elif hw.Arista:
-        before.add_cmd(Command("conf s"))
-        if do_commit:
-            after.add_cmd(Command("commit"))
-        else:
-            after.add_cmd(Command("abort"))  # просто exit оставит висеть configure session
-        if do_finalize:
-            after.add_cmd(Command("write memory"))
-    elif hw.ASR or hw.XRV or hw.XR:
-        # коммит сам сохраняет изменения в стартап do_finalize не применим
-        before.add_cmd(Command("configure exclusive"))
-        if do_commit:
-            after.add_cmd(Command("commit"))
-        after.add_cmd(Command("exit"))
-    elif hw.Cisco or hw.Nexus:
-        # классический ios и nxos не умеет коммиты
-        before.add_cmd(Command("conf t"))
-        after.add_cmd(Command("exit"))
-        if do_finalize:
-            after.add_cmd(Command("copy running-config startup-config", timeout=40))
-    elif hw.Juniper:
-        # коммит сам сохраняет изменения в стартап do_finalize не применим
-        before.add_cmd(Command("configure exclusive"))
-        if do_commit:
-            after.add_cmd(Command("commit", timeout=30))
-        after.add_cmd(Command("exit"))
-    elif hw.PC:
-        if hw.soft.startswith(("Cumulus", "SwitchDev")):
-            if os.environ.get("ETCKEEPER_CHECK", False):
-                before.add_cmd(Command("etckeeper check"))
-    elif hw.Nokia:
-        # коммит сам сохраняет изменения в стартап do_finalize не применим
-        before.add_cmd(Command("configure private"))
-        if do_commit:
-            after.add_cmd(Command("commit"))
-    elif hw.RouterOS:
-        # FIXME: пока не удалось победить \x1b[c после включения safe mode
-        # if len(cmds) > 99:
-        #     raise Exception("RouterOS does not support more 100 actions in safe mode")
-        # before.add_cmd(RosDevice.SAFE_MODE)
-        pass
-        # after.add_cmd(RosDevice.SAFE_MODE)
-    elif hw.Aruba:
-        before.add_cmd(Command("conf t"))
-        after.add_cmd(Command("end"))
-        if do_commit:
-            after.add_cmd(Command("commit apply"))
-        if do_finalize:
-            after.add_cmd(Command("write memory"))
-    elif hw.Ribbon:
-        # коммит сам сохраняет изменения в стартап do_finalize не применим
-        before.add_cmd(Command("configure exclusive"))
-        if do_commit:
-            after.add_cmd(Command("commit", timeout=30))
-        after.add_cmd(Command("exit"))
-    elif hw.B4com.CS2148P:
-        before.add_cmd(Command("conf t"))
-        after.add_cmd(Command("end"))
-        if do_finalize:
-            after.add_cmd(Command("write", timeout=40))
-    elif hw.B4com:
-        before.add_cmd(Command("conf t"))
-        if do_commit:
-            after.add_cmd(Command("commit"))
-            after.add_cmd(Command("end"))
-        if do_finalize:
-            after.add_cmd(Command("write", timeout=40))
-    elif hw.H3C:
-        before.add_cmd(Command("system-view"))
-        if do_finalize:
-            after.add_cmd(Command("save force", timeout=90))
-    else:
-        raise Exception("unknown hw %s" % hw)
-
-    return before, after
+def apply(hw, do_commit, do_finalize, path):
+    return registry_connector.get().match(hw).apply(hw, do_commit, do_finalize, path)

annet/generators/__init__.py

@@ -157,7 +157,7 @@ def run_partial_generators(
 
 
 @tracing.function(name="run_partial_generator")
-def _run_partial_generator(gen: "PartialGenerator", run_args: GeneratorPartialRunArgs) -> Optional[GeneratorPartialResult]:
+def _run_partial_generator(gen: "PartialGenerator", run_args: GeneratorPartialRunArgs) -> GeneratorPartialResult | None:
     logger = get_logger(generator=_make_generator_ctx(gen))
     device = run_args.device
     output = ""
@@ -207,7 +207,9 @@ def _run_partial_generator(gen: "PartialGenerator", run_args: GeneratorPartialRu
 
     if run_args.use_acl:
         try:
-            with tracing_connector.get().start_as_current_span("apply_acl", tracer_name=__name__, min_duration="0.01") as acl_span:
+            with tracing_connector.get().start_as_current_span(
+                "apply_acl", tracer_name=__name__, min_duration="0.01"
+            ) as acl_span:
                 tracing_connector.get().set_device_attributes(acl_span, run_args.device)
                 config = patching.apply_acl(
                     config=config,
@@ -265,8 +267,8 @@ def check_entire_generators_required_packages(gens, device_packages: FrozenSet[s
 
 @tracing.function
 def run_file_generators(
-        gens: Iterable[Union["JSONFragment", "Entire"]],
-        device: "Device",
+    gens: Iterable[Union["JSONFragment", "Entire"]],
+    device: "Device",
 ) -> RunGeneratorResult:
     """Run generators that generate files or file parts."""
     ret = RunGeneratorResult()
@@ -295,21 +297,22 @@ def run_file_generators(
 @tracing.function(min_duration="0.5")
 def _run_entire_generator(gen: "Entire", device: "Device") -> Optional[GeneratorResult]:
     logger = get_logger(generator=_make_generator_ctx(gen))
-    if not gen.supports_device(device):
-        logger.info("generator %s is not supported for device %s", gen, device.hostname)
-        return
-
     span = tracing_connector.get().get_current_span()
     if span:
         tracing_connector.get().set_device_attributes(span, device)
         tracing_connector.get().set_dimensions_attributes(span, gen, device)
 
-    path = gen.path(device)
-    if not path:
-        raise RuntimeError("entire generator should return non-empty path")
-
-    logger.info("Generating ENTIRE ...")
     with GeneratorPerfMesurer(gen, trace_min_duration="0.5") as pm:
+        if not gen.supports_device(device):
+            logger.debug("generator %s is not supported for device %s", gen, device.hostname)
+            return
+
+        path = gen.path(device)
+        if not path:
+            raise RuntimeError("entire generator should return non-empty path")
+
+        logger.info("Generating ENTIRE ...")
+
         output = gen(device)
 
     return GeneratorEntireResult(
@@ -329,35 +332,37 @@ def _make_generator_ctx(gen):
 
 
 def _run_json_fragment_generator(
-        gen: "JSONFragment",
-        device: "Device",
+    gen: "JSONFragment",
+    device: "Device",
 ) -> Optional[GeneratorResult]:
     logger = get_logger(generator=_make_generator_ctx(gen))
-    if not gen.supports_device(device):
-        logger.info("generator %s is not supported for device %s", gen, device.hostname)
-        return
-
-    path = gen.path(device)
-    if not path:
-        raise RuntimeError("json fragment generator should return non-empty path")
-
-    acl_item_or_list_of_items = gen.acl(device)
-    safe_acl_item_or_list_of_items = gen.acl_safe(device)
-    if not acl_item_or_list_of_items:
-        raise RuntimeError("json fragment generator should return non-empty acl")
-    if isinstance(acl_item_or_list_of_items, list):
-        acl = acl_item_or_list_of_items
-    else:
-        acl = [acl_item_or_list_of_items]
-    if isinstance(safe_acl_item_or_list_of_items, list):
-        acl_safe = safe_acl_item_or_list_of_items
-    else:
-        acl_safe = [safe_acl_item_or_list_of_items]
-
-    logger.info("Generating JSON_FRAGMENT ...")
+
     with GeneratorPerfMesurer(gen) as pm:
+        if not gen.supports_device(device):
+            logger.info("generator %s is not supported for device %s", gen, device.hostname)
+            return
+
+        path = gen.path(device)
+        if not path:
+            raise RuntimeError("json fragment generator should return non-empty path")
+
+        acl_item_or_list_of_items = gen.acl(device)
+        safe_acl_item_or_list_of_items = gen.acl_safe(device)
+        if not acl_item_or_list_of_items:
+            raise RuntimeError("json fragment generator should return non-empty acl")
+        if isinstance(acl_item_or_list_of_items, list):
+            acl = acl_item_or_list_of_items
+        else:
+            acl = [acl_item_or_list_of_items]
+        if isinstance(safe_acl_item_or_list_of_items, list):
+            acl_safe = safe_acl_item_or_list_of_items
+        else:
+            acl_safe = [safe_acl_item_or_list_of_items]
+
+        logger.info("Generating JSON_FRAGMENT ...")
+
         config = gen(device)
-    reload_cmds = gen.get_reload_cmds(device)
+        reload_cmds = gen.get_reload_cmds(device)
     return GeneratorJSONFragmentResult(
         name=gen.__class__.__name__,
         tags=gen.TAGS,
@@ -407,7 +412,8 @@ def _load_gen_module(module_path: str):
     except ModuleNotFoundError as e:
         try:  # maybe it's a path to module
             module_abs_path = os.path.abspath(module_path)
-            module = importlib.machinery.SourceFileLoader(re.sub(r"[./]", "_", module_abs_path).strip("_"), module_abs_path).load_module()
+            module = importlib.machinery.SourceFileLoader(re.sub(r"[./]", "_", module_abs_path).strip("_"),
+                                                          module_abs_path).load_module()
         except ModuleNotFoundError:
             raise e
     return module

annet/generators/entire.py

@@ -3,19 +3,10 @@ from __future__ import annotations
 import pkgutil
 import re
 import types
-from typing import (
-    FrozenSet,
-    Iterable,
-    List,
-    Optional,
-    Set,
-    Union,
-)
-
-from annet.lib import (
-    flatten,
-    mako_render,
-)
+from typing import FrozenSet, Iterable, List, Optional, Set, Union
+
+from annet.lib import flatten, mako_render
+
 from .base import BaseGenerator, _filter_str
 from .exceptions import NotSupportedDevice
 
@@ -39,8 +30,7 @@ class Entire(BaseGenerator):
     def run(self, device) -> Union[None, str, Iterable[Union[str, tuple]]]:
         raise NotImplementedError
 
-    def reload(self, device) -> Optional[
-        str]:  # pylint: disable=unused-argument
+    def reload(self, device) -> Optional[str]:  # pylint: disable=unused-argument
         return
 
     def get_reload_cmds(self, device) -> str:

annet/generators/perf.py

@@ -1,27 +1,19 @@
-from __future__ import annotations
-
 import time
-from typing import (
-    Optional,
-    Union,
-)
+from typing import Optional, Union
 
 from annet import tracing
 from annet.tracing import tracing_connector
-from annet.types import (
-    GeneratorPartialRunArgs,
-    GeneratorPerf,
-)
-from .entire import Entire
-from .partial import PartialGenerator
+from annet.types import GeneratorPartialRunArgs, GeneratorPerf
+
+from .base import BaseGenerator
 
 
 class GeneratorPerfMesurer:
     def __init__(
-            self,
-            gen: Union[PartialGenerator, Entire],
-            run_args: Optional[GeneratorPartialRunArgs] = None,
-            trace_min_duration: tracing.MinDurationT = None
+        self,
+        gen: BaseGenerator,
+        run_args: Optional[GeneratorPartialRunArgs] = None,
+        trace_min_duration: tracing.MinDurationT = None
     ):
         self._gen = gen
         self._run_args = run_args
@@ -44,8 +36,7 @@ class GeneratorPerfMesurer:
         self._span = self._span_ctx.__enter__()  # pylint: disable=unnecessary-dunder-call
 
         if self._span:
-            self._span.set_attributes(
-                {"generator.class": self._gen.__class__.__name__})
+            self._span.set_attributes({"generator.class": self._gen.__class__.__name__})
             if self._run_args:
                 tracing_connector.get().set_device_attributes(
                     self._span, self._run_args.device,

annet/vendors/base.py

@@ -1,6 +1,7 @@
 import abc
 from typing import ClassVar
 
+from annet.annlib.command import CommandList
 from annet.annlib.netdev.views.hardware import HardwareView
 from annet.annlib.tabparser import CommonFormatter
 
@@ -12,6 +13,11 @@ class AbstractVendor(abc.ABC):
     def match(self) -> list[str]:
         raise NotImplementedError
 
+    def apply(
+        self, hw: HardwareView, do_commit: bool, do_finalize: bool, path: str
+    ) -> tuple[CommandList, CommandList]:
+        return CommandList(), CommandList()
+
     @property
     @abc.abstractmethod
     def reverse(self) -> str:

annet/vendors/library/arista.py

@@ -1,3 +1,4 @@
+from annet.annlib.command import Command, CommandList
 from annet.annlib.netdev.views.hardware import HardwareView
 from annet.annlib.tabparser import AristaFormatter
 from annet.vendors.base import AbstractVendor
@@ -8,6 +9,19 @@ from annet.vendors.registry import registry
 class AristaVendor(AbstractVendor):
     NAME = "arista"
 
+    def apply(self, hw: HardwareView, do_commit: bool, do_finalize: bool, path: str) -> tuple[CommandList, CommandList]:
+        before, after = CommandList(), CommandList()
+
+        before.add_cmd(Command("conf s"))
+        if do_commit:
+            after.add_cmd(Command("commit"))
+        else:
+            after.add_cmd(Command("abort"))  # просто exit оставит висеть configure session
+        if do_finalize:
+            after.add_cmd(Command("write memory"))
+
+        return before, after
+
     def match(self) -> list[str]:
         return ["Arista"]
 

annet/vendors/library/aruba.py

@@ -1,3 +1,4 @@
+from annet.annlib.command import Command, CommandList
 from annet.annlib.netdev.views.hardware import HardwareView
 from annet.annlib.tabparser import ArubaFormatter
 from annet.vendors.base import AbstractVendor
@@ -8,6 +9,18 @@ from annet.vendors.registry import registry
 class ArubaVendor(AbstractVendor):
     NAME = "aruba"
 
+    def apply(self, hw: HardwareView, do_commit: bool, do_finalize: bool, path: str) -> tuple[CommandList, CommandList]:
+        before, after = CommandList(), CommandList()
+
+        before.add_cmd(Command("conf t"))
+        after.add_cmd(Command("end"))
+        if do_commit:
+            after.add_cmd(Command("commit apply"))
+        if do_finalize:
+            after.add_cmd(Command("write memory"))
+
+        return before, after
+
     def match(self) -> list[str]:
         return ["Aruba"]
 

annet/vendors/library/b4com.py

@@ -1,3 +1,4 @@
+from annet.annlib.command import Command, CommandList
 from annet.annlib.netdev.views.hardware import HardwareView
 from annet.annlib.tabparser import B4comFormatter
 from annet.vendors.base import AbstractVendor
@@ -8,6 +9,24 @@ from annet.vendors.registry import registry
 class B4ComVendor(AbstractVendor):
     NAME = "b4com"
 
+    def apply(self, hw: HardwareView, do_commit: bool, do_finalize: bool, path: str) -> tuple[CommandList, CommandList]:
+        before, after = CommandList(), CommandList()
+
+        if hw.B4com.CS2148P:
+            before.add_cmd(Command("conf t"))
+            after.add_cmd(Command("end"))
+            if do_finalize:
+                after.add_cmd(Command("write", timeout=40))
+        else:
+            before.add_cmd(Command("conf t"))
+            if do_commit:
+                after.add_cmd(Command("commit"))
+                after.add_cmd(Command("end"))
+            if do_finalize:
+                after.add_cmd(Command("write", timeout=40))
+
+        return before, after
+
     def match(self) -> list[str]:
         return ["B4com"]
 

annet/vendors/library/cisco.py

@@ -1,3 +1,4 @@
+from annet.annlib.command import Command, CommandList
 from annet.annlib.netdev.views.hardware import HardwareView
 from annet.annlib.tabparser import CiscoFormatter
 from annet.vendors.base import AbstractVendor
@@ -8,6 +9,16 @@ from annet.vendors.registry import registry
 class CiscoVendor(AbstractVendor):
     NAME = "cisco"
 
+    def apply(self, hw: HardwareView, do_commit: bool, do_finalize: bool, path: str) -> tuple[CommandList, CommandList]:
+        before, after = CommandList(), CommandList()
+
+        before.add_cmd(Command("conf t"))
+        after.add_cmd(Command("exit"))
+        if do_finalize:
+            after.add_cmd(Command("copy running-config startup-config", timeout=40))
+
+        return before, after
+
     def match(self) -> list[str]:
         return ["Cisco"]
 

annet/vendors/library/h3c.py

@@ -1,3 +1,4 @@
+from annet.annlib.command import Command, CommandList
 from annet.annlib.netdev.views.hardware import HardwareView
 from annet.annlib.tabparser import HuaweiFormatter
 from annet.vendors.base import AbstractVendor
@@ -8,6 +9,15 @@ from annet.vendors.registry import registry
 class H3CVendor(AbstractVendor):
     NAME = "h3c"
 
+    def apply(self, hw: HardwareView, do_commit: bool, do_finalize: bool, path: str) -> tuple[CommandList, CommandList]:
+        before, after = CommandList(), CommandList()
+
+        before.add_cmd(Command("system-view"))
+        if do_finalize:
+            after.add_cmd(Command("save force", timeout=90))
+
+        return before, after
+
     def match(self) -> list[str]:
         return ["H3C"]
 

annet/vendors/library/huawei.py

@@ -1,3 +1,4 @@
+from annet.annlib.command import Command, CommandList
 from annet.annlib.netdev.views.hardware import HardwareView
 from annet.annlib.tabparser import HuaweiFormatter
 from annet.vendors.base import AbstractVendor
@@ -8,6 +9,18 @@ from annet.vendors.registry import registry
 class HuaweiVendor(AbstractVendor):
     NAME = "huawei"
 
+    def apply(self, hw: HardwareView, do_commit: bool, do_finalize: bool, path: str) -> tuple[CommandList, CommandList]:
+        before, after = CommandList(), CommandList()
+
+        before.add_cmd(Command("system-view"))
+        if do_commit and (hw.Huawei.CE or hw.Huawei.NE):
+            after.add_cmd(Command("commit"))
+        after.add_cmd(Command("q"))
+        if do_finalize:
+            after.add_cmd(Command("save", timeout=20))
+
+        return before, after
+
     def match(self) -> list[str]:
         return ["Huawei"]
 

annet/vendors/library/iosxr.py

@@ -1,3 +1,4 @@
+from annet.annlib.command import Command, CommandList
 from annet.annlib.netdev.views.hardware import HardwareView
 from annet.annlib.tabparser import AsrFormatter
 from annet.vendors.base import AbstractVendor
@@ -8,6 +9,16 @@ from annet.vendors.registry import registry
 class IosXrVendor(AbstractVendor):
     NAME = "iosxr"
 
+    def apply(self, hw: HardwareView, do_commit: bool, do_finalize: bool, path: str) -> tuple[CommandList, CommandList]:
+        before, after = CommandList(), CommandList()
+
+        before.add_cmd(Command("configure exclusive"))
+        if do_commit:
+            after.add_cmd(Command("commit"))
+        after.add_cmd(Command("exit"))
+
+        return before, after
+
     def match(self) -> list[str]:
         return ["Cisco.ASR", "Cisco.XR", "Cisco.XRV"]
 

annet/vendors/library/juniper.py

@@ -1,3 +1,4 @@
+from annet.annlib.command import Command, CommandList
 from annet.annlib.netdev.views.hardware import HardwareView
 from annet.annlib.tabparser import JuniperFormatter
 from annet.vendors.base import AbstractVendor
@@ -8,6 +9,16 @@ from annet.vendors.registry import registry
 class JuniperVendor(AbstractVendor):
     NAME = "juniper"
 
+    def apply(self, hw: HardwareView, do_commit: bool, do_finalize: bool, path: str) -> tuple[CommandList, CommandList]:
+        before, after = CommandList(), CommandList()
+
+        before.add_cmd(Command("configure exclusive"))
+        if do_commit:
+            after.add_cmd(Command("commit", timeout=30))
+        after.add_cmd(Command("exit"))
+
+        return before, after
+
     def match(self) -> list[str]:
         return ["Juniper"]
 

annet/vendors/library/nexus.py

@@ -1,3 +1,4 @@
+from annet.annlib.command import Command, CommandList
 from annet.annlib.netdev.views.hardware import HardwareView
 from annet.annlib.tabparser import NexusFormatter
 from annet.vendors.base import AbstractVendor
@@ -8,6 +9,16 @@ from annet.vendors.registry import registry
 class NexusVendor(AbstractVendor):
     NAME = "nexus"
 
+    def apply(self, hw: HardwareView, do_commit: bool, do_finalize: bool, path: str) -> tuple[CommandList, CommandList]:
+        before, after = CommandList(), CommandList()
+
+        before.add_cmd(Command("conf t"))
+        after.add_cmd(Command("exit"))
+        if do_finalize:
+            after.add_cmd(Command("copy running-config startup-config", timeout=40))
+
+        return before, after
+
     def match(self) -> list[str]:
         return ["Cisco.Nexus"]
 

annet/vendors/library/nokia.py

@@ -1,3 +1,4 @@
+from annet.annlib.command import Command, CommandList
 from annet.annlib.netdev.views.hardware import HardwareView
 from annet.annlib.tabparser import NokiaFormatter
 from annet.vendors.base import AbstractVendor
@@ -8,6 +9,15 @@ from annet.vendors.registry import registry
 class NokiaVendor(AbstractVendor):
     NAME = "nokia"
 
+    def apply(self, hw: HardwareView, do_commit: bool, do_finalize: bool, path: str) -> tuple[CommandList, CommandList]:
+        before, after = CommandList(), CommandList()
+
+        before.add_cmd(Command("configure private"))
+        if do_commit:
+            after.add_cmd(Command("commit"))
+
+        return before, after
+
     def match(self) -> list[str]:
         return ["Nokia"]
 

annet/vendors/library/optixtrans.py

@@ -1,19 +1,16 @@
 from annet.annlib.netdev.views.hardware import HardwareView
 from annet.annlib.tabparser import OptixtransFormatter
-from annet.vendors.base import AbstractVendor
 from annet.vendors.registry import registry
 
+from .huawei import HuaweiVendor
+
 
 @registry.register
-class OptixTransVendor(AbstractVendor):
+class OptixTransVendor(HuaweiVendor):
     NAME = "optixtrans"
 
     def match(self) -> list[str]:
-        return ["OptiXtrans"]
-
-    @property
-    def reverse(self) -> str:
-        return "undo"
+        return ["Huawei.OptiXtrans"]
 
     @property
     def hardware(self) -> HardwareView:
@@ -22,6 +19,5 @@ class OptixTransVendor(AbstractVendor):
     def make_formatter(self, **kwargs) -> OptixtransFormatter:
         return OptixtransFormatter(**kwargs)
 
-    @property
-    def exit(self) -> str:
-        return "quit"
+    def svi_name(self, num: int) -> str:
+        return f"vlan{num}"

annet/vendors/library/pc.py

@@ -1,3 +1,6 @@
+import os
+
+from annet.annlib.command import Command, CommandList
 from annet.annlib.netdev.views.hardware import HardwareView
 from annet.annlib.tabparser import CommonFormatter
 from annet.vendors.base import AbstractVendor
@@ -15,6 +18,15 @@ class PCVendor(AbstractVendor):
     def reverse(self) -> str:
         return "-"
 
+    def apply(self, hw: HardwareView, do_commit: bool, do_finalize: bool, path: str) -> tuple[CommandList, CommandList]:
+        before, after = CommandList(), CommandList()
+
+        if hw.soft.startswith(("Cumulus", "SwitchDev")):
+            if os.environ.get("ETCKEEPER_CHECK", False):
+                before.add_cmd(Command("etckeeper check"))
+
+        return before, after
+
     @property
     def hardware(self) -> HardwareView:
         return HardwareView("PC")

annet/vendors/library/ribbon.py

@@ -1,3 +1,4 @@
+from annet.annlib.command import Command, CommandList
 from annet.annlib.netdev.views.hardware import HardwareView
 from annet.annlib.tabparser import RibbonFormatter
 from annet.vendors.base import AbstractVendor
@@ -8,6 +9,16 @@ from annet.vendors.registry import registry
 class RibbonVendor(AbstractVendor):
     NAME = "ribbon"
 
+    def apply(self, hw: HardwareView, do_commit: bool, do_finalize: bool, path: str) -> tuple[CommandList, CommandList]:
+        before, after = CommandList(), CommandList()
+
+        before.add_cmd(Command("configure exclusive"))
+        if do_commit:
+            after.add_cmd(Command("commit", timeout=30))
+        after.add_cmd(Command("exit"))
+
+        return before, after
+
     def match(self) -> list[str]:
         return ["Ribbon"]
 

annet/vendors/library/routeros.py

@@ -1,3 +1,4 @@
+from annet.annlib.command import Command, CommandList
 from annet.annlib.netdev.views.hardware import HardwareView
 from annet.annlib.tabparser import RosFormatter
 from annet.vendors.base import AbstractVendor
@@ -8,6 +9,18 @@ from annet.vendors.registry import registry
 class RouterOSVendor(AbstractVendor):
     NAME = "routeros"
 
+    def apply(self, hw: HardwareView, do_commit: bool, do_finalize: bool, path: str) -> tuple[CommandList, CommandList]:
+        before, after = CommandList(), CommandList()
+
+        # FIXME: пока не удалось победить \x1b[c после включения safe mode
+        # if len(cmds) > 99:
+        #     raise Exception("RouterOS does not support more 100 actions in safe mode")
+        # before.add_cmd(RosDevice.SAFE_MODE)
+        pass
+        # after.add_cmd(RosDevice.SAFE_MODE)
+
+        return before, after
+
     def match(self) -> list[str]:
         return ["RouterOS"]
 

{annet-3.1.4.dist-info → annet-3.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: annet
-Version: 3.1.4
+Version: 3.3.0
 Summary: annet
 Home-page: https://github.com/annetutil/annet
 License: MIT
@@ -21,7 +21,7 @@ Requires-Dist: yarl>=1.8.2
 Requires-Dist: adaptix==3.0.0b7
 Requires-Dist: dataclass-rest==0.4
 Provides-Extra: netbox
-Requires-Dist: annetbox[sync]>=0.3.0; extra == "netbox"
+Requires-Dist: annetbox[sync]>=0.4.0; extra == "netbox"
 Dynamic: home-page
 Dynamic: license
 Dynamic: license-file

{annet-3.1.4.dist-info → annet-3.3.0.dist-info}/RECORD

@@ -49,8 +49,8 @@ annet/adapters/netbox/v41/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 annet/adapters/netbox/v41/models.py,sha256=nunqgxffobE2C3_g1bkHXWffStBdAJxHfx0eAbvnWAU,2068
 annet/adapters/netbox/v41/storage.py,sha256=h6e7V4Od5o7P3Ssr_vUQ1IBjsxdnug4YGvG1BokwKuk,3795
 annet/adapters/netbox/v42/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-annet/adapters/netbox/v42/models.py,sha256=xo5TB6VE0KNvQwUB_zasl2bDbNuKCCjcsHB3P-JVLgk,1784
-annet/adapters/netbox/v42/storage.py,sha256=atiThgkDGVdJStvFCDn4JtPZfEAhUNbw5IH_W-ijFDg,3795
+annet/adapters/netbox/v42/models.py,sha256=UWrnRh1k3a048XsVmqh50dF91dunrolIJzmTLT0QExw,1920
+annet/adapters/netbox/v42/storage.py,sha256=iMhdrdAAYZXS_VHoqi_5azFB3BVmp4daK4TVCI9vhxM,4666
 annet/annlib/__init__.py,sha256=fT1l4xV5fqqg8HPw9HqmZVN2qwS8i6X1aIm2zGDjxKY,252
 annet/annlib/command.py,sha256=uuBddMQphtn8P5MO5kzIa8_QrtMns-k05VeKv1bcAuA,1043
 annet/annlib/diff.py,sha256=MZ6eQAU3cadQp8KaSE6uAYFtcfMDCIe_eNuVROnYkCk,4496
@@ -76,17 +76,17 @@ annet/annlib/rbparser/ordering.py,sha256=XknggB92N4IbhHBZZHL0StwCCITt8mkho4D4Nu5
 annet/annlib/rbparser/platform.py,sha256=d1jFH8hGMOf_qiveKE4H-c0OfNnzwmt2VYpSiSOn9qc,42
 annet/annlib/rbparser/syntax.py,sha256=iZ7Y-4QQBw4L3UtjEh54qisiRDhobl7HZxFNdP8mi54,3577
 annet/annlib/rulebook/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-annet/annlib/rulebook/common.py,sha256=h2x-NWnOboED4jUHdywbB0lcl1wbI1hmlVF4n2l87OU,17016
+annet/annlib/rulebook/common.py,sha256=2ZMu7Z47Ip7d8yMjihDNKKhwl9aV5U4IeCcyXM4CjeA,13527
 annet/api/__init__.py,sha256=EMGUgt3pTi2MDF8DfXlS8PGX2zT0iGLY01R6QlQHvGg,33305
 annet/configs/context.yml,sha256=RVLrKLIHpCty7AGwOnmqf7Uu0iZQCn-AjYhophDJer8,259
 annet/configs/logging.yaml,sha256=EUagfir99QqA73Scc3k7sfQccbU3E1SvEQdyhLFtCl4,997
-annet/generators/__init__.py,sha256=wJbNLI1bMppE71tnjJrqNGtbtLZ_gdRd-HytdZPzBiY,16451
+annet/generators/__init__.py,sha256=p71npgfOVie1jtbkGd_Uw_sD1STl6-HdddcfcPOBMVg,16632
 annet/generators/base.py,sha256=rgQLcQBPZm4ecbKmRhVOpBR-GFJAiVfdb_y5f2-LUR8,3670
-annet/generators/entire.py,sha256=mfnKtn5GcHiqjGjuhETkOVm-0ppMELAPsmg5jQrYTZc,2922
+annet/generators/entire.py,sha256=jFTbx9BKoiXMstq_QeesRH2yi01LHyYHxgCtvvwBmeM,2872
 annet/generators/exceptions.py,sha256=GPXTBgn2xZ3Ev_bdOPlfCLGRC1kQHe_dEq88S-uyi5s,151
 annet/generators/jsonfragment.py,sha256=Cl43t9_OtNWRxesk3B69h60KvD37tiK9W7sLLmppAT4,4379
 annet/generators/partial.py,sha256=XI01KDA--XwjSEU33SOQCCJZRXFq5boRz1uJA8lVA1g,3502
-annet/generators/perf.py,sha256=K72ivUuXbNXrsHrLeKWhGmczGYWsB7kUDdDzqOX6j3c,2370
+annet/generators/perf.py,sha256=IaAcfEVtX7UNO11VOCXzp-FPj_tOx_CDQ34HGThCn4c,2225
 annet/generators/ref.py,sha256=QVdeL8po1D0kBsVLOpCjFR81D8yNTk-kaQj5WUM4hng,438
 annet/generators/result.py,sha256=zMAvGOYQU803bGy6datZduHLgrEqK2Zba_Jcf9Qn9p0,4976
 annet/generators/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -180,25 +180,25 @@ annet/rulebook/texts/ribbon.rul,sha256=609LyLTDCtXPVQNAzqS-VEyCpW3byHP8TCMJLFMn5
 annet/rulebook/texts/routeros.order,sha256=M71uy_hf0KAjLNS3zZY3uih4m2xLUcu26FEoVVjC6k0,905
 annet/rulebook/texts/routeros.rul,sha256=ipfxjj0mjFef6IsUlupqx4BY_Je_OUb8u_U1019O1DE,1203
 annet/vendors/__init__.py,sha256=gQcDFlKeWDZB6vxJ_MdPWEoE-C5dg-YgXvgGkuV9YLw,569
-annet/vendors/base.py,sha256=KLf_lv-Vm3uPH_Kuovdr1V257EALmAS_gYhKEL196FA,919
+annet/vendors/base.py,sha256=J9Bz7pDdBE7nCceHm2t9EzZ0MJChEugKLWaoOKZlP9Q,1144
 annet/vendors/registry.py,sha256=zJUj5vJ105kD7xvxt84MYNTNYZI08AKCNTCmE0YtMzo,2539
 annet/vendors/library/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-annet/vendors/library/arista.py,sha256=SF6RxoOsQwp7MUmwnAS0V8HrmHkGKDdL5jj3pPOmVLo,732
-annet/vendors/library/aruba.py,sha256=coAEyEsHkILvcIx6IZeBywJWXLgcRPffu56AYQrTedY,768
-annet/vendors/library/b4com.py,sha256=jRemG0mur0cItmGLmjrqDKt5GDsJ56nIyB6JCsenWhA,655
-annet/vendors/library/cisco.py,sha256=ZwBKBWVJS1DmUl1DnklyxQr85kEg7wef75emOFahIZc,725
-annet/vendors/library/h3c.py,sha256=x1QwQqcjD4AUTtJI5CJ9g90N14piNFUuaJ9uAorST3k,652
-annet/vendors/library/huawei.py,sha256=cw69jgFfce4l2Lv_fs2x54b30LBdkjhsuwg8RglFCJ8,736
-annet/vendors/library/iosxr.py,sha256=prC4WZct6Z-msFd14lTcr6zkL0HMlz7apG3Kry8VpNU,752
-annet/vendors/library/juniper.py,sha256=1rVdNtOmAd96F_jMSfxKJqmcVhvXcjGxPg6zJ72qK-Q,855
-annet/vendors/library/nexus.py,sha256=OAAfJB4OON34iEJb1jIl_v7aIwcK10yxcPOV_l1E6AI,667
-annet/vendors/library/nokia.py,sha256=GWZ6f2L_mSy9RMsRR-8FJKXFA7x9pj_tODy2CD7fxXw,771
-annet/vendors/library/optixtrans.py,sha256=425akCUYKdeM7qV1ZUvfRjho_bWoWhHotoo0j5plGmY,699
-annet/vendors/library/pc.py,sha256=UOfaBtlFUGiAF9qsjlhgWNDJz31SqlMdtyYqoYtn6GM,641
-annet/vendors/library/ribbon.py,sha256=3DJBZ5wOoi0mkfqLpNeRJIkThVuRqSiJ_67G7eoKZzs,782
-annet/vendors/library/routeros.py,sha256=pvMG3MJpc4-pUBYnEzOXWfK0vqM7l0SlxwlGywaPovk,661
-annet-3.1.4.dist-info/licenses/AUTHORS,sha256=rh3w5P6gEgqmuC-bw-HB68vBCr-yIBFhVL0PG4hguLs,878
-annet-3.1.4.dist-info/licenses/LICENSE,sha256=yPxl7dno02Pw7gAcFPIFONzx_gapwDoPXsIsh6Y7lC0,1079
+annet/vendors/library/arista.py,sha256=tp01vFmPDa2H3-3MSHzJ0iWVwe3kI_IKNViHa0VWB8M,1301
+annet/vendors/library/aruba.py,sha256=vGRUlbe48LRZs67o19wLOLnbDW-GDbAASWwsPuAzUro,1256
+annet/vendors/library/b4com.py,sha256=GMgOLd_JT-96gWb4KKGlZcaJ5Cfbp_cwXtYbwrPimNM,1389
+annet/vendors/library/cisco.py,sha256=ccKLOqxPQPFKUxb-SMdYR6-CCcILRvUB8shXZa7y_VU,1175
+annet/vendors/library/h3c.py,sha256=-gY4DKAMqMcoMcVCWORxM945SfI9r8zekPXFqqKi52o,1044
+annet/vendors/library/huawei.py,sha256=rDI3ScgJ7Cq8SlnGTZsZT8JcM8PxjyeGsQUMA5Xo4i8,1260
+annet/vendors/library/iosxr.py,sha256=5QUN0ERGhBvXRA6G2FC7fHeE-GNuk58Wog28sqzFU2k,1173
+annet/vendors/library/juniper.py,sha256=1Zih2jPdq34ZUZpjUX5nZKxOq4eiqgypaVqkmTmvTbc,1288
+annet/vendors/library/nexus.py,sha256=GW3fWz0W10V9AuW17FFDM9vVUoCQgUVIGDufJaU6B2w,1117
+annet/vendors/library/nokia.py,sha256=IPAwBux11cM2YdEK_b9dal7NAdmcMAo9HCLbWxjdZzo,1151
+annet/vendors/library/optixtrans.py,sha256=xNurVzIrr7cefVUCzhxI-7xnCnayk1kIyRfU8Ug5t8M,631
+annet/vendors/library/pc.py,sha256=4lPTvtvjGDCUuZ6yDyi-U4X9HPZRtj_CoYZM38jxwzo,1085
+annet/vendors/library/ribbon.py,sha256=55tlhY8weGTE9x-CDpFUMvEX6pg0KnrXnd-P8bxceGk,1215
+annet/vendors/library/routeros.py,sha256=0Hi-tDBjgBwfewiZi0EAnXkgGGHlOc_uwpY8xd9p3TM,1253
+annet-3.3.0.dist-info/licenses/AUTHORS,sha256=rh3w5P6gEgqmuC-bw-HB68vBCr-yIBFhVL0PG4hguLs,878
+annet-3.3.0.dist-info/licenses/LICENSE,sha256=yPxl7dno02Pw7gAcFPIFONzx_gapwDoPXsIsh6Y7lC0,1079
 annet_generators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 annet_generators/example/__init__.py,sha256=OJ77uj8axc-FIyIu_Xdcnzmde3oQW5mk5qbODkhuVc8,355
 annet_generators/example/hostname.py,sha256=RloLzNVetEoWPLITzfJ13Nk3CC0yi-cZB1RTd6dnuhI,2541
@@ -211,8 +211,8 @@ annet_generators/rpl_example/generator.py,sha256=EWah19gOH8G-QyNyWqxCqdRi0BK7GbM
 annet_generators/rpl_example/items.py,sha256=d99HSXDHFjZq511EvGhIqRTWK3F4ZsCWfdUqFYQcyhE,772
 annet_generators/rpl_example/mesh.py,sha256=z_WgfDZZ4xnyh3cSf75igyH09hGvtexEVwy1gCD_DzA,288
 annet_generators/rpl_example/route_policy.py,sha256=z6nPb0VDeQtKD1NIg9sFvmUxBD5tVs2frfNIuKdM-5c,2318
-annet-3.1.4.dist-info/METADATA,sha256=Q7ZpCjkgFIsvMzZDWHS4rr9JIsPjumv3CQDbHr8iFdc,815
-annet-3.1.4.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-annet-3.1.4.dist-info/entry_points.txt,sha256=5lIaDGlGi3l6QQ2ry2jZaqViP5Lvt8AmsegdD0Uznck,192
-annet-3.1.4.dist-info/top_level.txt,sha256=QsoTZBsUtwp_FEcmRwuN8QITBmLOZFqjssRfKilGbP8,23
-annet-3.1.4.dist-info/RECORD,,
+annet-3.3.0.dist-info/METADATA,sha256=5C_NTl6Czb2mm3dWoXxyGwGlcuGcdN3kAVYBBWD6Ie0,815
+annet-3.3.0.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
+annet-3.3.0.dist-info/entry_points.txt,sha256=5lIaDGlGi3l6QQ2ry2jZaqViP5Lvt8AmsegdD0Uznck,192
+annet-3.3.0.dist-info/top_level.txt,sha256=QsoTZBsUtwp_FEcmRwuN8QITBmLOZFqjssRfKilGbP8,23
+annet-3.3.0.dist-info/RECORD,,

{annet-3.1.4.dist-info → annet-3.3.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (79.0.1)
+Generator: setuptools (80.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any