annet 0.16.29__tar.gz → 0.16.31__tar.gz

{annet-0.16.29/annet.egg-info → annet-0.16.31}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: annet
-Version: 0.16.29
+Version: 0.16.31
 Summary: annet
 Home-page: https://github.com/annetutil/annet
 License: MIT
@@ -24,3 +24,9 @@ Requires-Dist: adaptix==3.0.0b7
 Requires-Dist: dataclass-rest==0.4
 Provides-Extra: netbox
 Requires-Dist: annetbox[sync]>=0.1.10; extra == "netbox"
+Dynamic: home-page
+Dynamic: license
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

{annet-0.16.29 → annet-0.16.31}/annet/adapters/file/provider.py

@@ -102,6 +102,9 @@ class Device(DeviceCls, DumpableView):
     def add_subif(self, interface: str, subif: int) -> Interface:
         raise NotImplementedError
 
+    def find_interface(self, name: str) -> Optional[Interface]:
+        raise NotImplementedError
+
     def neighbours_fqdns(self) -> list[str]:
         return []
 

{annet-0.16.29 → annet-0.16.31}/annet/adapters/netbox/common/manufacturer.py

@@ -54,6 +54,8 @@ def get_breed(manufacturer: str, model: str):
         return "bcom-os"
     elif manufacturer == "MikroTik":
         return "routeros"
+    elif manufacturer == "Moxa":
+        return "moxa"
     elif manufacturer == "PC":
         return "pc"
     return ""

{annet-0.16.29 → annet-0.16.31}/annet/adapters/netbox/common/models.py

@@ -214,7 +214,7 @@ class NetboxDevice(Entity):
         return type(self) is type(other) and self.url == other.url
 
     def is_pc(self) -> bool:
-        return self.device_type.manufacturer.name == "Mellanox" or self.breed == "pc"
+        return self.device_type.manufacturer.name in ("Mellanox", "Moxa") or self.breed == "pc"
 
     def _make_interface(self, name: str, type: InterfaceType) -> Interface:
         return Interface(
@@ -281,3 +281,9 @@ class NetboxDevice(Entity):
         )
         self.interfaces.append(target_port)
         return target_port
+
+    def find_interface(self, name: str) -> Optional[Interface]:
+        for iface in self.interfaces:
+            if iface.name == name:
+                return iface
+        return None

{annet-0.16.29 → annet-0.16.31}/annet/annlib/netdev/devdb/data/devdb.json

@@ -116,6 +116,7 @@
     "PC.Whitebox.Edgecore": "[Ee]dge-?[Cc]ore",
     "PC.Whitebox.Edgecore.AS": "AS",
     "PC.Whitebox.Edgecore.AS9736": "AS9736",
+    "PC.Whitebox.Edgecore.AS9817": "AS9817",
     "PC.Whitebox.NVIDIA": "NVIDIA",
     "PC.Whitebox.NVIDIA.SN": " SN",
     "PC.Whitebox.NVIDIA.SN.SN5400": " SN5400",

{annet-0.16.29 → annet-0.16.31}/annet/api/__init__.py

@@ -274,12 +274,14 @@ def patch(args: cli_args.ShowPatchOptions, loader: ann_gen.Loader):
 
 def _patch_worker(device_id, args: cli_args.ShowPatchOptions, stdin, loader: ann_gen.Loader, filterer: filtering.Filterer):
     for res, _, patch_tree in res_diff_patch(device_id, args, stdin, loader, filterer):
+        old_files = res.old_files
         new_files = res.get_new_files(args.acl_safe)
         new_json_fragment_files = res.get_new_file_fragments(args.acl_safe)
         if new_files:
             for path, (cfg_text, _cmds) in new_files.items():
                 label = res.device.hostname + os.sep + path
-                yield label, cfg_text, False
+                if old_files.get(path) != cfg_text:
+                    yield label, cfg_text, False
         elif res.old_json_fragment_files or new_json_fragment_files:
             for path, (new_json_cfg, _cmds) in new_json_fragment_files.items():
                 label = res.device.hostname + os.sep + path
