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

pylegend/__init__.py

@@ -15,14 +15,16 @@
 from pylegend._typing import (
     PyLegendSequence,
 )
-from pylegend.legend_api_tds_client import (
-    LegendApiTdsClient,
-    legend_api_tds_client,
+from pylegend.legacy_api_tds_client import (
+    LegacyApiTdsClient,
+    legacy_api_tds_client,
 )
 from pylegend.core.request import (
     LegendClient,
     AuthScheme,
     LocalhostEmptyAuthScheme,
+    HeaderTokenAuthScheme,
+    CookieAuthScheme,
     ResponseReader,
 )
 from pylegend.core.project_cooridnates import (
@@ -36,12 +38,14 @@ from pylegend.core.language import agg
 __all__: PyLegendSequence[str] = [
     "__version__",
 
-    "LegendApiTdsClient",
-    "legend_api_tds_client",
+    "LegacyApiTdsClient",
+    "legacy_api_tds_client",
 
     "LegendClient",
     "AuthScheme",
     "LocalhostEmptyAuthScheme",
+    "HeaderTokenAuthScheme",
+    "CookieAuthScheme",
     "ResponseReader",
 
     "VersionedProjectCoordinates",

pylegend/core/{databse → database}/sql_to_string/__init__.py

@@ -13,9 +13,9 @@
 # limitations under the License.
 
 from pylegend._typing import PyLegendSequence
-from pylegend.core.databse.sql_to_string.generator import SqlToStringGenerator
-from pylegend.core.databse.sql_to_string.config import SqlToStringConfig, SqlToStringFormat
-from pylegend.core.databse.sql_to_string.db_extension import SqlToStringDbExtension
+from pylegend.core.database.sql_to_string.generator import SqlToStringGenerator
+from pylegend.core.database.sql_to_string.config import SqlToStringConfig, SqlToStringFormat
+from pylegend.core.database.sql_to_string.db_extension import SqlToStringDbExtension
 
 __all__: PyLegendSequence[str] = [
     "SqlToStringGenerator",

pylegend/core/{databse → database}/sql_to_string/db_extension.py

@@ -18,7 +18,7 @@ from pylegend._typing import (
     PyLegendList,
     PyLegendSequence
 )
-from pylegend.core.databse.sql_to_string.config import SqlToStringConfig
+from pylegend.core.database.sql_to_string.config import SqlToStringConfig
 from pylegend.core.sql.metamodel import (
     QuerySpecification,
     Select,
@@ -130,6 +130,7 @@ from pylegend.core.sql.metamodel_extension import (
     MinuteExpression,
     SecondExpression,
     EpochExpression,
+    WindowExpression,
 )
 
 
@@ -448,6 +449,8 @@ def expression_processor(
         return extension.process_second_expression(expression, config)
     elif isinstance(expression, EpochExpression):
         return extension.process_epoch_expression(expression, config)
+    elif isinstance(expression, WindowExpression):
+        return extension.process_window_expression(expression, config)
 
     else:
         raise ValueError("Unsupported expression type: " + str(type(expression)))  # pragma: no cover
@@ -674,7 +677,7 @@ def function_call_processor(
 
     window = ""
     if function_call.window:
-        window = " OVER" + extension.process_window(function_call.window, config)
+        window = " " + extension.process_window(function_call.window, config)
 
     first_sep = config.format.separator(1) if function_call.arguments else ""
     sep0 = config.format.separator(0) if function_call.arguments else ""
@@ -838,14 +841,14 @@ def window_processor(
     if window.windowRef:
         return window.windowRef
 
-    partitions = " PARTITION BY " + (", ".join([extension.process_expression(e, config) for e in window.partitions])) \
+    partitions = "PARTITION BY " + (", ".join([extension.process_expression(e, config) for e in window.partitions])) \
         if window.partitions else ""
 
-    order_by = " ORDER BY " + (", ".join([extension.process_sort_item(o, config) for o in window.orderBy])) \
+    order_by = "ORDER BY " + (", ".join([extension.process_sort_item(o, config) for o in window.orderBy])) \
         if window.orderBy else ""
 
     # TODO: Handle window frame
-    return f"{partitions}{order_by}{''}"
+    return f"OVER ({partitions}{' ' if (partitions != '') and (order_by != '') else ''}{order_by}){''}"
 
 
 def table_function_processor(
@@ -1162,6 +1165,9 @@ class SqlToStringDbExtension:
     def process_epoch_expression(self, expr: EpochExpression, config: SqlToStringConfig) -> str:
         return f"DATE_PART('epoch', {self.process_expression(expr.value, config)})"
 
+    def process_window_expression(self, expr: WindowExpression, config: SqlToStringConfig) -> str:
+        return f"{self.process_expression(expr.nested, config)} {self.process_window(expr.window, config)}"
+
     def process_qualified_name(self, qualified_name: QualifiedName, config: SqlToStringConfig) -> str:
         return qualified_name_processor(qualified_name, self, config)
 

pylegend/core/{databse → database}/sql_to_string/generator.py

@@ -18,8 +18,8 @@ from pylegend._typing import (
     PyLegendDict
 )
 from pylegend.utils.class_utils import find_sub_classes
-from pylegend.core.databse.sql_to_string.config import SqlToStringConfig
-from pylegend.core.databse.sql_to_string.db_extension import SqlToStringDbExtension
+from pylegend.core.database.sql_to_string.config import SqlToStringConfig
+from pylegend.core.database.sql_to_string.db_extension import SqlToStringDbExtension
 from pylegend.core.sql.metamodel import QuerySpecification
 
 __all__: PyLegendSequence[str] = [

pylegend/core/language/__init__.py

@@ -15,7 +15,7 @@
 from pylegend._typing import (
     PyLegendSequence,
 )
-from pylegend.core.language.primitives import (
+from pylegend.core.language.shared.primitives import (
     PyLegendPrimitive,
     PyLegendPrimitiveOrPythonPrimitive,
     PyLegendBoolean,
@@ -27,7 +27,7 @@ from pylegend.core.language.primitives import (
     PyLegendDateTime,
     PyLegendStrictDate,
 )
-from pylegend.core.language.expression import (
+from pylegend.core.language.shared.expression import (
     PyLegendExpressionBooleanReturn,
     PyLegendExpressionStringReturn,
     PyLegendExpressionNumberReturn,
@@ -37,7 +37,7 @@ from pylegend.core.language.expression import (
     PyLegendExpressionDateTimeReturn,
     PyLegendExpressionStrictDateReturn,
 )
-from pylegend.core.language.literal_expressions import (
+from pylegend.core.language.shared.literal_expressions import (
     PyLegendBooleanLiteralExpression,
     PyLegendStringLiteralExpression,
     PyLegendIntegerLiteralExpression,
@@ -46,7 +46,7 @@ from pylegend.core.language.literal_expressions import (
     PyLegendStrictDateLiteralExpression,
     convert_literal_to_literal_expression,
 )
-from pylegend.core.language.column_expressions import (
+from pylegend.core.language.shared.column_expressions import (
     PyLegendColumnExpression,
     PyLegendBooleanColumnExpression,
     PyLegendStringColumnExpression,
@@ -57,9 +57,9 @@ from pylegend.core.language.column_expressions import (
     PyLegendDateTimeColumnExpression,
     PyLegendStrictDateColumnExpression,
 )
-from pylegend.core.language.tds_row import TdsRow
-from pylegend.core.language.aggregate_specification import AggregateSpecification, agg
-from pylegend.core.language.primitive_collection import (
+from pylegend.core.language.legacy_api.legacy_api_tds_row import LegacyApiTdsRow
+from pylegend.core.language.legacy_api.aggregate_specification import LegacyApiAggregateSpecification, agg
+from pylegend.core.language.shared.primitive_collection import (
     PyLegendPrimitiveCollection,
     PyLegendIntegerCollection,
     PyLegendFloatCollection,
@@ -71,7 +71,7 @@ from pylegend.core.language.primitive_collection import (
     PyLegendStrictDateCollection,
     create_primitive_collection,
 )
-from pylegend.core.language.functions import (
+from pylegend.core.language.shared.functions import (
     today,
     now,
 )
@@ -115,8 +115,8 @@ __all__: PyLegendSequence[str] = [
     "PyLegendStrictDateLiteralExpression",
     "convert_literal_to_literal_expression",
 
-    "TdsRow",
-    "AggregateSpecification",
+    "LegacyApiTdsRow",
+    "LegacyApiAggregateSpecification",
     "agg",
 
     "PyLegendPrimitiveCollection",

pylegend/core/language/legacy_api/__init__.py

@@ -0,0 +1,13 @@
+# Copyright 2025 Goldman Sachs
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.

pylegend/core/language/{aggregate_specification.py → legacy_api/aggregate_specification.py}

@@ -16,27 +16,27 @@ from pylegend._typing import (
     PyLegendSequence,
     PyLegendCallable
 )
-from pylegend.core.language.primitive_collection import PyLegendPrimitiveCollection
+from pylegend.core.language.shared.primitive_collection import PyLegendPrimitiveCollection
 from pylegend.core.language import (
-    TdsRow,
+    LegacyApiTdsRow,
     PyLegendPrimitive,
     PyLegendPrimitiveOrPythonPrimitive,
 )
 
 __all__: PyLegendSequence[str] = [
-    "AggregateSpecification",
+    "LegacyApiAggregateSpecification",
     "agg",
 ]
 
 
-class AggregateSpecification:
-    __map_fn: PyLegendCallable[[TdsRow], PyLegendPrimitiveOrPythonPrimitive]
+class LegacyApiAggregateSpecification:
+    __map_fn: PyLegendCallable[[LegacyApiTdsRow], PyLegendPrimitiveOrPythonPrimitive]
     __aggregate_fn: PyLegendCallable[[PyLegendPrimitiveCollection], PyLegendPrimitive]
     __name: str
 
     def __init__(
             self,
-            map_fn: PyLegendCallable[[TdsRow], PyLegendPrimitiveOrPythonPrimitive],
+            map_fn: PyLegendCallable[[LegacyApiTdsRow], PyLegendPrimitiveOrPythonPrimitive],
             aggregate_fn: PyLegendCallable[[PyLegendPrimitiveCollection], PyLegendPrimitive],
             name: str
     ) -> None:
@@ -47,7 +47,7 @@ class AggregateSpecification:
     def get_name(self) -> str:
         return self.__name
 
-    def get_map_fn(self) -> PyLegendCallable[[TdsRow], PyLegendPrimitiveOrPythonPrimitive]:
+    def get_map_fn(self) -> PyLegendCallable[[LegacyApiTdsRow], PyLegendPrimitiveOrPythonPrimitive]:
         return self.__map_fn
 
     def get_aggregate_fn(self) -> PyLegendCallable[[PyLegendPrimitiveCollection], PyLegendPrimitive]:
@@ -55,8 +55,8 @@ class AggregateSpecification:
 
 
 def agg(
-    map_fn: PyLegendCallable[[TdsRow], PyLegendPrimitiveOrPythonPrimitive],
+    map_fn: PyLegendCallable[[LegacyApiTdsRow], PyLegendPrimitiveOrPythonPrimitive],
     aggregate_fn: PyLegendCallable[[PyLegendPrimitiveCollection], PyLegendPrimitive],
     name: str
-) -> AggregateSpecification:
-    return AggregateSpecification(map_fn=map_fn, aggregate_fn=aggregate_fn, name=name)
+) -> LegacyApiAggregateSpecification:
+    return LegacyApiAggregateSpecification(map_fn=map_fn, aggregate_fn=aggregate_fn, name=name)

pylegend/core/language/legacy_api/legacy_api_tds_row.py

@@ -0,0 +1,32 @@
+# Copyright 2023 Goldman Sachs
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from pylegend._typing import (
+    PyLegendSequence,
+)
+from pylegend.core.language.shared.tds_row import AbstractTdsRow
+from pylegend.core.tds.tds_frame import PyLegendTdsFrame
+
+__all__: PyLegendSequence[str] = [
+    "LegacyApiTdsRow",
+]
+
+
+class LegacyApiTdsRow(AbstractTdsRow):
+    def __init__(self, frame_name: str, frame: PyLegendTdsFrame) -> None:
+        super().__init__(frame_name, frame)
+
+    @staticmethod
+    def from_tds_frame(frame_name: str, frame: PyLegendTdsFrame) -> "LegacyApiTdsRow":
+        return LegacyApiTdsRow(frame_name=frame_name, frame=frame)

pylegend/core/language/legendql_api/__init__.py

@@ -0,0 +1,13 @@
+# Copyright 2025 Goldman Sachs
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.