pylegend 0.3.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

pylegend/core/tds/{legend_api/frames/legend_api_tds_frame.py → legacy_api/frames/legacy_api_tds_frame.py}

@@ -24,45 +24,45 @@ from pylegend._typing import (
     PyLegendUnion,
 )
 from pylegend.core.language import (
-    TdsRow,
+    LegacyApiTdsRow,
     PyLegendBoolean,
     PyLegendPrimitiveOrPythonPrimitive,
-    AggregateSpecification,
+    LegacyApiAggregateSpecification,
 )
 
 __all__: PyLegendSequence[str] = [
-    "LegendApiTdsFrame"
+    "LegacyApiTdsFrame"
 ]
 
 
-class LegendApiTdsFrame(PyLegendTdsFrame):
+class LegacyApiTdsFrame(PyLegendTdsFrame):
 
     @abstractmethod
-    def head(self, count: int = 5) -> "LegendApiTdsFrame":
+    def head(self, count: int = 5) -> "LegacyApiTdsFrame":
         pass  # pragma: no cover
 
     @abstractmethod
-    def take(self, count: int = 5) -> "LegendApiTdsFrame":
+    def take(self, count: int = 5) -> "LegacyApiTdsFrame":
         pass  # pragma: no cover
 
     @abstractmethod
-    def limit(self, count: int = 5) -> "LegendApiTdsFrame":
+    def limit(self, count: int = 5) -> "LegacyApiTdsFrame":
         pass  # pragma: no cover
 
     @abstractmethod
-    def drop(self, count: int = 5) -> "LegendApiTdsFrame":
+    def drop(self, count: int = 5) -> "LegacyApiTdsFrame":
         pass  # pragma: no cover
 
     @abstractmethod
-    def slice(self, start_row: int, end_row_exclusive: int) -> "LegendApiTdsFrame":
+    def slice(self, start_row: int, end_row_exclusive: int) -> "LegacyApiTdsFrame":
         pass  # pragma: no cover
 
     @abstractmethod
-    def distinct(self) -> "LegendApiTdsFrame":
+    def distinct(self) -> "LegacyApiTdsFrame":
         pass  # pragma: no cover
 
     @abstractmethod
-    def restrict(self, column_name_list: PyLegendList[str]) -> "LegendApiTdsFrame":
+    def restrict(self, column_name_list: PyLegendList[str]) -> "LegacyApiTdsFrame":
         pass  # pragma: no cover
 
     @abstractmethod
@@ -70,11 +70,11 @@ class LegendApiTdsFrame(PyLegendTdsFrame):
             self,
             column_name_list: PyLegendList[str],
             direction_list: PyLegendOptional[PyLegendList[str]] = None
-    ) -> "LegendApiTdsFrame":
+    ) -> "LegacyApiTdsFrame":
         pass  # pragma: no cover
 
     @abstractmethod
-    def concatenate(self, other: "LegendApiTdsFrame") -> "LegendApiTdsFrame":
+    def concatenate(self, other: "LegacyApiTdsFrame") -> "LegacyApiTdsFrame":
         pass  # pragma: no cover
 
     @abstractmethod
@@ -82,55 +82,55 @@ class LegendApiTdsFrame(PyLegendTdsFrame):
             self,
             column_names: PyLegendList[str],
             renamed_column_names: PyLegendList[str]
