lsrestclient 1.3.0__py3-none-any.whl → 1.3.2__py3-none-any.whl

lsrestclient/auth.py

@@ -1,10 +1,14 @@
 from lsrestclient import LsRestClient, DownStreamError
 
 
-def auth_am_login_apikey(org_id: str, api_key: str) -> str:
+def auth_am_login(**kwargs) -> str:
     am = LsRestClient.from_env("AM_BASE_URL", "am", True)
-    r = am.post("/auth/login", body=dict(ORG_ID=org_id, API_KEY=api_key))
+    r = am.post("/auth/login", body=kwargs)
     if r.status_code == 200:
         return r.json().get("access_token")
     else:
         raise DownStreamError("/auth/login", r.status_code, r.content)
+
+
+def auth_am_login_apikey(org_id: str, api_key: str) -> str:
+    return auth_am_login(ORG_ID=org_id, API_KEY=api_key)

lsrestclient/exceptions.py

@@ -13,13 +13,13 @@ class ConnectionError(Exception):
     """Exception class for connection errors.
 
     Args:
-        url (Optional[str]): The URL that the connection could not be established to.
+            url (Optional[str]): The URL that the connection could not be established to.
 
     Attributes:
-        url (Optional[str]): The URL that the connection could not be established to.
+            url (Optional[str]): The URL that the connection could not be established to.
 
     Raises:
-        ConnectionError: If a connection could not be established to the given URL.
+            ConnectionError: If a connection could not be established to the given URL.
 
     """
 
@@ -51,8 +51,12 @@ def raise_errors(r, exceptions: List[Type[Exception]]):
         try:
             json = r.json()
             detail = pydash.get(json, "detail", json)
-            error_class = pydash.get(detail, "error_class")
-            payload = pydash.get(detail, "error_payload", {})
+            error_class = pydash.get(detail, "error_class", None)
+            if error_class is not None:
+                payload = pydash.get(detail, "error_payload", {})
+            else:
+                error_class = pydash.get(detail, "ERROR_CLASS", None)
+                payload = {}
 
             if error_class in exceptions_by_class:
                 e = exceptions_by_class[error_class](**payload)

lsrestclient/mock.py

@@ -7,37 +7,40 @@ from lsrestclient import LsRestClient
 
 @dataclasses.dataclass
 class LsRestClientMockModel:
-    mock_name: str
-    client: LsRestClient
-    mock: MagicMock
+	mock_name: str
+	client: LsRestClient
+	mock: MagicMock
 
 
 class LsRestClientMocker(object):
-    def __init__(self):
-        super().__init__()
-        self.mocks = {}
+	def __init__(self):
+		super().__init__()
+		self.mocks = {}
 
-    def mock(self, client: LsRestClient, method: str, url: str, *args, **kwargs):
-        mock_name = LsRestClient.mock_name(client.name, method, url)
-        mock = MagicMock(*args, **kwargs)
-        client.mock(mock_name, mock)
-        self.mocks[mock_name] = LsRestClientMockModel(client=client, mock_name=mock_name, mock=mock)
-        return mock
+	def mock(self, client: LsRestClient | str, method: str, url: str, *args, **kwargs):
+		if isinstance(client, str):
+			client = LsRestClient.client(client)
 
-    def unmock_all(self):
-        for mock_name, mock_model in self.mocks.items():
-            mock_model.client.unmock(mock_name)
-        self.mocks = {}
+		mock_name = LsRestClient.mock_name(client.name, method, url)
+		mock = MagicMock(*args, **kwargs)
+		client.mock(mock_name, mock)
+		self.mocks[mock_name] = LsRestClientMockModel(client=client, mock_name=mock_name, mock=mock)
+		return mock
 
-    def unmock(self, client: LsRestClient, method: str, url: str):
-        mock_name = LsRestClient.mock_name(client.name, method, url)
-        mock_model = self.mocks[mock_name]
-        mock_model.client.unmock(mock_name)
-        del self.mocks[mock_name]
+	def unmock_all(self):
+		for mock_name, mock_model in self.mocks.items():
+			mock_model.client.unmock(mock_name)
+		self.mocks = {}
+
+	def unmock(self, client: LsRestClient, method: str, url: str):
+		mock_name = LsRestClient.mock_name(client.name, method, url)
+		mock_model = self.mocks[mock_name]
+		mock_model.client.unmock(mock_name)
+		del self.mocks[mock_name]
 
 
 @contextmanager
 def lsrestclient_mock_context():
-    mocker = LsRestClientMocker()
-    yield mocker
-    mocker.unmock_all()
+	mocker = LsRestClientMocker()
+	yield mocker
+	mocker.unmock_all()

{lsrestclient-1.3.0.dist-info → lsrestclient-1.3.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lsrestclient
-Version: 1.3.0
+Version: 1.3.2
 Summary: REST Api Client
 Author: mba
 Author-email: bartel@electronic-shop.lu

{lsrestclient-1.3.0.dist-info → lsrestclient-1.3.2.dist-info}/RECORD

@@ -1,13 +1,13 @@
 lsrestclient/__init__.py,sha256=lI62SHmG0m-iukB5UEwdN5dO4cKnDasOt-TmR1rgaWI,72
-lsrestclient/auth.py,sha256=-XT9OajBudLDZJ7tZyM0E5EXF2R1Kc-jyeKMf24wyJA,403
+lsrestclient/auth.py,sha256=n9f5GkKDbzin2jHNv9D7QBb6ZLEjamfbkzzlGpNdjLE,468
 lsrestclient/client.py,sha256=cQXgRZ7oktHR7u8i9sib9FhBV5JxdmA-QOz6UGkYzUc,9922
 lsrestclient/contexts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lsrestclient/contexts/bearer_token.py,sha256=GZZOzAI2Ng_9DvFCbhv1Wwb8wJ1AYCca3fQeNtt2NaU,753
-lsrestclient/exceptions.py,sha256=6Ad0XjS2H17xCRwPiIsB0fsvU4liFS6OPAun-A4Hpuk,1926
+lsrestclient/exceptions.py,sha256=Ze_UZutdTxmKXq8brvciZclUIkEGpmFrybPzk6TVk4Q,2105
 lsrestclient/fixtures.py,sha256=dFkAYQXL6xqTOwRofb03Nsu_cIjq1sG10Rh8dxWfz9s,239
-lsrestclient/mock.py,sha256=4HOVjYLdctKBLTKf02fw8glrVncZtzEW0h_-gVDC3Jc,1253
+lsrestclient/mock.py,sha256=YanA_CQdaIMd99VkHaWDv3ro59fjBJsNUTZb14rC5aE,1213
 lsrestclient/response.py,sha256=IXjIgb7ySF-1FdAtnOKRyF3k9LJRB3OFyzIIZI1GNxM,1816
-lsrestclient-1.3.0.dist-info/METADATA,sha256=aDaZBE450bqszB6-jgqvxubT-mq-zHspmOhPVkqlBro,6656
-lsrestclient-1.3.0.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
-lsrestclient-1.3.0.dist-info/entry_points.txt,sha256=7lN1XN3lq5Jv5PlpOdIlFrLlFlwzE5MaEWSgMhKASOM,47
-lsrestclient-1.3.0.dist-info/RECORD,,
+lsrestclient-1.3.2.dist-info/METADATA,sha256=x5C7teRjs6eW2uHkRrCQjySg9Db1KIeS1dQmB7y88Dc,6656
+lsrestclient-1.3.2.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
+lsrestclient-1.3.2.dist-info/entry_points.txt,sha256=7lN1XN3lq5Jv5PlpOdIlFrLlFlwzE5MaEWSgMhKASOM,47
+lsrestclient-1.3.2.dist-info/RECORD,,