nebu 0.1.88__py3-none-any.whl → 0.1.93__py3-none-any.whl

nebu/namespaces/namespace.py

@@ -3,6 +3,7 @@ from typing import Dict, List, Optional
 import requests
 
 from nebu.config import GlobalConfig
+from nebu.logging import logger
 from nebu.namespaces.models import (
     V1Namespace,
     V1NamespaceMetaRequest,
@@ -22,6 +23,7 @@ class Namespace:
         labels: Optional[Dict[str, str]] = None,
         owner: Optional[str] = None,
         config: Optional[GlobalConfig] = None,
+        api_key: Optional[str] = None,
     ):
         self.config = config or GlobalConfig.read()
         if not self.config:
@@ -30,7 +32,7 @@ class Namespace:
         if not current_server:
             raise ValueError("No server config found")
         self.current_server = current_server
-        self.api_key = current_server.api_key
+        self.api_key = api_key or current_server.api_key
         self.nebu_host = current_server.server
         self.name = name
         self.labels = labels
@@ -43,7 +45,7 @@ class Namespace:
         )
         response.raise_for_status()
         resp_json = response.json()
-        print(resp_json)
+        logger.debug(resp_json)
         existing_namespaces = V1Namespaces.model_validate(resp_json)
         self.namespace: Optional[V1Namespace] = next(
             (
@@ -56,7 +58,7 @@ class Namespace:
 
         # If not found, create
         if not self.namespace:
-            print(f"Creating namespace {name}")
+            logger.info(f"Creating namespace {name}")
             # Create metadata and namespace request
             metadata = V1NamespaceMetaRequest(name=name, labels=labels, owner=owner)
 
@@ -71,20 +73,23 @@ class Namespace:
             )
             create_response.raise_for_status()
             self.namespace = V1Namespace.model_validate(create_response.json())
-            print(f"Created Namespace {self.namespace.metadata.name}")
+            logger.info(f"Created Namespace {self.namespace.metadata.name}")
         else:
-            print(f"Found Namespace {self.namespace.metadata.name}, no update needed")
+            logger.info(
+                f"Found Namespace {self.namespace.metadata.name}, no update needed"
+            )
 
     @classmethod
     def load(
         cls,
         name: str,
         config: Optional[GlobalConfig] = None,
+        api_key: Optional[str] = None,
     ):
         """
         Get a Namespace from the remote server.
         """
-        namespaces = cls.get(name=name, config=config)
+        namespaces = cls.get(name=name, config=config, api_key=api_key)
         if not namespaces:
             raise ValueError("Namespace not found")
         namespace_v1 = namespaces[0]
@@ -97,7 +102,7 @@ class Namespace:
         out.current_server = out.config.get_current_server_config()
         if not out.current_server:
             raise ValueError("No server config found")
-        out.api_key = out.current_server.api_key
+        out.api_key = api_key or out.current_server.api_key
         out.nebu_host = out.current_server.server
         out.namespaces_url = f"{out.nebu_host}/v1/namespaces"
         out.name = name
@@ -109,6 +114,7 @@ class Namespace:
         cls,
         name: Optional[str] = None,
         config: Optional[GlobalConfig] = None,
+        api_key: Optional[str] = None,
     ) -> List[V1Namespace]:
         """
         Get a list of Namespaces that optionally match the name filter.
@@ -119,11 +125,12 @@ class Namespace:
         current_server = config.get_current_server_config()
         if not current_server:
             raise ValueError("No server config found")
+        api_key = api_key or current_server.api_key
         namespaces_url = f"{current_server.server}/v1/namespaces"
 
         response = requests.get(
             namespaces_url,
-            headers={"Authorization": f"Bearer {current_server.api_key}"},
+            headers={"Authorization": f"Bearer {api_key}"},
         )
         response.raise_for_status()