thoth-dbmanager 0.4.0__py3-none-any.whl → 0.4.2__py3-none-any.whl

thoth_dbmanager/plugins/supabase.py

@@ -0,0 +1,224 @@
+"""
+Supabase plugin implementation.
+"""
+import logging
+from typing import Any, Dict, List
+from pathlib import Path
+from urllib.parse import urlparse
+
+from ..core.interfaces import DbPlugin, DbAdapter
+from ..core.registry import register_plugin
+from ..adapters.supabase import SupabaseAdapter
+
+logger = logging.getLogger(__name__)
+
+
+@register_plugin("supabase")
+class SupabasePlugin(DbPlugin):
+    """
+    Supabase database plugin implementation.
+    Extends PostgreSQL functionality with Supabase-specific features.
+    """
+    
+    plugin_name = "Supabase Plugin"
+    plugin_version = "1.0.0"
+    supported_db_types = ["supabase", "supabase-postgresql"]
+    required_dependencies = ["psycopg2-binary", "SQLAlchemy", "supabase", "postgrest-py"]
+    
+    def __init__(self, db_root_path: str, db_mode: str = "dev", **kwargs):
+        super().__init__(db_root_path, db_mode, **kwargs)
+        self.db_id = None
+        self.db_directory_path = None
+        self.project_url = None
+        self.api_key = None
+        
+        # LSH manager integration (for backward compatibility)
+        self._lsh_manager = None
+    
+    def create_adapter(self, **kwargs) -> DbAdapter:
+        """Create and return a Supabase adapter instance"""
+        return SupabaseAdapter(kwargs)
+    
+    def validate_connection_params(self, **kwargs) -> bool:
+        """Validate connection parameters for Supabase"""
+        # Support both direct database connection and REST API connection
+        use_rest_api = kwargs.get('use_rest_api', False)
+        
+        if use_rest_api:
+            return self._validate_rest_params(**kwargs)
+        else:
+            return self._validate_direct_params(**kwargs)
+    
+    def _validate_rest_params(self, **kwargs) -> bool:
+        """Validate REST API connection parameters"""
+        required_params = ['project_url', 'api_key']
+        
+        for param in required_params:
+            if param not in kwargs:
+                logger.error(f"Missing required parameter for REST API: {param}")
+                return False
+        
+        # Validate project URL format
+        project_url = kwargs.get('project_url')
+        try:
+            parsed = urlparse(project_url)
+            if not parsed.netloc.endswith('.supabase.co'):
+                logger.error("Invalid Supabase project URL format")
+                return False
+        except Exception:
+            logger.error("Invalid project URL format")
+            return False
+        
+        # Validate API key format
+        api_key = kwargs.get('api_key')
+        if not api_key or not isinstance(api_key, str):
+            logger.error("API key must be a non-empty string")
+            return False
+        
+        return True
+    
+    def _validate_direct_params(self, **kwargs) -> bool:
+        """Validate direct database connection parameters"""
+        required_params = ['host', 'port', 'database', 'user', 'password']
+        
+        for param in required_params:
+            if param not in kwargs:
+                logger.error(f"Missing required parameter: {param}")
+                return False
+        
+        # Validate types
+        try:
+            port = int(kwargs['port'])
+            if port <= 0 or port > 65535:
+                logger.error(f"Invalid port number: {port}")
+                return False
+        except (ValueError, TypeError):
+            logger.error(f"Port must be a valid integer: {kwargs.get('port')}")
+            return False
+        
+        # Validate required string parameters are not empty
+        string_params = ['host', 'database', 'user', 'password']
+        for param in string_params:
+            if not kwargs.get(param) or not isinstance(kwargs[param], str):
+                logger.error(f"Parameter {param} must be a non-empty string")
+                return False
+        
+        # Validate SSL parameters if provided
+        ssl_params = ['sslcert', 'sslkey', 'sslrootcert']
+        for param in ssl_params:
+            if param in kwargs and kwargs[param]:
+                if not isinstance(kwargs[param], str):
+                    logger.error(f"SSL parameter {param} must be a string")
+                    return False
+        
+        return True
+    
+    def initialize(self, **kwargs) -> None:
+        """Initialize the Supabase plugin"""
+        super().initialize(**kwargs)
+        
+        # Store connection parameters
+        self.project_url = kwargs.get('project_url')
+        self.api_key = kwargs.get('api_key')
+        
+        # Set up database directory path (for LSH and other features)
+        if 'database' in kwargs:
+            self.db_id = kwargs['database']
+        elif 'project_url' in kwargs:
+            # Extract database name from project URL
+            parsed = urlparse(kwargs['project_url'])
+            self.db_id = parsed.netloc.split('.')[0]
+        
+        if self.db_id:
+            self._setup_directory_path(self.db_id)
+        
+        logger.info(f"Supabase plugin initialized for project: {self.db_id}")
+    
+    def _setup_directory_path(self, db_id: str) -> None:
+        """Set up the database directory path"""
+        if isinstance(self.db_root_path, str):
+            self.db_root_path = Path(self.db_root_path)
+        
+        self.db_directory_path = Path(self.db_root_path) / f"{self.db_mode}_databases" / db_id
+        self.db_id = db_id
+        
+        # Reset LSH manager when directory path changes
+        self._lsh_manager = None
+    
+    @property
+    def lsh_manager(self):
+        """Lazy load LSH manager for backward compatibility"""
+        if self._lsh_manager is None and self.db_directory_path:
+            from ..lsh.manager import LshManager
+            self._lsh_manager = LshManager(self.db_directory_path)
+        return self._lsh_manager
+    
+    # LSH integration methods for backward compatibility
+    def set_lsh(self) -> str:
+        """Set LSH for backward compatibility"""
+        try:
+            if self.lsh_manager and self.lsh_manager.load_lsh():
+                return "success"
+            else:
+                return "error"
+        except Exception as e:
+            logger.error(f"Error loading LSH: {e}")
+            return "error"
+    
+    def query_lsh(self, keyword: str, signature_size: int = 30, n_gram: int = 3, top_n: int = 10) -> Dict[str, Dict[str, List[str]]]:
+        """Query LSH for backward compatibility"""
+        if self.lsh_manager:
+            try:
+                return self.lsh_manager.query(
+                    keyword=keyword,
+                    signature_size=signature_size,
+                    n_gram=n_gram,
+                    top_n=top_n
+                )
+            except Exception as e:
+                logger.error(f"LSH query failed: {e}")
+                raise Exception(f"Error querying LSH for {self.db_id}: {e}")
+        else:
+            raise Exception(f"LSH not available for {self.db_id}")
+    
+    def get_connection_info(self) -> Dict[str, Any]:
+        """Get connection information"""
+        base_info = super().get_plugin_info()
+        
+        if self.adapter:
+            adapter_info = self.adapter.get_connection_info()
+            base_info.update(adapter_info)
+        
+        base_info.update({
+            "db_id": self.db_id,
+            "db_directory_path": str(self.db_directory_path) if self.db_directory_path else None,
+            "lsh_available": self.lsh_manager is not None,
+            "project_url": self.project_url,
+            "connection_mode": "REST API" if self.connection_params.get('use_rest_api') else "Direct Database"
+        })
+        
+        return base_info
+    
+    def get_example_data(self, table_name: str, number_of_rows: int = 30) -> Dict[str, List[Any]]:
+        """Get example data through adapter"""
+        if self.adapter:
+            return self.adapter.get_example_data(table_name, number_of_rows)
+        else:
+            raise RuntimeError("Plugin not initialized")
+    
+    def get_required_parameters(self) -> Dict[str, Any]:
+        """Get required connection parameters for Supabase"""
+        use_rest_api = self.connection_params.get('use_rest_api', False)
+        
+        if use_rest_api:
+            return {
+                "required": ["project_url", "api_key"],
+                "optional": ["schema", "timeout", "pool_size"],
+                "connection_mode": "REST API"
+            }
+        else:
+            return {
+                "required": ["host", "port", "database", "user", "password"],
+                "optional": ["schema", "sslmode", "sslcert", "sslkey", "sslrootcert", "pool_size", "connect_timeout"],
+                "connection_mode": "Direct Database"
+            }

