sqlseed 0.2.0__tar.gz → 0.2.1__tar.gz

{sqlseed-0.2.0 → sqlseed-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlseed
-Version: 0.2.0
+Version: 0.2.1
 Summary: Declarative SQLite test data generation toolkit
 Project-URL: Homepage, https://github.com/sunbos/sqlseed
 Project-URL: Documentation, https://github.com/sunbos/sqlseed#readme

{sqlseed-0.2.0 → sqlseed-0.2.1}/pyproject.toml

@@ -84,7 +84,8 @@ packages = ["src/sqlseed"]
 exclude = [
     "docs/",
     "tests/",
-    "plugins/",
+    "plugins/sqlseed-ai/",
+    "plugins/mcp-server-sqlseed/",
     ".github/",
     ".trae/",
     ".gemini/",

sqlseed-0.2.1/src/sqlseed/plugins/__init__.py

@@ -0,0 +1,12 @@
+from __future__ import annotations
+
+from sqlseed.plugins.hookspecs import PROJECT_NAME, SqlseedHookSpec, hookimpl, hookspec
+from sqlseed.plugins.manager import PluginManager
+
+__all__ = [
+    "PROJECT_NAME",
+    "PluginManager",
+    "SqlseedHookSpec",
+    "hookimpl",
+    "hookspec",
+]

sqlseed-0.2.1/src/sqlseed/plugins/hookspecs.py

@@ -0,0 +1,119 @@
+from __future__ import annotations
+
+from typing import Any
+
+import pluggy
+
+hookspec = pluggy.HookspecMarker("sqlseed")
+hookimpl = pluggy.HookimplMarker("sqlseed")
+
+PROJECT_NAME = "sqlseed"
+
+
+class SqlseedHookSpec:
+    @hookspec
+    def sqlseed_register_providers(self, registry: Any) -> None:
+        raise NotImplementedError
+
+    @hookspec
+    def sqlseed_register_column_mappers(self, mapper: Any) -> None:
+        raise NotImplementedError
+
+    @hookspec(firstresult=True)
+    def sqlseed_ai_analyze_table(
+        self,
+        table_name: str,
+        columns: list[Any],
+        indexes: list[dict[str, Any]],
+        sample_data: list[dict[str, Any]],
+        foreign_keys: list[Any],
+        all_table_names: list[str],
+    ) -> dict[str, Any] | None:
+        """
+        [AI Hook] 分析整张表，返回完整的列配置建议。
+        """
+
+    @hookspec
+    def sqlseed_before_generate(
+        self,
+        table_name: str,
+        count: int,
+        config: Any,
+    ) -> None:
+        raise NotImplementedError
+
+    @hookspec
+    def sqlseed_after_generate(
+        self,
+        table_name: str,
+        count: int,
+        elapsed: float,
+    ) -> None:
+        raise NotImplementedError
+
+    @hookspec
+    def sqlseed_transform_row(
+        self,
+        table_name: str,
+        row: dict[str, Any],
+    ) -> dict[str, Any] | None:
+        """
+        Transform/modify each generated row.
+        Return modified row, or None to keep unchanged.
+        Note: This hook is in the hot path - performance sensitive.
+        """
+
+    @hookspec
+    def sqlseed_transform_batch(
+        self,
+        table_name: str,
+        batch: list[dict[str, Any]],
+    ) -> list[dict[str, Any]] | None:
+        """
+        Transform/modify a batch of generated data.
+        Multiple plugins can chain: each plugin's output feeds into the next.
+        """
+
+    @hookspec
+    def sqlseed_before_insert(
+        self,
+        table_name: str,
+        batch_number: int,
+        batch_size: int,
+    ) -> None:
+        raise NotImplementedError
+
+    @hookspec
+    def sqlseed_after_insert(
+        self,
+        table_name: str,
+        batch_number: int,
+        rows_inserted: int,
+    ) -> None:
+        raise NotImplementedError
+
+    @hookspec
+    def sqlseed_shared_pool_loaded(
+        self,
+        table_name: str,
+        shared_pool: Any,
+    ) -> None:
+        """
+        Called after a table's generated values are loaded into the shared pool.
+        Other plugins can use this to track cross-table associations.
+        """
+
+    @hookspec(firstresult=True)
+    def sqlseed_pre_generate_templates(
+        self,
+        table_name: str,
+        column_name: str,
+        column_type: str,
+        count: int,
+        sample_data: list[Any],
+    ) -> list[Any] | None:
+        """
+        [AI Hook] Pre-generate candidate value pool for columns that cannot match
+        a deterministic generator. Called before DataStream creation.
+        Returns a list of template values, or None if the plugin does not handle this column.
+        """

sqlseed-0.2.1/src/sqlseed/plugins/manager.py

@@ -0,0 +1,37 @@
+from __future__ import annotations
+
+from typing import Any
+
+import pluggy
+
+from sqlseed._utils.logger import get_logger
+from sqlseed.plugins.hookspecs import PROJECT_NAME, SqlseedHookSpec
+
+logger = get_logger(__name__)
+
+
+class PluginManager:
+    def __init__(self) -> None:
+        self._pm = pluggy.PluginManager(PROJECT_NAME)
+        self._pm.add_hookspecs(SqlseedHookSpec)
+
+    def load_plugins(self) -> None:
+        self._pm.load_setuptools_entrypoints(PROJECT_NAME)
+        logger.debug("Loaded plugins", plugins=self._pm.get_plugins())
+
+    def register(self, plugin: Any, name: str | None = None) -> None:
+        self._pm.register(plugin, name=name)
+        logger.debug("Registered plugin", name=name or str(plugin))
+
+    def unregister(self, plugin: Any) -> None:
+        self._pm.unregister(plugin)
+
+    @property
+    def hook(self) -> Any:
+        return self._pm.hook
+
+    def get_plugins(self) -> set[Any]:
+        return self._pm.get_plugins()
+
+    def is_registered(self, plugin: Any) -> bool:
+        return self._pm.is_registered(plugin)