mcp-server-sqlseed 0.1.0__py3-none-any.whl

mcp_server_sqlseed/__init__.py

@@ -0,0 +1,7 @@
+from __future__ import annotations
+
+from mcp_server_sqlseed.server import mcp
+
+
+def main() -> None:
+    mcp.run()

mcp_server_sqlseed/__main__.py

@@ -0,0 +1,6 @@
+from __future__ import annotations
+
+from mcp_server_sqlseed import main
+
+if __name__ == "__main__":
+    main()

mcp_server_sqlseed/config.py

@@ -0,0 +1,9 @@
+from __future__ import annotations
+
+from pydantic import BaseModel
+
+
+class MCPServerConfig(BaseModel):
+    db_path: str | None = None
+    host: str = "localhost"
+    port: int = 8000

mcp_server_sqlseed/server.py

@@ -0,0 +1,148 @@
+from __future__ import annotations
+
+import hashlib
+import json
+from typing import Any
+
+from mcp.server.fastmcp import FastMCP
+
+mcp = FastMCP("sqlseed")
+
+
+def _serialize_schema_context(ctx: dict[str, Any]) -> dict[str, Any]:
+    return {
+        "table_name": ctx["table_name"],
+        "columns": [
+            {
+                "name": c.name,
+                "type": c.type,
+                "nullable": c.nullable,
+                "default": c.default,
+                "is_primary_key": c.is_primary_key,
+                "is_autoincrement": c.is_autoincrement,
+            }
+            for c in ctx["columns"]
+        ],
+        "foreign_keys": [
+            {"column": fk.column, "ref_table": fk.ref_table, "ref_column": fk.ref_column}
+            for fk in ctx["foreign_keys"]
+        ],
+        "indexes": ctx["indexes"],
+        "sample_data": ctx["sample_data"],
+        "all_table_names": ctx["all_table_names"],
+    }
+
+
+def _compute_schema_hash(schema_ctx: dict[str, Any]) -> str:
+    hash_input = json.dumps(
+        {
+            "columns": [
+                {"name": c.name, "type": c.type, "nullable": c.nullable}
+                for c in schema_ctx["columns"]
+            ],
+            "foreign_keys": [
+                {"column": fk.column, "ref_table": fk.ref_table}
+                for fk in schema_ctx["foreign_keys"]
+            ],
+        },
+        sort_keys=True,
+    )
+    return hashlib.sha256(hash_input.encode()).hexdigest()[:16]
+
+
+@mcp.resource("sqlseed://schema/{db_path}/{table_name}")
+def get_schema_resource(db_path: str, table_name: str) -> str:
+    from sqlseed.core.orchestrator import DataOrchestrator
+
+    with DataOrchestrator(db_path) as orch:
+        ctx = orch.get_schema_context(table_name)
+        serializable_ctx = _serialize_schema_context(ctx)
+        return json.dumps(serializable_ctx, ensure_ascii=False, indent=2)
+
+
+@mcp.tool()
+def sqlseed_inspect_schema(db_path: str, table_name: str | None = None) -> dict[str, Any]:
+    """Inspect database schema. Returns column info, foreign keys, indexes,
+    sample data, and schema_hash for specified table or all tables."""
+    from sqlseed.core.orchestrator import DataOrchestrator
+
+    with DataOrchestrator(db_path) as orch:
+        tables = [table_name] if table_name else orch._db.get_table_names()
+        result: dict[str, Any] = {}
+        for tbl in tables:
+            ctx = orch.get_schema_context(tbl)
+            result[tbl] = _serialize_schema_context(ctx)
+            result[tbl]["schema_hash"] = _compute_schema_hash(ctx)
+        return result
+
+
+@mcp.tool()
+def sqlseed_generate_yaml(db_path: str, table_name: str, max_retries: int = 3) -> str:
+    """Generate YAML config for a table using AI analysis with self-correction.
+    Returns YAML string for human review. Requires sqlseed-ai plugin and API key."""
+    import yaml
+    from sqlseed_ai.analyzer import SchemaAnalyzer
+    from sqlseed_ai.refiner import AiConfigRefiner, AISuggestionFailedError
+
+    analyzer = SchemaAnalyzer()
+    refiner = AiConfigRefiner(analyzer, db_path)
+
+    try:
+        result = refiner.generate_and_refine(
+            table_name=table_name,
+            max_retries=max_retries,
+        )
+    except AISuggestionFailedError as e:
+        return f"# AI suggestion failed: {e}"
+    except Exception as e:
+        return f"# Error: {e}"
+
+    if result:
+        output = {"db_path": db_path, "provider": "mimesis", "locale": "zh_CN", "tables": [result]}
+        return yaml.dump(output, allow_unicode=True, sort_keys=False, default_flow_style=False)
+    return "# No AI suggestions available. Ensure sqlseed-ai plugin is installed and API key is configured."
+
+
+@mcp.tool()
+def sqlseed_execute_fill(
+    db_path: str,
+    table_name: str,
+    count: int = 1000,
+    yaml_config: str | None = None,
+) -> dict[str, Any]:
+    """Execute data generation for a table. Optionally provide YAML config string for column rules."""
+    from sqlseed.core.orchestrator import DataOrchestrator
+
+    with DataOrchestrator(db_path) as orch:
+        column_configs = None
+        clear_before = False
+        seed = None
+
+        if yaml_config:
+            import yaml as _yaml
+
+            from sqlseed.config.models import GeneratorConfig
+
+            data = _yaml.safe_load(yaml_config)
+            config = GeneratorConfig(**data)
+            for t in config.tables:
+                if t.name == table_name:
+                    column_configs = t.columns
+                    clear_before = t.clear_before
+                    seed = t.seed
+                    break
+
+        result = orch.fill_table(
+            table_name=table_name,
+            count=count,
+            column_configs=column_configs,
+            clear_before=clear_before,
+            seed=seed,
+        )
+
+        return {
+            "table_name": result.table_name,
+            "count": result.count,
+            "elapsed": result.elapsed,
+            "errors": result.errors,
+        }