-    ) -> "LegendApiTdsFrame":
+    ) -> "LegacyApiTdsFrame":
         pass  # pragma: no cover
 
     @abstractmethod
     def filter(
             self,
-            filter_function: PyLegendCallable[[TdsRow], PyLegendUnion[bool, PyLegendBoolean]]
-    ) -> "LegendApiTdsFrame":
+            filter_function: PyLegendCallable[[LegacyApiTdsRow], PyLegendUnion[bool, PyLegendBoolean]]
+    ) -> "LegacyApiTdsFrame":
         pass  # pragma: no cover
 
     @abstractmethod
     def extend(
             self,
-            functions_list: PyLegendList[PyLegendCallable[[TdsRow], PyLegendPrimitiveOrPythonPrimitive]],
+            functions_list: PyLegendList[PyLegendCallable[[LegacyApiTdsRow], PyLegendPrimitiveOrPythonPrimitive]],
             column_names_list: PyLegendList[str]
-    ) -> "LegendApiTdsFrame":
+    ) -> "LegacyApiTdsFrame":
         pass  # pragma: no cover
 
     @abstractmethod
     def join(
             self,
-            other: "LegendApiTdsFrame",
-            join_condition: PyLegendCallable[[TdsRow, TdsRow], PyLegendUnion[bool, PyLegendBoolean]],
+            other: "LegacyApiTdsFrame",
+            join_condition: PyLegendCallable[[LegacyApiTdsRow, LegacyApiTdsRow], PyLegendUnion[bool, PyLegendBoolean]],
             join_type: str = 'LEFT_OUTER'
-    ) -> "LegendApiTdsFrame":
+    ) -> "LegacyApiTdsFrame":
         pass  # pragma: no cover
 
     @abstractmethod
     def join_by_columns(
             self,
-            other: "LegendApiTdsFrame",
+            other: "LegacyApiTdsFrame",
             self_columns: PyLegendList[str],
             other_columns: PyLegendList[str],
             join_type: str = 'LEFT_OUTER'
-    ) -> "LegendApiTdsFrame":
+    ) -> "LegacyApiTdsFrame":
         pass  # pragma: no cover
 
     def join_by_function(
             self,
-            other: "LegendApiTdsFrame",
-            join_condition: PyLegendCallable[[TdsRow, TdsRow], PyLegendUnion[bool, PyLegendBoolean]],
+            other: "LegacyApiTdsFrame",
+            join_condition: PyLegendCallable[[LegacyApiTdsRow, LegacyApiTdsRow], PyLegendUnion[bool, PyLegendBoolean]],
             join_type: str = 'LEFT_OUTER'
-    ) -> "LegendApiTdsFrame":
+    ) -> "LegacyApiTdsFrame":
         return self.join(other, join_condition, join_type)
 
     @abstractmethod
     def group_by(
             self,
             grouping_columns: PyLegendList[str],
-            aggregations: PyLegendList[AggregateSpecification],
-    ) -> "LegendApiTdsFrame":
+            aggregations: PyLegendList[LegacyApiAggregateSpecification],
+    ) -> "LegacyApiTdsFrame":
         pass  # pragma: no cover

pylegend/core/tds/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.

pylegend/core/tds/legendql_api/frames/__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/tds/legendql_api/frames/functions/__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/tds/legendql_api/frames/functions/legendql_api_asofjoin_function.py

