fix

Author: ppandit-sfdc

src/datacustomcode/cli.py

@@ -50,7 +50,11 @@ def version():
 @click.option("--client-id", prompt=True)
 @click.option("--client-secret", prompt=True)
 @click.option("--login-url", prompt=True)
-@click.option("--dataspace", default="default", help="Dataspace name (optional, for non-default dataspaces)")
+@click.option(
+    "--dataspace",
+    default="default",
+    help="Dataspace name (optional, for non-default dataspaces)",
+)
 def configure(
     username: str,
     password: str,

src/datacustomcode/credentials.py

@@ -65,7 +65,9 @@ def from_env(cls) -> Credentials:
             credentials_data = {}
             for k, v in ENV_CREDENTIALS.items():
                 if k == "dataspace":
-                    credentials_data[k] = os.environ.get(v)
+                    dataspace_value = os.environ.get(v)
+                    if dataspace_value is not None:
+                        credentials_data[k] = dataspace_value
                 else:
                     credentials_data[k] = os.environ[v]
             return cls(**credentials_data)
@@ -103,7 +105,7 @@ def update_ini(self, profile: str = "default", ini_file: str = INI_FILE):
         config[profile]["client_id"] = self.client_id
         config[profile]["client_secret"] = self.client_secret
         config[profile]["login_url"] = self.login_url
-        
+
         if self.dataspace is not None:
             config[profile]["dataspace"] = self.dataspace
 

src/datacustomcode/io/reader/query_api.py

@@ -88,7 +88,7 @@ def __init__(
             credentials.client_id,
             credentials.client_secret,
         ]
-        
+
         connection_kwargs = {}
         if credentials.dataspace is not None:
             connection_kwargs["dataspace"] = credentials.dataspace