@@ -468,6 +470,9 @@ class PCDeployerJob(DeployerJob):
         elif not new_files and not new_json_fragment_files:
             return
 
+        enable_reload = self.args.entire_reload is not cli_args.EntireReloadFlag.no
+        force_reload = self.args.entire_reload is cli_args.EntireReloadFlag.force
+
         upload_files: Dict[str, bytes] = {}
         reload_cmds: Dict[str, bytes] = {}
         generator_types: Dict[str, GeneratorType] = {}
@@ -483,18 +488,23 @@ class PCDeployerJob(DeployerJob):
                     old_text = jsontools.format_json(old_json_cfg)
                     new_text = jsontools.format_json(file_content_or_json_cfg)
                     diff_content = "\n".join(_diff_file(old_text, new_text))
-                if diff_content:
-                    self._has_diff = True
-                upload_files[file], reload_cmds[file] = file_content.encode(), cmds.encode()
-                generator_types[file] = generator_type
-                self.cmd_lines.append("= Deploy cmds %s/%s " % (device.hostname, file))
-                self.cmd_lines.extend([cmds, ""])
-                self.cmd_lines.append("= %s/%s " % (device.hostname, file))
-                self.cmd_lines.extend([file_content, ""])
-                self.diff_lines.append("= %s/%s " % (device.hostname, file))
-                self.diff_lines.extend([diff_content, ""])
-
-        if upload_files:
+
+                if diff_content or force_reload:
+                    self._has_diff |= True
+
+                    upload_files[file] = file_content.encode()
+                    generator_types[file] = generator_type
+                    self.cmd_lines.append("= %s/%s " % (device.hostname, file))
+                    self.cmd_lines.extend([file_content, ""])
+                    self.diff_lines.append("= %s/%s " % (device.hostname, file))
+                    self.diff_lines.extend([diff_content, ""])
+
+                    if enable_reload:
+                        reload_cmds[file] = cmds.encode()
+                        self.cmd_lines.append("= Deploy cmds %s/%s " % (device.hostname, file))
+                        self.cmd_lines.extend([cmds, ""])
+
+        if self._has_diff:
             self.deploy_cmds[device] = {
                 "files": upload_files,
                 "cmds": reload_cmds,
@@ -534,13 +544,12 @@ class Deployer:
         self._filterer = filtering.filterer_connector.get()
 
     def parse_result(self, job: DeployerJob, result: ann_gen.OldNewResult):
-        entire_reload = self.args.entire_reload
         logger = get_logger(job.device.hostname)
 
         job.parse_result(result)
         self.failed_configs.update(job.failed_configs)
 
-        if job.has_diff() or entire_reload is entire_reload.force:
+        if job.has_diff():
             self.cmd_lines.extend(job.cmd_lines)
             self.deploy_cmds.update(job.deploy_cmds)
             self.diffs.update(job.diffs)

{annet-0.16.29 → annet-0.16.31}/annet/bgp_models.py

@@ -286,13 +286,30 @@ def _used_policies(peer: Union[Peer, PeerGroup]) -> Iterable[str]:
         yield peer.export_policy
 
 
+def _used_redistribute_policies(opts: Union[GlobalOptions, VrfOptions]) -> Iterable[str]:
+    for red in opts.ipv4_unicast.redistributes:
+        if red.policy:
+            yield red.policy
+    for red in opts.ipv6_unicast.redistributes:
+        if red.policy:
+            yield red.policy
+    for red in opts.ipv4_labeled_unicast.redistributes:
+        if red.policy:
+            yield red.policy
+    for red in opts.ipv6_labeled_unicast.redistributes:
+        if red.policy:
+            yield red.policy
+
+
 def extract_policies(config: BgpConfig) -> Sequence[str]:
     result: list[str] = []
     for vrf in config.global_options.vrf.values():
         for group in vrf.groups:
             result.extend(_used_policies(group))
+        result.extend(_used_redistribute_policies(vrf))
     for group in config.global_options.groups:
         result.extend(_used_policies(group))
     for peer in config.peers:
         result.extend(_used_policies(peer))
+    result.extend(_used_redistribute_policies(config.global_options))
     return result

{annet-0.16.29 → annet-0.16.31}/annet/mesh/basemodel.py

@@ -133,6 +133,9 @@ class BaseMeshModel:
             raise AttributeError(f"{self.__class__.__name__} has no field {key}")
         super().__setattr__(key, value)
 
+    def is_empty(self):
+        return not self.__dict__
+
 
 ModelT = TypeVar("ModelT", bound=BaseMeshModel)
 

{annet-0.16.29 → annet-0.16.31}/annet/mesh/executor.py

@@ -57,6 +57,11 @@ class MeshExecutor:
             rule_global_opts = MeshGlobalOptions(rule.match, device)
             logger.debug("Running device handler: %s", handler_name)
             rule.handler(rule_global_opts)
+
+            if rule_global_opts.is_empty():
+                # nothing was set
+                continue
+
             try:
                 global_opts = merge(global_opts, rule_global_opts)
             except MergeForbiddenError as e:
@@ -79,7 +84,7 @@ class MeshExecutor:
             rule: MatchedDirectPair,
             ports: list[tuple[str, str]],
             all_connected_ports: list[tuple[str, str]],
-    ) -> Pair:
+    ) -> Optional[Pair]:
         session = MeshSession()
         handler_name = self._handler_name(rule.handler)
         logger.debug("Running direct handler: %s", handler_name)
