thoth-dbmanager 0.5.3__py3-none-any.whl → 0.5.9__py3-none-any.whl

thoth_dbmanager/adapters/postgresql.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 """
 PostgreSQL adapter implementation.
 """
@@ -29,6 +43,8 @@ class PostgreSQLAdapter(DbAdapter):
         super().__init__(connection_params)
         self.engine = None
         self.raw_connection = None
+        # Schema support (default 'public')
+        self.schema = connection_params.get('schema', 'public')
     
     def connect(self) -> None:
         """Establish PostgreSQL connection"""
@@ -129,19 +145,23 @@ class PostgreSQLAdapter(DbAdapter):
     
     def get_tables_as_documents(self) -> List[TableDocument]:
         """Get tables as document objects"""
+        # Use pg_catalog to avoid requiring SELECT privileges on tables
+        # relkind: 'r' = ordinary table, 'p' = partitioned table
         query = """
         SELECT 
-            schemaname as schema_name,
-            tablename as table_name,
-            COALESCE(obj_description(c.oid), '') as comment
-        FROM pg_tables pt
-        LEFT JOIN pg_class c ON c.relname = pt.tablename
-        LEFT JOIN pg_namespace n ON n.oid = c.relnamespace AND n.nspname = pt.schemaname
-        WHERE schemaname NOT IN ('information_schema', 'pg_catalog', 'pg_toast')
-        ORDER BY schemaname, tablename
+            n.nspname AS schema_name,
+            c.relname AS table_name,
+            COALESCE(d.description, '') AS comment
+        FROM pg_class c
+        JOIN pg_namespace n ON n.oid = c.relnamespace
+        LEFT JOIN pg_description d ON d.objoid = c.oid AND d.objsubid = 0
+        WHERE c.relkind IN ('r','p')
+          AND n.nspname = :schema
+          AND n.nspname NOT IN ('pg_catalog','information_schema','pg_toast')
+        ORDER BY c.relname
         """
         
-        results = self.execute_query(query)
+        results = self.execute_query(query, {"schema": self.schema})
         documents = []
         
         for row in results:
@@ -179,10 +199,11 @@ class PostgreSQLAdapter(DbAdapter):
         ) pk ON pk.column_name = c.column_name AND pk.table_name = c.table_name AND pk.table_schema = c.table_schema
         WHERE c.table_name = :table_name
         AND c.table_schema NOT IN ('information_schema', 'pg_catalog')
+        AND c.table_schema = :schema
         ORDER BY c.ordinal_position
         """
         
-        results = self.execute_query(query, {"table_name": table_name})
+        results = self.execute_query(query, {"table_name": table_name, "schema": self.schema})
         documents = []
         
         for row in results:
@@ -205,21 +226,26 @@ class PostgreSQLAdapter(DbAdapter):
         """Get foreign keys as document objects"""
         query = """
         SELECT
-            tc.constraint_name,
-            tc.table_schema as schema_name,
-            tc.table_name as source_table,
-            kcu.column_name as source_column,
-            ccu.table_name as target_table,
-            ccu.column_name as target_column
-        FROM information_schema.table_constraints tc
-        JOIN information_schema.key_column_usage kcu ON tc.constraint_name = kcu.constraint_name
-        JOIN information_schema.constraint_column_usage ccu ON ccu.constraint_name = tc.constraint_name
-        WHERE tc.constraint_type = 'FOREIGN KEY'
-        AND tc.table_schema NOT IN ('information_schema', 'pg_catalog')
-        ORDER BY tc.table_schema, tc.table_name, kcu.ordinal_position
+            con.conname AS constraint_name,
+            ns.nspname AS schema_name,
+            rel.relname AS source_table,
+            a.attname AS source_column,
+            frel.relname AS target_table,
+            fa.attname AS target_column
+        FROM pg_constraint con
+        JOIN pg_class rel ON rel.oid = con.conrelid
+        JOIN pg_namespace ns ON ns.oid = rel.relnamespace
+        JOIN pg_class frel ON frel.oid = con.confrelid
+        JOIN unnest(con.conkey) WITH ORDINALITY AS src(attnum, ord) ON true
+        JOIN pg_attribute a ON a.attrelid = con.conrelid AND a.attnum = src.attnum
+        JOIN unnest(con.confkey) WITH ORDINALITY AS dst(attnum, ord) ON dst.ord = src.ord
+        JOIN pg_attribute fa ON fa.attrelid = con.confrelid AND fa.attnum = dst.attnum
+        WHERE con.contype = 'f'
+        AND ns.nspname = :schema
+        ORDER BY ns.nspname, rel.relname, src.ord
         """
         
-        results = self.execute_query(query)
+        results = self.execute_query(query, {"schema": self.schema})
         documents = []
         
         for row in results:

thoth_dbmanager/adapters/sqlite.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 """
 SQLite adapter implementation.
 """
@@ -206,7 +220,6 @@ class SQLiteAdapter(DbAdapter):
     
     def get_schemas_as_documents(self) -> List[SchemaDocument]:
         """Get schemas as document objects"""
-        # SQLite has limited schema support, mainly 'main', 'temp', and attached databases
         query = "PRAGMA database_list"
         
         results = self.execute_query(query)

thoth_dbmanager/adapters/sqlserver.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 """
 SQL Server adapter implementation.
 """
@@ -23,6 +37,7 @@ class SQLServerAdapter(DbAdapter):
         self.database = connection_params.get('database')
         self.user = connection_params.get('user')
         self.password = connection_params.get('password')
+        self.schema = connection_params.get('schema', 'dbo')  # Default to 'dbo' for SQL Server
         self.driver = connection_params.get('driver', 'ODBC Driver 17 for SQL Server')
         
     def connect(self) -> None:
@@ -142,10 +157,11 @@ class SQLServerAdapter(DbAdapter):
     
     def get_tables(self) -> List[str]:
         """Get list of tables in the database."""
-        query = """
+        query = f"""
         SELECT TABLE_NAME as name
         FROM INFORMATION_SCHEMA.TABLES
         WHERE TABLE_TYPE = 'BASE TABLE'
+        AND TABLE_SCHEMA = '{self.schema}'
         ORDER BY TABLE_NAME
         """
         result = self.execute_query(query)
