[DOP-25645] insert license

Author: TiGrib

data_rentgen/consumer/extractors/batch_extraction_result.py

@@ -68,7 +68,7 @@ def __init__(self):
         self._outputs: dict[tuple, OutputDTO] = {}
         self._column_lineage: dict[tuple, ColumnLineageDTO] = {}
         self._schemas: dict[tuple, SchemaDTO] = {}
-        self._sql_queries: dict[str, SQLQueryDTO] = {}
+        self._sql_queries: dict[tuple, SQLQueryDTO] = {}
         self._users: dict[tuple, UserDTO] = {}
 
     def __repr__(self):
@@ -173,7 +173,7 @@ def get_location(self, location_key: tuple) -> LocationDTO:
     def get_schema(self, schema_key: tuple) -> SchemaDTO:
         return self._schemas[schema_key]
 
-    def get_sql_query(self, sql_query_key: str) -> SQLQueryDTO:
+    def get_sql_query(self, sql_query_key: tuple) -> SQLQueryDTO:
         return self._sql_queries[sql_query_key]
 
     def get_user(self, user_key: tuple) -> UserDTO:

data_rentgen/consumer/openlineage/job_facets/sql.py

@@ -1,3 +1,5 @@
+# SPDX-FileCopyrightText: 2024-2025 MTS PJSC
+# SPDX-License-Identifier: Apache-2.0
 from data_rentgen.consumer.openlineage.job_facets.base import OpenLineageJobFacet
 
 

data_rentgen/db/models/sql_query.py

@@ -1,3 +1,5 @@
+# SPDX-FileCopyrightText: 2024-2025 MTS PJSC
+# SPDX-License-Identifier: Apache-2.0
 from uuid import UUID
 
 from sqlalchemy import UUID as SQL_UUID

data_rentgen/db/repositories/sql_query.py

@@ -1,3 +1,5 @@
+# SPDX-FileCopyrightText: 2024-2025 MTS PJSC
+# SPDX-License-Identifier: Apache-2.0
 from uuid import UUID
 
 from sqlalchemy import select

data_rentgen/dto/sql_query.py

@@ -1,3 +1,5 @@
+# SPDX-FileCopyrightText: 2024-2025 MTS PJSC
+# SPDX-License-Identifier: Apache-2.0
 from __future__ import annotations
 
 from dataclasses import dataclass, field
@@ -8,6 +10,10 @@ class SQLQueryDTO:
     query: str
     id: int | None = field(default=None, compare=False)
 
+    @property
+    def unique_key(self) -> tuple:
+        return (self.query,)
+
     def merge(self, new: SQLQueryDTO) -> SQLQueryDTO:
         if new.id is None:
             return self