{thoth_dbmanager-0.4.0.dist-info → thoth_dbmanager-0.4.2.dist-info}/METADATA

@@ -1,13 +1,13 @@
 Metadata-Version: 2.4
-Name: thoth-dbmanager
-Version: 0.4.0
+Name: thoth_dbmanager
+Version: 0.4.2
 Summary: A Python library for managing SQL databases with support for multiple database types, LSH-based similarity search, and a modern plugin architecture.
 Author-email: Marco Pancotti <mp@tylconsulting.it>
 License: MIT
-Project-URL: Homepage, https://github.com/mptyl/thoth-dbmanager
-Project-URL: Bug Tracker, https://github.com/mptyl/thoth-dbmanager/issues
-Project-URL: Documentation, https://github.com/mptyl/thoth-dbmanager#readme
-Project-URL: Source Code, https://github.com/mptyl/thoth-dbmanager
+Project-URL: Homepage, https://github.com/mptyl/thoth_dbmanager
+Project-URL: Bug Tracker, https://github.com/mptyl/thoth_dbmanager/issues
+Project-URL: Documentation, https://github.com/mptyl/thoth_dbmanager#readme
+Project-URL: Source Code, https://github.com/mptyl/thoth_dbmanager
 Keywords: database,sql,lsh,similarity-search,orm
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.8
@@ -45,6 +45,8 @@ Requires-Dist: supabase>=2.0.0; extra == "supabase"
 Requires-Dist: postgrest-py>=0.16.0; extra == "supabase"
 Requires-Dist: gotrue-py>=2.0.0; extra == "supabase"
 Provides-Extra: sqlite