@@ -0,0 +1,156 @@
+# 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.
+
+from pylegend._typing import (
+    PyLegendList,
+    PyLegendSequence,
+    PyLegendUnion,
+    PyLegendCallable,
+    PyLegendOptional,
+)
+from pylegend.core.language import (
+    PyLegendBoolean,
+    PyLegendPrimitive,
+    convert_literal_to_literal_expression,
+)
+from pylegend.core.language.legendql_api.legendql_api_tds_row import LegendQLApiTdsRow
+from pylegend.core.language.shared.helpers import generate_pure_lambda
+from pylegend.core.sql.metamodel import (
+    QuerySpecification,
+)
+from pylegend.core.tds.legendql_api.frames.legendql_api_applied_function_tds_frame import LegendQLApiAppliedFunction
+from pylegend.core.tds.legendql_api.frames.legendql_api_base_tds_frame import LegendQLApiBaseTdsFrame
+from pylegend.core.tds.legendql_api.frames.legendql_api_tds_frame import LegendQLApiTdsFrame
+from pylegend.core.tds.tds_column import TdsColumn
+from pylegend.core.tds.tds_frame import FrameToPureConfig
+from pylegend.core.tds.tds_frame import FrameToSqlConfig
+
+__all__: PyLegendSequence[str] = [
+    "LegendQLApiAsOfJoinFunction"
+]
+
+
+class LegendQLApiAsOfJoinFunction(LegendQLApiAppliedFunction):
+    __base_frame: LegendQLApiBaseTdsFrame
+    __other_frame: LegendQLApiBaseTdsFrame
+    __match_function: PyLegendCallable[[LegendQLApiTdsRow, LegendQLApiTdsRow], PyLegendUnion[bool, PyLegendBoolean]]
+    __join_condition: PyLegendOptional[
+        PyLegendCallable[[LegendQLApiTdsRow, LegendQLApiTdsRow], PyLegendUnion[bool, PyLegendBoolean]]
+    ]
+
+    @classmethod
+    def name(cls) -> str:
+        return "asOfJoin"
+
+    def __init__(
+            self,
+            base_frame: LegendQLApiBaseTdsFrame,
+            other_frame: LegendQLApiTdsFrame,
+            match_function: PyLegendCallable[[LegendQLApiTdsRow, LegendQLApiTdsRow], PyLegendUnion[bool, PyLegendBoolean]],
+            join_condition: PyLegendOptional[
+                PyLegendCallable[[LegendQLApiTdsRow, LegendQLApiTdsRow], PyLegendUnion[bool, PyLegendBoolean]]
+            ] = None
+    ) -> None:
+        self.__base_frame = base_frame
+        if not isinstance(other_frame, LegendQLApiBaseTdsFrame):
+            raise ValueError("Expected LegendQLApiBaseTdsFrame")  # pragma: no cover
+        self.__other_frame = other_frame
+        self.__match_function = match_function
+        self.__join_condition = join_condition
+
+    def to_sql(self, config: FrameToSqlConfig) -> QuerySpecification:
+        raise RuntimeError("AsOfJoin SQL translation not supported yet")
+
+    def to_pure(self, config: FrameToPureConfig) -> str:
+        left_row = LegendQLApiTdsRow.from_tds_frame("l", self.__base_frame)
+        right_row = LegendQLApiTdsRow.from_tds_frame("r", self.__other_frame)
+        match_expr = self.__match_function(left_row, right_row)
+        match_string = (match_expr.to_pure_expression(config.push_indent(2))
+                        if isinstance(match_expr, PyLegendPrimitive) else
+                        convert_literal_to_literal_expression(match_expr).to_pure_expression(config.push_indent(2)))
+        if self.__join_condition is not None:
+            join_expr = self.__join_condition(left_row, right_row)
+            join_string = (join_expr.to_pure_expression(config.push_indent(2))
+                           if isinstance(join_expr, PyLegendPrimitive) else
+                           convert_literal_to_literal_expression(join_expr).to_pure_expression(config.push_indent(2)))
+            join_arg = (
+                f",{config.separator(2, True)}"
+                f"{generate_pure_lambda('l, r', join_string)}{config.separator(1)}"
+            )
+        else:
+            join_arg = f"{config.separator(1)}"
+
+        return (f"{self.__base_frame.to_pure(config)}{config.separator(1)}" +
+                f"->asOfJoin({config.separator(2)}"
+                f"{self.__other_frame.to_pure(config.push_indent(2))},{config.separator(2, True)}"
+                f"{generate_pure_lambda('l, r', match_string)}"
+                f"{join_arg}"
+                f")")
+
+    def base_frame(self) -> LegendQLApiBaseTdsFrame:
+        return self.__base_frame
+
+    def tds_frame_parameters(self) -> PyLegendList["LegendQLApiBaseTdsFrame"]:
+        return [self.__other_frame]
+
+    def calculate_columns(self) -> PyLegendSequence["TdsColumn"]:
+        return (
+            [c.copy() for c in self.__base_frame.columns()] +
+            [c.copy() for c in self.__other_frame.columns()]
+        )
+
+    def validate(self) -> bool:
+        copy = self.__match_function  # For MyPy
+        if not isinstance(copy, type(lambda x: 0)) or (copy.__code__.co_argcount != 2):
+            raise TypeError("AsOfJoin match function should be a lambda which takes two arguments (TDSRow, TDSRow)")
+
+        left_row = LegendQLApiTdsRow.from_tds_frame("left", self.__base_frame)
+        right_row = LegendQLApiTdsRow.from_tds_frame("right", self.__other_frame)
+
+        try:
+            result = self.__match_function(left_row, right_row)
+        except Exception as e:
+            raise RuntimeError(
+                "AsOfJoin match function incompatible. Error occurred while evaluating. Message: " + str(e)
+            ) from e
+
+        if not isinstance(result, (bool, PyLegendBoolean)):
+            raise RuntimeError("AsOfJoin match function incompatible. Returns non boolean - " + str(type(result)))
+
+        if self.__join_condition is not None:
+            copy = self.__join_condition  # For MyPy
+            if not isinstance(copy, type(lambda x: 0)) or (copy.__code__.co_argcount != 2):
+                raise TypeError("AsOfJoin join function should be a lambda which takes two arguments (TDSRow, TDSRow)")
+
+            try:
+                result = self.__join_condition(left_row, right_row)
+            except Exception as e:
+                raise RuntimeError(
+                    "AsOfJoin join function incompatible. Error occurred while evaluating. Message: " + str(e)
+                ) from e
+
+            if not isinstance(result, (bool, PyLegendBoolean)):
+                raise RuntimeError("AsOfJoin join function incompatible. Returns non boolean - " + str(type(result)))
+
+        left_cols = [c.get_name() for c in self.__base_frame.columns()]
+        right_cols = [c.get_name() for c in self.__other_frame.columns()]
+
+        final_cols = left_cols + right_cols
+        if len(final_cols) != len(set(final_cols)):
+            raise ValueError(
+                "Found duplicate columns in joined frames. Use rename function to ensure there are no duplicate columns "
+                f"in joined frames. Columns - Left Frame: {left_cols}, Right Frame: {right_cols}"
+            )
+
+        return True

pylegend/core/tds/legendql_api/frames/functions/legendql_api_concatenate_function.py

@@ -0,0 +1,139 @@
+# 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.
+
+from pylegend._typing import (
+    PyLegendList,
+    PyLegendSequence
+)
+from pylegend.core.tds.legendql_api.frames.legendql_api_applied_function_tds_frame import LegendQLApiAppliedFunction
+from pylegend.core.sql.metamodel import (
+    QuerySpecification,
+    Union,
+    AliasedRelation,
+    SingleColumn,
+    Select,
+    QualifiedNameReference,
+    QualifiedName,
+    TableSubquery,
+    Query,
+)
+from pylegend.core.tds.sql_query_helpers import create_sub_query
+from pylegend.core.tds.tds_column import TdsColumn
+from pylegend.core.tds.tds_frame import FrameToSqlConfig
+from pylegend.core.tds.tds_frame import FrameToPureConfig
+from pylegend.core.tds.legendql_api.frames.legendql_api_tds_frame import LegendQLApiTdsFrame
+from pylegend.core.tds.legendql_api.frames.legendql_api_base_tds_frame import LegendQLApiBaseTdsFrame
+
+
+__all__: PyLegendSequence[str] = [
+    "LegendQLApiConcatenateFunction"
+]
+
+
+class LegendQLApiConcatenateFunction(LegendQLApiAppliedFunction):
+    __base_frame: LegendQLApiBaseTdsFrame
+    __other_frame: LegendQLApiBaseTdsFrame
+
+    @classmethod
+    def name(cls) -> str:
+        return "concatenate"
+
+    def __init__(self, base_frame: LegendQLApiBaseTdsFrame, other: LegendQLApiTdsFrame) -> None:
+        self.__base_frame = base_frame
+        if not isinstance(other, LegendQLApiBaseTdsFrame):
+            raise ValueError("Expected LegendQLApiBaseTdsFrame")  # pragma: no cover
+        self.__other_frame = other
+
+    def to_sql(self, config: FrameToSqlConfig) -> QuerySpecification:
+        base_query = self.__base_frame.to_sql_query_object(config)
+        other_query = self.__other_frame.to_sql_query_object(config)
+
+        db_extension = config.sql_to_string_generator().get_db_extension()
+        root_alias = db_extension.quote_identifier("root")
+        columns = [db_extension.quote_identifier(c.get_name()) for c in self.__base_frame.columns()]
+
+        return QuerySpecification(
+            select=Select(
+                selectItems=[
+                    SingleColumn(
+                        alias=x,
+                        expression=QualifiedNameReference(name=QualifiedName(parts=[root_alias, x]))
+                    )
+                    for x in columns
+                ],
+                distinct=False
+            ),
+            from_=[
+                AliasedRelation(
+                    relation=TableSubquery(
+                        query=Query(
+                            queryBody=Union(
+                                left=create_sub_query(base_query, config, "left"),
+                                right=create_sub_query(other_query, config, "right"),
+                                distinct=False
+                            ),
+                            limit=None, offset=None, orderBy=[]
+                        )
+                    ),
+                    alias=root_alias,
+                    columnNames=columns
+                )
+            ],
+            where=None,
+            groupBy=[],
+            having=None,
+            orderBy=[],
+            limit=None,
+            offset=None
+        )
+
+    def to_pure(self, config: FrameToPureConfig) -> str:
+        return (f"{self.__base_frame.to_pure(config)}{config.separator(1)}"
+                f"->concatenate({config.separator(2)}"
+                f"{self.__other_frame.to_pure(config.push_indent(2))}"
+                f"{config.separator(1)})")
+
+    def base_frame(self) -> LegendQLApiBaseTdsFrame:
+        return self.__base_frame
+
+    def tds_frame_parameters(self) -> PyLegendList["LegendQLApiBaseTdsFrame"]:
+        return [self.__other_frame]
+
+    def calculate_columns(self) -> PyLegendSequence["TdsColumn"]:
+        return [c.copy() for c in self.__base_frame.columns()]
+
+    def validate(self) -> bool:
+        base_frame_cols = self.__base_frame.columns()
+        other_frame_cols = self.__other_frame.columns()
+
+        if len(base_frame_cols) != len(other_frame_cols):
+            cols1 = "[" + ", ".join([str(c) for c in base_frame_cols]) + "]"
+            cols2 = "[" + ", ".join([str(c) for c in other_frame_cols]) + "]"
+            raise ValueError(
+                "Cannot concatenate two Tds Frames with different column counts. \n"
+                f"Frame 1 cols - (Count: {len(base_frame_cols)}) - {cols1} \n"
+                f"Frame 2 cols - (Count: {len(other_frame_cols)}) - {cols2} \n"
+            )
+
+        for i in range(0, len(base_frame_cols)):
+            base_col = base_frame_cols[i]
+            other_col = other_frame_cols[i]
+
+            if (base_col.get_name() != other_col.get_name()) or (base_col.get_type() != other_col.get_type()):
+                raise ValueError(
+                    f"Column name/type mismatch when concatenating Tds Frames at index {i}. "
+                    f"Frame 1 column - {base_col}, Frame 2 column - {other_col}"
+                )
+
+        return True

pylegend/core/tds/legendql_api/frames/functions/legendql_api_distinct_function.py

@@ -0,0 +1,69 @@
+# 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.
+
+from pylegend._typing import (
+    PyLegendList,
+    PyLegendSequence
+)
+from pylegend.core.tds.legendql_api.frames.legendql_api_applied_function_tds_frame import LegendQLApiAppliedFunction
+from pylegend.core.tds.sql_query_helpers import copy_query, create_sub_query
+from pylegend.core.sql.metamodel import (
+    QuerySpecification,
+)
+from pylegend.core.tds.tds_column import TdsColumn
+from pylegend.core.tds.tds_frame import FrameToSqlConfig
+from pylegend.core.tds.tds_frame import FrameToPureConfig
+from pylegend.core.tds.legendql_api.frames.legendql_api_base_tds_frame import LegendQLApiBaseTdsFrame
+
+
+__all__: PyLegendSequence[str] = [
+    "LegendQLApiDistinctFunction"
+]
+
+
+class LegendQLApiDistinctFunction(LegendQLApiAppliedFunction):
+    __base_frame: LegendQLApiBaseTdsFrame
+
+    @classmethod
+    def name(cls) -> str:
+        return "distinct"
+
+    def __init__(self, base_frame: LegendQLApiBaseTdsFrame) -> None:
+        self.__base_frame = base_frame
+
+    def to_sql(self, config: FrameToSqlConfig) -> QuerySpecification:
+        base_query = self.__base_frame.to_sql_query_object(config)
+        should_create_sub_query = (base_query.offset is not None) or (base_query.limit is not None)
+        new_query = (
+            create_sub_query(base_query, config, "root") if should_create_sub_query else
+            copy_query(base_query)
+        )
+        new_query.select.distinct = True
+        return new_query
+
+    def to_pure(self, config: FrameToPureConfig) -> str:
+        return (f"{self.__base_frame.to_pure(config)}{config.separator(1)}"
+                f"->distinct()")
+
+    def base_frame(self) -> LegendQLApiBaseTdsFrame:
+        return self.__base_frame
+
+    def tds_frame_parameters(self) -> PyLegendList[LegendQLApiBaseTdsFrame]:
+        return []
+
+    def calculate_columns(self) -> PyLegendSequence[TdsColumn]:
+        return [c.copy() for c in self.__base_frame.columns()]
+
+    def validate(self) -> bool:
+        return True

