Handling exception of invalid header and strip token (#825)

CHANGELOG.md

@@ -1,3 +1,11 @@
+## [UNRELEASED] neptune-client 0.14.3
+
+## Features
+- Stripping whitespaces from Neptune API Token ([#825](https://github.com/neptune-ai/neptune-client/pull/825))
+
+## Fixes
+- Raise proper exception when invalid token were provided ([#825](https://github.com/neptune-ai/neptune-client/pull/825))
+
 ## neptune-client 0.14.2
 
 ### Features

neptune/new/internal/backends/utils.py

@@ -72,6 +72,7 @@
     ClientHttpError,
     NeptuneFeatureNotAvailableException,
     MetadataInconsistency,
+    NeptuneInvalidApiTokenException,
 )
 from neptune.new.internal.backends.api_model import ClientConfig
 from neptune.new.internal.operation import Operation, CopyAttribute
@@ -99,6 +100,10 @@ def wrapper(*args, **kwargs):
 
             try:
                 return func(*args, **kwargs)
+            except requests.exceptions.InvalidHeader as e:
+                if "X-Neptune-Api-Token" in e.args[0]:
+                    raise NeptuneInvalidApiTokenException()
+                raise
             except requests.exceptions.SSLError as e:
                 raise SSLError() from e
             except (

neptune/new/internal/credentials.py

@@ -46,6 +46,7 @@ def from_token(cls, api_token: Optional[str] = None) -> "Credentials":
         if api_token is None:
             raise NeptuneMissingApiTokenException()
 
+        api_token = api_token.strip()
         token_dict = Credentials._api_token_to_dict(api_token)
         # TODO: Consider renaming 'api_address' (breaking backward compatibility)
         if "api_address" not in token_dict: