pylegend 0.2.0__tar.gz → 0.2.2__tar.gz

{pylegend-0.2.0 → pylegend-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pylegend
-Version: 0.2.0
+Version: 0.2.2
 Summary: Python language binding for Legend data management platform
 Home-page: https://github.com/finos/pylegend
 License: Apache-2.0
@@ -13,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: ijson (>=3.1.4)
 Requires-Dist: numpy (>=1.20.0) ; python_version < "3.12"
 Requires-Dist: numpy (>=1.26.0) ; python_version >= "3.12"

{pylegend-0.2.0 → pylegend-0.2.2}/pylegend/core/request/auth.py

@@ -13,6 +13,9 @@
 # limitations under the License.
 
 from abc import ABCMeta, abstractmethod
+
+from requests import PreparedRequest
+
 from pylegend._typing import (
     PyLegendSequence,
     PyLegendOptional,
@@ -22,6 +25,7 @@ from requests.auth import AuthBase
 __all__: PyLegendSequence[str] = [
     "AuthScheme",
     "LocalhostEmptyAuthScheme",
+    "BearerAuthScheme"
 ]
 
 
@@ -36,3 +40,22 @@ class LocalhostEmptyAuthScheme(AuthScheme):
 
     def get_auth_base(self) -> PyLegendOptional[AuthBase]:
         return None
+
+
+class BearerAuth(AuthBase):
+    def __init__(self, headerName: str, token: str) -> None:
+        self.headerName = headerName
+        self.token = token
+
+    def __call__(self, r: PreparedRequest) -> PreparedRequest:
+        r.headers[self.headerName] = self.token
+        return r
+
+
+class BearerAuthScheme(AuthScheme):
+    def __init__(self, headerName: str, token: str) -> None:
+        self.headerName = headerName
+        self.token = token
+
+    def get_auth_base(self) -> PyLegendOptional[AuthBase]:
+        return BearerAuth(self.headerName, self.token)

{pylegend-0.2.0 → pylegend-0.2.2}/pylegend/core/request/legend_client.py

@@ -35,12 +35,18 @@ class LegendClient(ServiceClient):
             self,
             host: str,
             port: int,
-            auth_scheme: AuthScheme = LocalhostEmptyAuthScheme(),
             secure_http: bool = True,
+            path_prefix: PyLegendOptional[str] = "/api",
+            auth_scheme: AuthScheme = LocalhostEmptyAuthScheme(),
             retry_count: int = 2
     ) -> None:
         super().__init__(
-            host=host, port=port, auth_scheme=auth_scheme, secure_http=secure_http, retry_count=retry_count
+            host=host,
+            port=port,
+            secure_http=secure_http,
+            path_prefix=path_prefix,
+            auth_scheme=auth_scheme,
+            retry_count=retry_count
         )
 
     def get_sql_string_schema(
@@ -49,7 +55,7 @@ class LegendClient(ServiceClient):
     ) -> PyLegendSequence[TdsColumn]:
         response = super()._execute_service(
             method=RequestMethod.POST,
-            path="api/sql/v1/execution/getSchemaFromQueryString",
+            path="sql/v1/execution/getSchemaFromQueryString",
             data=sql,
             headers={"Content-Type": "text/plain"},
             stream=False
@@ -64,7 +70,7 @@ class LegendClient(ServiceClient):
     ) -> ResponseReader:
         iter_content = super()._execute_service(
             method=RequestMethod.POST,
-            path="api/sql/v1/execution/executeQueryString",
+            path="sql/v1/execution/executeQueryString",
             data=sql,
             headers={"Content-Type": "text/plain"},
             stream=True

{pylegend-0.2.0 → pylegend-0.2.2}/pylegend/core/request/service_client.py

@@ -40,10 +40,19 @@ class RequestMethod(Enum):
 
 class ServiceClient(metaclass=ABCMeta):
 
-    def __init__(self, host: str, port: int, auth_scheme: AuthScheme, secure_http: bool, retry_count: int) -> None:
+    def __init__(
+            self,
+            host: str,
+            port: int,
+            secure_http: bool,
+            path_prefix: PyLegendOptional[str],
+            auth_scheme: AuthScheme,
+            retry_count: int
+    ) -> None:
         self.__host = host
         self.__port = port
         self.__auth_scheme = auth_scheme
+        self.__path_prefix = path_prefix
         self.__secure_http = secure_http
         if retry_count < 1:
             raise ValueError("Retry count should be a number greater than 1. Got " + str(retry_count))
@@ -66,7 +75,9 @@ class ServiceClient(metaclass=ABCMeta):
     ) -> requests.Response:
 
         scheme = "https" if self.__secure_http else "http"
-        url = f"{scheme}://{self.__host}:{self.__port}/{path}"
+        prefix = (self.__path_prefix if self.__path_prefix.startswith("/") else f"/{self.__path_prefix}") \
+            if self.__path_prefix is not None else ""
+        url = f"{scheme}://{self.__host}:{self.__port}{prefix}/{path}"
 
         request = requests.Request(
             method=method.name,

{pylegend-0.2.0 → pylegend-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pylegend"
-version = "0.2.0"
+version = "0.2.2"
 description = "Python language binding for Legend data management platform"
 authors = ["PyLegend Maintainers <legend@finos.org>"]
 license = "Apache-2.0"