mcp_server_sqlseed-0.1.0.dist-info/METADATA

@@ -0,0 +1,20 @@
+Metadata-Version: 2.4
+Name: mcp-server-sqlseed
+Version: 0.1.0
+Summary: MCP server for sqlseed - AI-powered SQLite test data generation
+Project-URL: Homepage, https://github.com/sunbos/sqlseed
+Project-URL: Repository, https://github.com/sunbos/sqlseed/tree/main/plugins/mcp-server-sqlseed
+Author-email: SunBo <1443584939@qq.com>
+License-Expression: AGPL-3.0-or-later
+Classifier: Development Status :: 3 - Alpha
+Classifier: License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Python: >=3.10
+Requires-Dist: mcp>=1.0
+Requires-Dist: sqlseed
+Provides-Extra: ai
+Requires-Dist: sqlseed-ai; extra == 'ai'

mcp_server_sqlseed-0.1.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+mcp_server_sqlseed/__init__.py,sha256=_ALPs9unfY9drR8wmUQkNMc1KgVVhoI2I0ZyYDZayR4,114
+mcp_server_sqlseed/__main__.py,sha256=ZDnilCg26WgugzNuf7WaQxP20KTQTAAb9hr9-XZ5AUM,111
+mcp_server_sqlseed/config.py,sha256=_e-0JCIPb_RZieDC48-4yJ-B1udpwvGOVPNl0PEiWz4,183
+mcp_server_sqlseed/server.py,sha256=_f9BdOYsPpMVjSmXV8js10V61Y6abaiU6NShv_XxFhQ,4937
+mcp_server_sqlseed-0.1.0.dist-info/METADATA,sha256=BaDqCVM0PAKRFU051gPoRgl5ouj27ifFx5TypXt3khA,888
+mcp_server_sqlseed-0.1.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+mcp_server_sqlseed-0.1.0.dist-info/entry_points.txt,sha256=Vl8J7LFyYfH9Bs5xEc2lI7CidwSpOGRuYQIJ1YTpwYU,63
+mcp_server_sqlseed-0.1.0.dist-info/RECORD,,

mcp_server_sqlseed-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.29.0
+Root-Is-Purelib: true
+Tag: py3-none-any

mcp_server_sqlseed-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+mcp-server-sqlseed = mcp_server_sqlseed:main