+Provides-Extra: qdrant
+Requires-Dist: qdrant-client>=1.7.0; extra == "qdrant"
 Provides-Extra: all
 Requires-Dist: psycopg2-binary>=2.9.0; extra == "all"
 Requires-Dist: mysql-connector-python>=8.0.0; extra == "all"
@@ -55,6 +57,7 @@ Requires-Dist: informixdb>=2.2.0; extra == "all"
 Requires-Dist: supabase>=2.0.0; extra == "all"
 Requires-Dist: postgrest-py>=0.16.0; extra == "all"
 Requires-Dist: gotrue-py>=2.0.0; extra == "all"
+Requires-Dist: qdrant-client>=1.7.0; extra == "all"
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0.0; extra == "dev"
 Requires-Dist: pytest-cov>=4.0.0; extra == "dev"

thoth_dbmanager-0.4.2.dist-info/RECORD

@@ -0,0 +1,41 @@
+thoth_dbmanager/ThothDbManager.py,sha256=xyPhB6_3Ao3fGb4LjgCe-MvVI_C-9UXxeazra2dhmkI,17846
+thoth_dbmanager/__init__.py,sha256=Aew8n8l8wMPAf9h6xzzOr7mHBS1-ThTT8js1iHKrAvE,3367
+thoth_dbmanager/documents.py,sha256=z-f7zo_CZHqoGM0qHT8-lSUx4NhnMNZTSajpoFtRxn4,5051
+thoth_dbmanager/dynamic_imports.py,sha256=q99hkh-D-Tnz15_pMJRa_e-gcrgOOYLr5QGhqRhXHgI,7939
+thoth_dbmanager/adapters/__init__.py,sha256=tKIMlo9-gbH_cqnqZJ9yw1zQZKUzsV4hljDUNzcoZXg,486
+thoth_dbmanager/adapters/mariadb.py,sha256=LTsf0gORiwqZkd6WtKcOsYLHyDgysxdqNesBscbJwNs,5709
+thoth_dbmanager/adapters/mysql.py,sha256=TrFbxoMMNWbmUcgkKQYOIfsstmMUmuLlGB7R4ZFEIYI,5698
+thoth_dbmanager/adapters/oracle.py,sha256=JSrsgohjz5PbVc8nI188MZ4QGBQls4ieNmwWfAKA7II,21468
+thoth_dbmanager/adapters/postgresql.py,sha256=qxdlxOV7Nvn8U4Lhat50w87Z2S8AzBfmLfEwKfz7dis,17299
+thoth_dbmanager/adapters/qdrant.py,sha256=sL8ldUbAczENb01MmcrkJBKw0dieWPAG2aj97jznj3k,6557
+thoth_dbmanager/adapters/sqlite.py,sha256=RTDszgnAtkE14LKFeoe9lBHgsqXqkmDk6jDCTmVpnoM,14659
+thoth_dbmanager/adapters/sqlserver.py,sha256=V555kUH54Fb1Atow0BfvbSHmoSwGnrB_RJGn718VQSI,23880
+thoth_dbmanager/adapters/supabase.py,sha256=bl2C6LpOpykPF3vIbdNRDk43aXLADzSk0wQuwTcEHZA,10348
+thoth_dbmanager/core/__init__.py,sha256=FlqNW0GZNv1rnwNgyXGzveLqaw0Z90y5AKhR_1DvHBE,269
+thoth_dbmanager/core/factory.py,sha256=sLj8tKI1ADqSlnU5F0NRtHHtpSimnATuwAqUYx5dfxI,9694
+thoth_dbmanager/core/interfaces.py,sha256=x2eHhIIQmrDEnt_0Ll3GicVWVI_rQip_x5k5sNwirFM,9280
+thoth_dbmanager/core/registry.py,sha256=2E9btPrPVnfI4z4JauKVEkN8mp8qy5gIVm3kKxUUD5E,7960
+thoth_dbmanager/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+thoth_dbmanager/helpers/multi_db_generator.py,sha256=LOrWEqh-RTVZXPF7vrlF6BmBGh1XC09kM6XkqkFNPQ4,23232
+thoth_dbmanager/helpers/preprocess_values.py,sha256=sKMjD50UL2CZG7-g7KGe3TJQeXmXC7n28LGge8CaP5c,6183
+thoth_dbmanager/helpers/schema.py,sha256=PhHgUxfFe7qUPxdpWQ3YfS593VQjyKtaSKMlChCyLNo,13978
+thoth_dbmanager/helpers/search.py,sha256=k04L7clSPfeQOlq_ifsH3PZ21ZK-rujh_Qy4hWrvSso,3970
+thoth_dbmanager/lsh/__init__.py,sha256=zAUTRuRX4MNc2nU93tTBkT2_IVL7OBkBL0WZPPgO3Ec,555
+thoth_dbmanager/lsh/core.py,sha256=171FqHW7ItAqAPk6g_AoayKTE3Bs1rRZxnt55MJVzjY,6813
+thoth_dbmanager/lsh/factory.py,sha256=2Bpkk-OygjaptZAw1yysxO1cxG3QTxmJ1yFGcXHqX3w,2411
+thoth_dbmanager/lsh/manager.py,sha256=NDKU2re6Xp7cU6pB5EtucZqQgPiT5nyugoUCCIAG29s,5777
+thoth_dbmanager/lsh/storage.py,sha256=Wp1fjVBVE3inRFTAZCxlc6vwZdlKDwY7wCfwfu7P5Xw,3085
+thoth_dbmanager/plugins/__init__.py,sha256=98iKwNiKnFFeJfw0qLGVOrP2Mo2wGPAolsx7R2B3emA,592
+thoth_dbmanager/plugins/mariadb.py,sha256=ElYa4Rexwrofcjcs0UQKan8fZpbU6-n9zghYR9SgRb4,17975
+thoth_dbmanager/plugins/mysql.py,sha256=mbDsIDV2H_BWYANU4JHMsUkxLQICuGtjKTTPbig2Ngs,16546
+thoth_dbmanager/plugins/oracle.py,sha256=k4Yxvz5MdsH3Sfty9lxbhr8igSnHvGbGujz3bLpNcHo,5230
+thoth_dbmanager/plugins/postgresql.py,sha256=GF6k4K0t7-Y08THWJzS0eWJkrQ1e4GfoKIcanC0Z5Ng,5401
+thoth_dbmanager/plugins/qdrant.py,sha256=XWmc4K6ILiwvB1BhyJBBtA7ohRzQmk1MiZ1N7S-YF78,1199
+thoth_dbmanager/plugins/sqlite.py,sha256=esgJqDp2aSu4a32WnmynfFyY9JfW5W8VjNTkaA-hZhM,6402
+thoth_dbmanager/plugins/sqlserver.py,sha256=mMb3F5FmSWV02FZwj-Ult-2TjuyeVA4Fl1iME1dbgLU,5289
+thoth_dbmanager/plugins/supabase.py,sha256=mWlaGAdpywx4-pU4Ffpmn24ze8sg0sM5kc6bFDoeYRg,8645
+thoth_dbmanager-0.4.2.dist-info/licenses/LICENSE,sha256=81-BOzGgwtY1XdYfkwMQB87AkOGXI9OMq0kjNcZA4UE,1071
+thoth_dbmanager-0.4.2.dist-info/METADATA,sha256=iMFzfRMoh0BqUahLBLt2Pny1oA7L6F1tB7iF8W4D9no,12183
+thoth_dbmanager-0.4.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+thoth_dbmanager-0.4.2.dist-info/top_level.txt,sha256=b9ttxm9RUc0KUCASEKRx6FqoREYJ1-KZWSpNuaM0uQ4,16
+thoth_dbmanager-0.4.2.dist-info/RECORD,,

thoth_dbmanager-0.4.2.dist-info/top_level.txt

@@ -0,0 +1 @@
+thoth_dbmanager

thoth_dbmanager-0.4.0.dist-info/RECORD

@@ -1,5 +0,0 @@
-thoth_dbmanager-0.4.0.dist-info/licenses/LICENSE,sha256=81-BOzGgwtY1XdYfkwMQB87AkOGXI9OMq0kjNcZA4UE,1071
-thoth_dbmanager-0.4.0.dist-info/METADATA,sha256=oWkABO_DVnwtvyZIHUzy8PwWiBF3WWk7e3anrY4VxIE,12053
-thoth_dbmanager-0.4.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-thoth_dbmanager-0.4.0.dist-info/top_level.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-thoth_dbmanager-0.4.0.dist-info/RECORD,,

thoth_dbmanager-0.4.0.dist-info/top_level.txt

@@ -1 +0,0 @@
-