pylegend 0.2.0__py3-none-any.whl → 0.2.2__py3-none-any.whl

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/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/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.dist-info → pylegend-0.2.2.dist-info}/METADATA

@@ -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.dist-info → pylegend-0.2.2.dist-info}/RECORD

@@ -38,10 +38,10 @@ pylegend/core/language/primitives/string.py,sha256=riI340Iytk7zfThbPIfnJ3dyGnOZy
 pylegend/core/language/tds_row.py,sha256=lrf129CatediTnbEh_d_IAIWjddoywxer_1L6fsBtUY,8175
 pylegend/core/project_cooridnates.py,sha256=t1G6gJzcRQRbCjGpCOk0vBiqd0F8Y_e15Nu4NuzYd6I,3751
 pylegend/core/request/__init__.py,sha256=IHt1dQ2lEa7ibDgZC2jUAnGXG_lG-UOC_I4V7JkC6qk,967
-pylegend/core/request/auth.py,sha256=XEl0VPvH-8_IU4hGL1Et7MJgSKFrBqJNVZEpIZo89Hg,1077
-pylegend/core/request/legend_client.py,sha256=B7Rx-y61o1t_n9j8lpVbLm4vOFCVAvjqt47UJs47PRM,2639
+pylegend/core/request/auth.py,sha256=IuoRYqolYPQA83lQuoOk7_7lPlJPvMeKovbpZwM8pm0,1701
+pylegend/core/request/legend_client.py,sha256=upAAQt9Zhtf_oczlfV-LoWR7cm7y5bsGVf2p2aU7s_E,2773
 pylegend/core/request/response_reader.py,sha256=TNMi2GKk4lkmf7VMeY5n2AwhxxuWbX4M8pgC2_pXFbE,1691
-pylegend/core/request/service_client.py,sha256=Zau4uh66qFvuNuf1Ksj5JSV86YdurJHh_FByQS4pLq8,3040
+pylegend/core/request/service_client.py,sha256=oNvMR6qNl5UcxpcAjIb4CCpd0MB3Z-Y5OJSyFPGeerM,3377
 pylegend/core/sql/__init__.py,sha256=LXTDJSDmHQXtnMDZouhZp9IZQVpY6ONkINbUYjtnMkE,578
 pylegend/core/sql/metamodel.py,sha256=B7aqblDyyiTsqWZUW2Is9_2FHTsYTkvhJ1FVxXxoSF8,20134
 pylegend/core/sql/metamodel_extension.py,sha256=A3B1anMf1x54sjBIjZJhQWXX5IKIsQXFYaL5vWJC4ow,15348
@@ -108,8 +108,8 @@ pylegend/extensions/tds/result_handler/to_pandas_df_result_handler.py,sha256=tvV
 pylegend/legend_api_tds_client.py,sha256=I0uRF7ADGvV5fHN-dEBIrdxTy7GC8jV91V62rgXn6c0,2319
 pylegend/utils/__init__.py,sha256=LXTDJSDmHQXtnMDZouhZp9IZQVpY6ONkINbUYjtnMkE,578
 pylegend/utils/class_utils.py,sha256=t4PpF3jAXS_D6p9TqlSppryNYNOuy5C-kbKn2Kgb4QU,973
-pylegend-0.2.0.dist-info/LICENSE,sha256=AGR96_qQPZO66Gjqq4G6r_g670K35VtW-IobTAkmZJM,11343
-pylegend-0.2.0.dist-info/LICENSE.spdx,sha256=i7TsBclLotUvMjx9vZ_6S8Pp0r4uknWGw1RwiKBBvQ4,207
-pylegend-0.2.0.dist-info/METADATA,sha256=UwGXNF0RYKLzLFT2er91aejVAe2i0JoWwlH0Pk34-qs,4216
-pylegend-0.2.0.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
-pylegend-0.2.0.dist-info/RECORD,,
+pylegend-0.2.2.dist-info/LICENSE,sha256=AGR96_qQPZO66Gjqq4G6r_g670K35VtW-IobTAkmZJM,11343
+pylegend-0.2.2.dist-info/LICENSE.spdx,sha256=i7TsBclLotUvMjx9vZ_6S8Pp0r4uknWGw1RwiKBBvQ4,207
+pylegend-0.2.2.dist-info/METADATA,sha256=lmfSjBstKvYcjXXTrlAkhGfvA3llmC_iOxK2SYbx8t8,4267
+pylegend-0.2.2.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+pylegend-0.2.2.dist-info/RECORD,,

{pylegend-0.2.0.dist-info → pylegend-0.2.2.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.7.0
+Generator: poetry-core 1.8.1
 Root-Is-Purelib: true
 Tag: py3-none-any