pylegend/core/tds/legendql_api/frames/functions/legendql_api_drop_function.py

@@ -0,0 +1,74 @@
+# 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.
+
+from pylegend._typing import (
+    PyLegendList,
+    PyLegendSequence
+)
+from pylegend.core.tds.legendql_api.frames.legendql_api_applied_function_tds_frame import LegendQLApiAppliedFunction
+from pylegend.core.tds.sql_query_helpers import copy_query, create_sub_query
+from pylegend.core.sql.metamodel import (
+    QuerySpecification,
+    LongLiteral,
+)
+from pylegend.core.tds.tds_column import TdsColumn
+from pylegend.core.tds.tds_frame import FrameToSqlConfig
+from pylegend.core.tds.tds_frame import FrameToPureConfig
+from pylegend.core.tds.legendql_api.frames.legendql_api_base_tds_frame import LegendQLApiBaseTdsFrame
+
+
+__all__: PyLegendSequence[str] = [
+    "LegendQLApiDropFunction"
+]
+
+
+class LegendQLApiDropFunction(LegendQLApiAppliedFunction):
+    __base_frame: LegendQLApiBaseTdsFrame
+    __row_count: int
+
+    @classmethod
+    def name(cls) -> str:
+        return "drop"
+
+    def __init__(self, base_frame: LegendQLApiBaseTdsFrame, row_count: int) -> None:
+        self.__base_frame = base_frame
+        self.__row_count = row_count
+
+    def to_sql(self, config: FrameToSqlConfig) -> QuerySpecification:
+        base_query = self.__base_frame.to_sql_query_object(config)
+        should_create_sub_query = (base_query.offset is not None) or (base_query.limit is not None)
+        new_query = (
+            create_sub_query(base_query, config, "root") if should_create_sub_query else
+            copy_query(base_query)
+        )
+        new_query.offset = LongLiteral(value=self.__row_count)
+        return new_query
+
+    def to_pure(self, config: FrameToPureConfig) -> str:
+        return (f"{self.__base_frame.to_pure(config)}{config.separator(1)}"
+                f"->drop({self.__row_count})")
+
+    def base_frame(self) -> LegendQLApiBaseTdsFrame:
+        return self.__base_frame
+
+    def tds_frame_parameters(self) -> PyLegendList["LegendQLApiBaseTdsFrame"]:
+        return []
+
+    def calculate_columns(self) -> PyLegendSequence["TdsColumn"]:
+        return [c.copy() for c in self.__base_frame.columns()]
+
+    def validate(self) -> bool:
+        if self.__row_count < 0:
+            raise ValueError("Row count argument of drop function cannot be negative")
+        return True