fiddling

hellt · hellt · commit 900bb524d890 · 2025-01-19T23:19:11.000+01:00
diff --git a/README.md b/README.md
@@ -9,6 +9,13 @@ uv tool install 'datamodel-code-generator[http]'
 To generate a model:
 
 ```bash
+# core
+datamodel-codegen --url https://raw.githubusercontent.com/eda-labs/openapi/refs/heads/main/core/core.json \
+--output-model-type pydantic_v2.BaseModel \
+--use-annotated \
+--enum-field-as-literal all \
+--output models/core.py
+
 # interfaces
 datamodel-codegen --url https://raw.githubusercontent.com/eda-labs/openapi/refs/heads/main/apps/interfaces.eda.nokia.com/v1alpha1/interfaces.json \
 --output-model-type pydantic_v2.BaseModel \
diff --git a/main.py b/main.py
@@ -6,6 +6,7 @@
     Metadata,
     SpecModel,
 )
+from src.banner import banner
 from src.client import Client
 
 logging.basicConfig(
@@ -19,6 +20,10 @@ def main():
     c.auth()
     logger.info(f"Access Token: {c.token}")
 
+    my_banner = banner("This is a test banner")
+    # print(my_banner.model_dump())
+    c.add_to_transaction_create(my_banner)
+
     # iface = Interface(
     #     apiVersion="interfaces.eda.nokia.com/v1alpha1",
     #     kind="Interface",
diff --git a/src/banner.py b/src/banner.py
@@ -1,4 +1,22 @@
-def set_banner(banner):
+from models.com.nokia.eda.siteinfo.v1alpha1 import Banner, Metadata, Spec
+
+
+def banner(motd_text: str) -> Banner:
     """
-    Set the banner for the SSH session.
+    Create a banner
     """
+
+    banner = Banner(
+        apiVersion="siteinfo.eda.nokia.com/v1alpha1",
+        kind="Banner",
+        metadata=Metadata(
+            name="banner",
+            namespace="clab-vlan",
+            labels={"app": "banner"},
+        ),
+        spec=Spec(
+            motd=motd_text,
+        ),
+    )
+
+    return banner
diff --git a/src/client.py b/src/client.py
@@ -1,4 +1,16 @@
+from typing import Any, Optional
+
 import httpx
+from pydantic import BaseModel
+
+from models.com.nokia.eda.siteinfo.v1alpha1 import Banner
+from models.core import (
+    Transaction,
+    TransactionContent,
+    TransactionCr,
+    TransactionType,
+    TransactionValue,
+)
 
 EDA_VERSION = "v24.12.1"
 
@@ -15,11 +27,31 @@ def __init__(self, base_url: str):
         self.base_url = base_url
         self.kc_url = f"{self.base_url}/core/httpproxy/v1/keycloak/"
         self.token = ""
+        self.transaction: Optional[Transaction] = None
 
     def auth(self) -> None:
         """Authenticate and get access token"""
         self.token = _get_access_token(self)
 
+    def add_to_transaction_create(self, resource: BaseModel) -> None:
+        """Add resource to transaction"""
+        content = TransactionContent(
+            **resource.model_dump(exclude_unset=True, exclude_defaults=True)
+        )
+
+        if self.transaction is None:
+            self.transaction = Transaction(
+                crs=[
+                    TransactionCr(
+                        type=TransactionType(create=TransactionValue(value=content))
+                    )
+                ],
+                description="",
+                dryRun=False,
+            )
+        # else:
+        #     self.transaction.resources.append(resource)
+
 
 def _get_client_secret(kc_url: str) -> str:
     with httpx.Client(verify=False) as client:
