traceback printer

Author: hellt

main.py

@@ -1,19 +1,22 @@
 from src.banner import banner
 from src.client import EDAClient
+from src.interface import interface
 from src.logging import setup_logging
 
 setup_logging()
 
 
 def main():
-    eda = EDAClient(base_url="https://devbox.panda-cobra.ts.net")
-
     my_banner = banner(
         ns="clab-vlan",
         name="my-banner",
         motd_text="Let's have some model driven automation going on!",
     )
-    eda.add_to_transaction_delete(my_banner)
+    my_interface = interface(ns="clab-vlan", name="my-interface")
+
+    eda = EDAClient(base_url="https://devbox.panda-cobra.ts.net")
+    # eda.add_to_transaction_delete(my_banner)
+    eda.add_to_transaction_replace(my_interface)
     _ = eda.commit_transaction()
 
 

src/client.py

@@ -70,6 +70,11 @@ def add_to_transaction_replace(self, resource: BaseModel) -> None:
 
         self.add_to_transaction(resource, _REPLACE)
 
+    def add_to_transaction_modify(self, resource: BaseModel) -> None:
+        """Add resource to the modify list of a transaction"""
+
+        self.add_to_transaction(resource, _MODIFY)
+
     def add_to_transaction_delete(self, resource: BaseModel) -> None:
         """Add resource to the delete list of a transaction"""
 
@@ -140,7 +145,7 @@ def commit_transaction(self) -> Any:
             exclude_unset=True, exclude_none=True, exclude_defaults=True
         )
 
-        logger.info(f"Committing transaction: {content}")
+        # logger.info(f"Committing transaction: {content}")
 
         response = self.post(
             url=self.transaction_endpoint,

src/interface.py

@@ -0,0 +1,24 @@
+import models.com.nokia.eda.interfaces.v1alpha1 as iface
+
+
+def interface(ns: str, name: str) -> iface.Interface:
+    iface_ = iface.Interface(
+        apiVersion="interfaces.eda.nokia.com/v1alpha1",
+        kind="Interface",
+        metadata=iface.Metadata(
+            name=name,
+            namespace=ns,
+        ),
+        spec=iface.SpecModel(
+            description="This is a test interface",
+            mtu=1500,
+            members=[
+                iface.Member(
+                    interface="ethernet-1-1",
+                    node="srl1",
+                ),
+            ],
+        ),
+    )
+
+    return iface_

src/logging.py

@@ -1,6 +1,7 @@
 import logging
 
 from rich.logging import RichHandler
+from rich.traceback import install
 
 
 def setup_logging():
@@ -12,5 +13,7 @@ def setup_logging():
         format="%(message)s",
         handlers=[RichHandler(rich_tracebacks=True, markup=True)],
     )
-    logger = logging.getLogger(__name__)
+
     logging.getLogger("httpx").setLevel(logging.WARNING)
+
+    install(show_locals=False)