@@ -102,6 +107,10 @@ class MeshExecutor:
         else:
             rule.handler(peer_neighbor, peer_device, session)
 
+        if peer_neighbor.is_empty() and peer_device.is_empty() and session.is_empty():
+            # nothing was set
+            return None
+
         try:
             neighbor_dto = merge(DirectPeerDTO(), peer_neighbor, session)
         except MergeForbiddenError as e:
@@ -124,18 +133,29 @@ class MeshExecutor:
         # we merge them according to remote fqdn
         neighbor_peers: dict[PeerKey, Pair] = {}
         # TODO batch resolve
-        for rule in self._registry.lookup_direct(device.fqdn, device.neighbours_fqdns):
+        rules = self._registry.lookup_direct(device.fqdn, device.neighbours_fqdns)
+        fqdns = {
+            rule.name_right if rule.direct_order else rule.name_left
+            for rule in rules
+        }
+        neigbors = {
+            d.fqdn: d for d in self._storage.make_devices(list(fqdns))
+        }
+        for rule in rules:
             handler_name = self._handler_name(rule.handler)
             if rule.direct_order:
-                neighbor_device = self._storage.make_devices([rule.name_right])[0]
+                neighbor_device = neigbors[rule.name_right]
             else:
-                neighbor_device = self._storage.make_devices([rule.name_left])[0]
+                neighbor_device = neigbors[rule.name_left]
             all_connected_ports = [
                 (p1.name, p2.name)
                 for p1, p2 in self._storage.search_connections(device, neighbor_device)
             ]
             for ports in rule.port_processor(all_connected_ports):
                 pair = self._execute_direct_pair(device, neighbor_device, rule, ports, all_connected_ports)
+                if pair is None:
+                    # nothing was set
+                    continue
                 addr = getattr(pair.connected, "addr", None)
                 if addr is None:
                     raise ValueError(f"Handler `{handler_name}` returned no peer addr")
@@ -171,6 +191,9 @@ class MeshExecutor:
                 peer_virtual = VirtualPeer(num=order_number)
 
                 rule.handler(peer_device, peer_virtual, session)
+                if peer_virtual.is_empty() and peer_device.is_empty() and session.is_empty():
+                    # nothing was set
+                    continue
 
                 try:
                     virtual_dto = merge(VirtualPeerDTO(), peer_virtual, session)
