annet 0.0__py3-none-any.whl

annet_generators/example/lldp.py

@@ -0,0 +1,53 @@
+
+from typing import List
+
+from annet.generators import BaseGenerator, PartialGenerator
+from annet.storage import Storage
+
+
+# ====
+class Lldp(PartialGenerator):
+    TAGS = ["mgmt", "lldp"]
+
+    def acl_huawei(self, device):
+        return """
+            lldp
+        """
+
+    def run_huawei(self, device):
+        yield "lldp enable"
+
+        if device.hw.CE:
+            yield "lldp transmit interval 10"
+
+    def acl_nexus(self, device):
+        return """
+            feature lldp
+            lldp
+        """
+
+    def run_nexus(self, device):
+        yield "feature lldp"
+        yield "lldp timer 10"
+
+    def acl_juniper(self, _):
+        return """
+        protocols    %cant_delete
+            lldp
+                *
+        """
+
+    def run_juniper(self, device):
+        with self.multiblock("protocols", "lldp"):
+            yield """
+                neighbour-port-info-display port-id
+                port-description-type interface-alias
+                port-id-subtype interface-name
+                interface all
+            """
+
+
+def get_generators(store: Storage) -> List[BaseGenerator]:
+    return [
+        Lldp(store),
+    ]