@@ -159,15 +175,17 @@ class SQLServerAdapter(DbAdapter):
             DATA_TYPE as type,
             IS_NULLABLE as nullable,
             COLUMN_DEFAULT as default_value,
-            CASE WHEN COLUMNPROPERTY(OBJECT_ID(TABLE_NAME), COLUMN_NAME, 'IsIdentity') = 1 THEN 1 ELSE 0 END as is_identity,
+            CASE WHEN COLUMNPROPERTY(OBJECT_ID('{self.schema}.' + TABLE_NAME), COLUMN_NAME, 'IsIdentity') = 1 THEN 1 ELSE 0 END as is_identity,
             CASE WHEN EXISTS (
                 SELECT 1 FROM INFORMATION_SCHEMA.KEY_COLUMN_USAGE 
                 WHERE TABLE_NAME = '{table_name}' 
+                AND TABLE_SCHEMA = '{self.schema}'
                 AND COLUMN_NAME = c.COLUMN_NAME 
                 AND CONSTRAINT_NAME LIKE 'PK_%'
             ) THEN 1 ELSE 0 END as is_primary_key
         FROM INFORMATION_SCHEMA.COLUMNS c
         WHERE TABLE_NAME = '{table_name}'
+        AND TABLE_SCHEMA = '{self.schema}'
         ORDER BY ORDINAL_POSITION
         """
         
@@ -255,6 +273,7 @@ class SQLServerAdapter(DbAdapter):
         SELECT COUNT(*) as count
         FROM INFORMATION_SCHEMA.TABLES
         WHERE TABLE_NAME = '{table_name}'
+        AND TABLE_SCHEMA = '{self.schema}'
         AND TABLE_TYPE = 'BASE TABLE'
         """
         result = self.execute_query(query)
@@ -276,13 +295,14 @@ class SQLServerAdapter(DbAdapter):
         if not self.engine:
             raise RuntimeError("Not connected to database")
 
-        query = """
+        query = f"""
         SELECT
             TABLE_NAME as name,
             TABLE_SCHEMA as schema_name,
             '' as comment
         FROM INFORMATION_SCHEMA.TABLES
         WHERE TABLE_TYPE = 'BASE TABLE'
+        AND TABLE_SCHEMA = '{self.schema}'
         ORDER BY TABLE_NAME
         """
 
@@ -323,9 +343,10 @@ class SQLServerAdapter(DbAdapter):
         """Get example data (most frequent values) for each column in a table."""
         inspector = inspect(self.engine)
         try:
-            columns = inspector.get_columns(table_name)
+            # For SQL Server, we need to specify the schema when inspecting columns
+            columns = inspector.get_columns(table_name, schema=self.schema)
         except SQLAlchemyError as e:
-            logger.error(f"Error inspecting columns for table {table_name}: {e}")
+            logger.error(f"Error inspecting columns for table {table_name} in schema {self.schema}: {e}")
             raise e
 
         if not columns:
@@ -374,16 +395,18 @@ class SQLServerAdapter(DbAdapter):
                 CASE WHEN EXISTS (
                     SELECT 1 FROM INFORMATION_SCHEMA.KEY_COLUMN_USAGE
                     WHERE TABLE_NAME = c.TABLE_NAME
+                    AND TABLE_SCHEMA = c.TABLE_SCHEMA
                     AND COLUMN_NAME = c.COLUMN_NAME
                     AND CONSTRAINT_NAME LIKE 'PK_%'
                 ) THEN 1 ELSE 0 END as is_primary_key
             FROM INFORMATION_SCHEMA.COLUMNS c
             WHERE c.TABLE_NAME = '{table_name}'
+            AND c.TABLE_SCHEMA = '{self.schema}'
             ORDER BY c.ORDINAL_POSITION
             """
         else:
             # Get all columns
-            query = """
+            query = f"""
             SELECT
                 c.TABLE_NAME as table_name,
                 c.COLUMN_NAME as column_name,
@@ -394,10 +417,12 @@ class SQLServerAdapter(DbAdapter):
                 CASE WHEN EXISTS (
                     SELECT 1 FROM INFORMATION_SCHEMA.KEY_COLUMN_USAGE
                     WHERE TABLE_NAME = c.TABLE_NAME
+                    AND TABLE_SCHEMA = c.TABLE_SCHEMA
                     AND COLUMN_NAME = c.COLUMN_NAME
                     AND CONSTRAINT_NAME LIKE 'PK_%'
                 ) THEN 1 ELSE 0 END as is_primary_key
             FROM INFORMATION_SCHEMA.COLUMNS c
+            WHERE c.TABLE_SCHEMA = '{self.schema}'
             ORDER BY c.TABLE_NAME, c.ORDINAL_POSITION
             """
 
@@ -445,7 +470,7 @@ class SQLServerAdapter(DbAdapter):
             raise RuntimeError("Not connected to database")
 
         if table_name:
-            where_clause = f"WHERE t.name = '{table_name}'"
+            where_clause = f"AND t.name = '{table_name}'"
         else:
             where_clause = ""
 
@@ -461,6 +486,8 @@ class SQLServerAdapter(DbAdapter):
         JOIN sys.columns c ON fkc.parent_object_id = c.object_id AND fkc.parent_column_id = c.column_id
         JOIN sys.columns rc ON fkc.referenced_object_id = rc.object_id AND fkc.referenced_column_id = rc.column_id
         JOIN sys.tables t ON fk.parent_object_id = t.object_id
+        JOIN sys.schemas s ON t.schema_id = s.schema_id
+        WHERE s.name = '{self.schema}'
         {where_clause}
         ORDER BY fk.name
         """
@@ -503,7 +530,7 @@ class SQLServerAdapter(DbAdapter):
             raise RuntimeError("Not connected to database")
 
         if table_name:
-            where_clause = f"WHERE t.name = '{table_name}'"
+            where_clause = f"AND t.name = '{table_name}'"
         else:
             where_clause = ""
 
@@ -518,8 +545,10 @@ class SQLServerAdapter(DbAdapter):
         JOIN sys.index_columns ic ON i.object_id = ic.object_id AND i.index_id = ic.index_id
         JOIN sys.columns c ON ic.object_id = c.object_id AND ic.column_id = c.column_id
         JOIN sys.tables t ON i.object_id = t.object_id
+        JOIN sys.schemas s ON t.schema_id = s.schema_id
+        WHERE s.name = '{self.schema}'
         {where_clause}
-        WHERE i.name IS NOT NULL
+        AND i.name IS NOT NULL
         ORDER BY i.name, ic.key_ordinal
         """
 

thoth_dbmanager/core/__init__.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 """
 Core components for Thoth SQL Database Manager.
 """

thoth_dbmanager/core/factory.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 """
 Factory for creating database manager instances with plugin support.
 """
@@ -6,6 +20,9 @@ from typing import Any, Dict, List, Optional
 from .registry import DbPluginRegistry
 from .interfaces import DbPlugin
 
+# Import plugins to ensure they are registered
+from .. import plugins  # This imports all plugins and registers them
+
 logger = logging.getLogger(__name__)
 
 

thoth_dbmanager/core/interfaces.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 """
 Abstract interfaces for database plugins and adapters.
 """

thoth_dbmanager/core/registry.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 """
 Plugin registry system for database plugins.
 """

thoth_dbmanager/documents.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 """
 Document models for Thoth SQL Database Manager.
 Provides type-safe document structures similar to thoth_vdb architecture.

thoth_dbmanager/dynamic_imports.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 """
 Dynamic import system for database-specific functionality.
 This module provides lazy loading of database managers and adapters.

thoth_dbmanager/helpers/__init__.py

@@ -0,0 +1,13 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.

thoth_dbmanager/helpers/multi_db_generator.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 import logging
 import random
 import re

thoth_dbmanager/helpers/preprocess_values.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 import logging
 import pickle
 from pathlib import Path

thoth_dbmanager/helpers/schema.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 import logging
 from dataclasses import dataclass, field
 from typing import Any, Dict, List, Optional, Tuple

thoth_dbmanager/helpers/search.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 import logging
 import pickle
 from pathlib import Path

thoth_dbmanager/lsh/__init__.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 """
 LSH (Locality Sensitive Hashing) module for database-independent LSH management.
 """

thoth_dbmanager/lsh/core.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 """
 Core LSH functionality extracted from helpers.
 """

thoth_dbmanager/lsh/factory.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 """
 Factory for creating LSH indices from database managers.
 """

thoth_dbmanager/lsh/manager.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 """
 LSH Manager for database-independent LSH operations.
 """

thoth_dbmanager/lsh/storage.py

@@ -1,3 +1,17 @@
+# Copyright 2025 Marco Pancotti
+#
+# 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.
+
 """
 Storage strategies for LSH data persistence.
 """