@@ -200,21 +223,34 @@ class MeshExecutor:
         # we can have multiple rules for the same pair
         # we merge them according to remote fqdn
         connected_peers: dict[PeerKey, Pair] = {}
-        for rule in self._registry.lookup_indirect(device.fqdn, all_fqdns):
+
+        rules = self._registry.lookup_indirect(device.fqdn, all_fqdns)
+        fqdns = {
+            rule.name_right if rule.direct_order else rule.name_left
+            for rule in rules
+        }
+        connected_devices = {
+            d.fqdn: d for d in self._storage.make_devices(list(fqdns))
+        }
+        for rule in rules:
             session = MeshSession()
             handler_name = self._handler_name(rule.handler)
             logger.debug("Running indirect handler: %s", handler_name)
             if rule.direct_order:
-                connected_device = self._storage.make_devices([rule.name_right])[0]
+                connected_device = connected_devices[rule.name_right]
                 peer_device = IndirectPeer(rule.match_left, device)
                 peer_connected = IndirectPeer(rule.match_right, connected_device)
                 rule.handler(peer_device, peer_connected, session)
             else:
-                connected_device = self._storage.make_devices([rule.name_left])[0]
+                connected_device = connected_devices[rule.name_left]
                 peer_connected = IndirectPeer(rule.match_left, connected_device)
                 peer_device = IndirectPeer(rule.match_right, device)
                 rule.handler(peer_connected, peer_device, session)
 
+            if peer_connected.is_empty() and peer_device.is_empty() and session.is_empty():
+                # nothing was set
+                continue
+
             try:
                 connected_dto = merge(IndirectPeerDTO(), peer_connected, session)
             except MergeForbiddenError as e:
@@ -265,7 +301,7 @@ class MeshExecutor:
     def _to_bgp_global(self, global_options: GlobalOptionsDTO) -> GlobalOptions:
         return to_bgp_global_options(global_options)
 
-    def _apply_interface_changes(
+    def _apply_direct_interface_changes(
             self, device: Device, neighbor: Device, ports: list[str], changes: InterfaceChanges,
     ) -> str:
         # filter ports according to processed in pair
@@ -299,6 +335,24 @@ class MeshExecutor:
         target_interface.add_addr(changes.addr, changes.vrf)
         return target_interface.name
 
+    def _apply_indirect_interface_changes(
+            self, device: Device, neighbor: Device, ifname: Optional[str], changes: InterfaceChanges,
+    ) -> Optional[str]:
+        if changes.lag is not None:
+            raise ValueError("LAG creation unsupported for indirect peers")
+        elif changes.subif is not None:
+            target_interface = device.add_subif(ifname, changes.subif)
+        elif changes.svi is not None:
+            target_interface = device.add_svi(changes.svi)
+        elif not ifname:
+            return None
+        else:
+            target_interface = device.find_interface(ifname)
+            if not target_interface:
+                raise ValueError(f"Interface {ifname} not found for device {device.fqdn}")
+        target_interface.add_addr(changes.addr, changes.vrf)
+        return target_interface.name
+
     def _apply_virtual_interface_changes(self, device: Device, local: VirtualLocalDTO) -> str:
         return device.add_svi(local.svi).name  # we check if SVI configured in execute method
 
@@ -308,8 +362,9 @@ class MeshExecutor:
         global_options = self._to_bgp_global(self._execute_globals(device))
 
         peers = []
+        target_interface: Optional[str]
         for direct_pair in self._execute_direct(device):
-            target_interface = self._apply_interface_changes(
+            target_interface = self._apply_direct_interface_changes(
                 device,
                 direct_pair.device,
                 direct_pair.ports,
@@ -325,7 +380,13 @@ class MeshExecutor:
             peers.append(self._virtual_to_bgp_peer(virtual_pair, target_interface))
 
         for connected_pair in self._execute_indirect(device, all_fqdns):
-            peers.append(self._to_bgp_peer(connected_pair, None))
+            target_interface = self._apply_indirect_interface_changes(
+                device,
+                connected_pair.device,
+                getattr(connected_pair.local, "ifname", None),
+                to_interface_changes(connected_pair.local),
+            )
+            peers.append(self._to_bgp_peer(connected_pair, target_interface))
 
         return BgpConfig(
             global_options=global_options,

{annet-0.16.29 → annet-0.16.31}/annet/mesh/peer_models.py

@@ -95,6 +95,10 @@ class IndirectPeerDTO(MeshSession, _OptionsDTO):
     description: str
     update_source: str
 
+    ifname: Optional[str]
+    subif: int
+    svi: Optional[int]
+
 
 class VirtualLocalDTO(_OptionsDTO):
     asnum: int

{annet-0.16.29 → annet-0.16.31}/annet/mesh/registry.py

@@ -21,6 +21,9 @@ class DirectPeer(DirectPeerDTO):
         self.ports = ports
         self.all_connected_ports = all_connected_ports
 
+    def is_empty(self):
+        return self.__dict__.keys() == {"match", "device", "ports", "all_connected_ports"}
+
 
 class IndirectPeer(IndirectPeerDTO):
     match: MatchedArgs
@@ -31,6 +34,9 @@ class IndirectPeer(IndirectPeerDTO):
         self.match = match
         self.device = device
 
+    def is_empty(self):
+        return self.__dict__.keys() == {"match", "device"}
+
 
 class VirtualLocal(VirtualLocalDTO):
     match: MatchedArgs
@@ -41,10 +47,16 @@ class VirtualLocal(VirtualLocalDTO):
         self.match = match
         self.device = device
 
+    def is_empty(self):
+        return self.__dict__.keys() == {"match", "device"}
+
 
 class VirtualPeer(VirtualPeerDTO):
     num: int
 
+    def is_empty(self):
+        return self.__dict__.keys() == {"num"}
+
 
 class GlobalOptions(GlobalOptionsDTO):
     match: MatchedArgs
@@ -55,6 +67,9 @@ class GlobalOptions(GlobalOptionsDTO):
         self.match = match
         self.device = device
 
+    def is_empty(self):
+        return self.__dict__.keys() == {"match", "device"}
+
 
 GlobalHandler = Callable[[GlobalOptions], None]
 

{annet-0.16.29 → annet-0.16.31}/annet/rpl_generators/community.py

@@ -164,7 +164,7 @@ class CommunityListGenerator(PartialGenerator, ABC):
                 yield self._huawei_community_filter(10, community_list, " ".join(members))
             elif community_list.logic == CommunityLogic.OR:
                 for i, member in enumerate(members):
-                    member_id = (i + 1) * 10 + 5
+                    member_id = (i + 1) * 10
                     yield self._huawei_community_filter(member_id, community_list, member)
             else:
                 raise NotImplementedError(f"Community logic {community_list.logic} is not implemented for huawei")

{annet-0.16.29 → annet-0.16.31}/annet/rpl_generators/cumulus_frr.py

@@ -11,10 +11,10 @@ from annet.rpl.statement_builder import NextHopActionValue, AsPathActionValue, C
 from .aspath import get_used_as_path_filters
 from .community import get_used_united_community_lists
 from .entities import (
-    AsPathFilter, IpPrefixList, mangle_ranged_prefix_list_name, CommunityList, CommunityLogic, CommunityType,
-    mangle_united_community_list_name,
+    AsPathFilter, IpPrefixList, CommunityList, CommunityLogic, CommunityType,
+    mangle_united_community_list_name, PrefixListNameGenerator,
 )
-from .prefix_lists import get_used_prefix_lists
+from .prefix_lists import get_used_prefix_lists, new_prefix_list_name_generator
 
 FRR_RESULT_MAP = {
     ResultType.ALLOW: "permit",
@@ -51,6 +51,12 @@ class CumulusPolicyGenerator(ABC):
     def get_prefix_lists(self, device: Any) -> Sequence[IpPrefixList]:
         raise NotImplementedError()
 
+    def get_used_prefix_lists(self, device: Any, name_generator: PrefixListNameGenerator) -> Sequence[IpPrefixList]:
+        return get_used_prefix_lists(
+            prefix_lists=self.get_prefix_lists(device),
+            name_generator=name_generator,
+        )
+
     @abstractmethod
     def get_community_lists(self, device: Any) -> list[CommunityList]:
         raise NotImplementedError()
@@ -61,11 +67,13 @@ class CumulusPolicyGenerator(ABC):
 
     def generate_cumulus_rpl(self, device: Any) -> Iterator[Sequence[str]]:
         policies = self.get_policies(device)
+        prefix_list_name_generator = new_prefix_list_name_generator(policies)
+
         communities = {c.name: c for c in self.get_community_lists(device)}
         yield from self._cumulus_as_path_filters(device, policies)
         yield from self._cumulus_communities(device, communities, policies)
-        yield from self._cumulus_prefix_lists(device, policies)
-        yield from self._cumulus_policy_config(device, communities, policies)
+        yield from self._cumulus_prefix_lists(device, policies, prefix_list_name_generator)
+        yield from self._cumulus_policy_config(device, communities, policies, prefix_list_name_generator)
 
     def _cumulus_as_path_filters(
             self,
@@ -100,11 +108,12 @@ class CumulusPolicyGenerator(ABC):
                 ("le", str(match.less_equal)) if match.less_equal is not None else ()
             )
 
-    def _cumulus_prefix_lists(self, device: Any, policies: list[RoutingPolicy]) -> Iterable[Sequence[str]]:
-        plists = {p.name: p for p in get_used_prefix_lists(
-            prefix_lists=self.get_prefix_lists(device),
-            policies=policies,
-        )}
+    def _cumulus_prefix_lists(
+            self, device: Any,
+            policies: list[RoutingPolicy],
+            prefix_list_name_generator: PrefixListNameGenerator,
+    ) -> Iterable[Sequence[str]]:
+        plists = {p.name: p for p in self.get_used_prefix_lists(device, prefix_list_name_generator)}
         if not plists.values():
             return
 
@@ -114,7 +123,7 @@ class CumulusPolicyGenerator(ABC):
                 cond: SingleCondition[PrefixMatchValue]
                 for cond in statement.match.find_all(MatchField.ip_prefix):
                     for name in cond.value.names:
-                        mangled_name = mangle_ranged_prefix_list_name(
+                        mangled_name = prefix_list_name_generator.get_prefix_name(
                             name=name,
                             greater_equal=cond.value.greater_equal,
                             less_equal=cond.value.less_equal,
@@ -125,7 +134,7 @@ class CumulusPolicyGenerator(ABC):
                         precessed_names.add(mangled_name)
                 for cond in statement.match.find_all(MatchField.ipv6_prefix):
                     for name in cond.value.names:
-                        mangled_name = mangle_ranged_prefix_list_name(
+                        mangled_name = prefix_list_name_generator.get_prefix_name(
                             name=name,
                             greater_equal=cond.value.greater_equal,
                             less_equal=cond.value.less_equal,
@@ -217,6 +226,7 @@ class CumulusPolicyGenerator(ABC):
             self,
             device: Any,
             condition: SingleCondition[Any],
+            prefix_list_name_generator: PrefixListNameGenerator,
     ) -> Iterator[Sequence[str]]:
         if condition.field == MatchField.community:
             for comm_name in self._get_match_community_names(condition):
@@ -236,7 +246,7 @@ class CumulusPolicyGenerator(ABC):
             return
         if condition.field == MatchField.ip_prefix:
             for name in condition.value.names:
-                mangled_name = mangle_ranged_prefix_list_name(
+                mangled_name = prefix_list_name_generator.get_prefix_name(
                     name=name,
                     greater_equal=condition.value.greater_equal,
                     less_equal=condition.value.less_equal,
@@ -245,7 +255,7 @@ class CumulusPolicyGenerator(ABC):
             return
         if condition.field == MatchField.ipv6_prefix:
             for name in condition.value.names:
-                mangled_name = mangle_ranged_prefix_list_name(
+                mangled_name = prefix_list_name_generator.get_prefix_name(
                     name=name,
                     greater_equal=condition.value.greater_equal,
                     less_equal=condition.value.less_equal,
@@ -421,11 +431,12 @@ class CumulusPolicyGenerator(ABC):
             device: Any,
             policy: RoutingPolicy,
             statement: RoutingPolicyStatement,
+            prefix_list_name_generator: PrefixListNameGenerator,
     ) -> Iterable[Sequence[str]]:
         yield "route-map", policy.name, FRR_RESULT_MAP[statement.result], str(statement.number)
 
         for condition in statement.match:
-            for row in self._cumulus_policy_match(device, condition):
+            for row in self._cumulus_policy_match(device, condition, prefix_list_name_generator):
                 yield FRR_INDENT, *row
         for action in statement.then:
             for row in self._cumulus_policy_then(communities, device, action):
@@ -439,6 +450,7 @@ class CumulusPolicyGenerator(ABC):
             device: Any,
             communities: dict[str, CommunityList],
             policies: list[RoutingPolicy],
+            prefix_list_name_generator: PrefixListNameGenerator,
     ) -> Iterable[Sequence[str]]:
         """ Route maps configuration """
 
@@ -454,5 +466,7 @@ class CumulusPolicyGenerator(ABC):
                     raise RuntimeError(
                         f"Multiple statements have same number {statement.number} for policy `{policy.name}`: "
                         f"`{statement.name}` and `{applied_stmts[statement.number]}`")
-                yield from self._cumulus_policy_statement(communities, device, policy, statement)
+                yield from self._cumulus_policy_statement(
+                    communities, device, policy, statement, prefix_list_name_generator,
+                )
                 applied_stmts[statement.number] = statement.name

{annet-0.16.29 → annet-0.16.31}/annet/rpl_generators/entities.py

@@ -1,3 +1,4 @@
+from collections import defaultdict
 from collections.abc import Sequence
 from dataclasses import dataclass
 from enum import Enum
@@ -56,15 +57,27 @@ def mangle_united_community_list_name(values: Sequence[str]) -> str:
     return "_OR_".join(values)
 
 
-def mangle_ranged_prefix_list_name(name: str, greater_equal: Optional[int], less_equal: Optional[int]) -> str:
-    if greater_equal is less_equal is None:
-        return name
-    if greater_equal is None:
-        ge_str = "unset"
-    else:
-        ge_str = str(greater_equal)
-    if less_equal is None:
-        le_str = "unset"
-    else:
-        le_str = str(less_equal)
-    return f"{name}_{ge_str}_{le_str}"
+class PrefixListNameGenerator:
+    def __init__(self):
+        self._prefix_lists = defaultdict(set)
+
+    def add_prefix(self, name: str, greater_equal: Optional[int], less_equal: Optional[int]) -> None:
+        self._prefix_lists[name].add((greater_equal, less_equal))
+
+    def is_used(self, name: str):
+        return name in self._prefix_lists
+
+    def get_prefix_name(self, name: str, greater_equal: Optional[int], less_equal: Optional[int]) -> str:
+        if len(self._prefix_lists[name]) == 1:
+            return name
+        if greater_equal is less_equal is None:
+            return name
+        if greater_equal is None:
+            ge_str = "unset"
+        else:
+            ge_str = str(greater_equal)
+        if less_equal is None:
+            le_str = "unset"
+        else:
+            le_str = str(less_equal)
+        return f"{name}_{ge_str}_{le_str}"