{
"event": "PreToolUse",
"tool_name": "Write",
"tool_input": {
"file_path": "\/var\/www\/mcp-servers\/mcp_db\/tools\/insert_tool.py",
"content": "\"\"\"Insert Tool - Fuegt Datensaetze ein.\"\"\"\n\nimport re\nimport time\nfrom datetime import datetime\n\nfrom constants import LOG_ENTRY_MAX_LENGTH, LOG_QUERY_MAX_LENGTH, MS_PER_SECOND\n\nfrom config import Config\nfrom infrastructure.db_connection import DatabaseConnection\nfrom shared.domain import LogEntry\nfrom shared.infrastructure import get_logger\n\n\ndef _validate_identifier(name: str) -> bool:\n \"\"\"Validiert Tabellen-\/Spaltennamen gegen SQL-Injection.\"\"\"\n return bool(re.match(r\"^[a-zA-Z0-9_]+$\", name))\n\n\ndef register_insert_tool(mcp) -> None:\n \"\"\"Registriert db_insert Tool.\"\"\"\n logger = get_logger(\"mcp-db\", Config)\n\n @mcp.tool()\n def db_insert(\n table: str,\n data: dict,\n database: str = \"ki_dev\",\n ) -> dict:\n \"\"\"\n Fuegt einen Datensatz ein.\n\n Args:\n table: Zieltabelle\n data: Dict mit Spalte:Wert Paaren\n database: Zieldatenbank (ki_dev oder ki_content)\n\n Returns:\n Dict mit status, inserted_id, error\n \"\"\"\n start = time.time()\n\n # Validierung: Tabellenname\n if not _validate_identifier(table):\n return {\n \"status\": \"denied\",\n \"error\": \"Invalid table name.\",\n }\n\n # Validierung: Datenbank\n if database not in Config.ALLOWED_DATABASES:\n return {\n \"status\": \"denied\",\n \"error\": f\"Database '{database}' not allowed.\",\n }\n\n # Validierung: Data nicht leer\n if not data:\n return {\n \"status\": \"denied\",\n \"error\": \"Data dict must not be empty.\",\n }\n\n # Validierung: Spaltennamen\n for col in data:\n if not _validate_identifier(col):\n return {\n \"status\": \"denied\",\n \"error\": f\"Invalid column name: {col}\",\n }\n\n try:\n with DatabaseConnection.get_connection(database) as conn:\n cursor = conn.cursor(buffered=True)\n\n # Build parameterized INSERT\n columns = list(data.keys())\n placeholders = \", \".join([\"%s\"] * len(columns))\n columns_str = \", \".join([f\"`{c}`\" for c in columns])\n values = tuple(data.values())\n\n query = f\"INSERT INTO `{table}` ({columns_str}) VALUES ({placeholders})\"\n cursor.execute(query, values)\n inserted_id = cursor.lastrowid\n conn.commit()\n cursor.close()\n\n duration = int((time.time() - start) * MS_PER_SECOND)\n\n # Log\n try:\n logger.log(\n LogEntry(\n timestamp=datetime.now(),\n client_name=\"mcp-db\",\n tool_name=\"db_insert\",\n request=f\"INSERT INTO {table} ({columns_str})\",\n status=\"success\",\n duration_ms=duration,\n )\n )\n except Exception:\n pass\n\n return {\n \"status\": \"success\",\n \"table\": table,\n \"inserted_id\": inserted_id,\n \"execution_ms\": duration,\n }\n\n except Exception as e:\n duration = int((time.time() - start) * MS_PER_SECOND)\n\n try:\n logger.log(\n LogEntry(\n timestamp=datetime.now(),\n client_name=\"mcp-db\",\n tool_name=\"db_insert\",\n request=f\"INSERT INTO {table}\",\n status=\"error\",\n duration_ms=duration,\n error_message=str(e)[:LOG_ENTRY_MAX_LENGTH],\n )\n )\n except Exception:\n pass\n\n return {\n \"status\": \"error\",\n \"error\": str(e)[:LOG_QUERY_MAX_LENGTH],\n \"execution_ms\": duration,\n }\n"
}
}
{
"tool_response": {
"type": "update",
"filePath": "\/var\/www\/mcp-servers\/mcp_db\/tools\/insert_tool.py",
"content": "\"\"\"Insert Tool - Fuegt Datensaetze ein.\"\"\"\n\nimport re\nimport time\nfrom datetime import datetime\n\nfrom constants import LOG_ENTRY_MAX_LENGTH, LOG_QUERY_MAX_LENGTH, MS_PER_SECOND\n\nfrom config import Config\nfrom infrastructure.db_connection import DatabaseConnection\nfrom shared.domain import LogEntry\nfrom shared.infrastructure import get_logger\n\n\ndef _validate_identifier(name: str) -> bool:\n \"\"\"Validiert Tabellen-\/Spaltennamen gegen SQL-Injection.\"\"\"\n return bool(re.match(r\"^[a-zA-Z0-9_]+$\", name))\n\n\ndef register_insert_tool(mcp) -> None:\n \"\"\"Registriert db_insert Tool.\"\"\"\n logger = get_logger(\"mcp-db\", Config)\n\n @mcp.tool()\n def db_insert(\n table: str,\n data: dict,\n database: str = \"ki_dev\",\n ) -> dict:\n \"\"\"\n Fuegt einen Datensatz ein.\n\n Args:\n table: Zieltabelle\n data: Dict mit Spalte:Wert Paaren\n database: Zieldatenbank (ki_dev oder ki_content)\n\n Returns:\n Dict mit status, inserted_id, error\n \"\"\"\n start = time.time()\n\n # Validierung: Tabellenname\n if not _validate_identifier(table):\n return {\n \"status\": \"denied\",\n \"error\": \"Invalid table name.\",\n }\n\n # Validierung: Datenbank\n if database not in Config.ALLOWED_DATABASES:\n return {\n \"status\": \"denied\",\n \"error\": f\"Database '{database}' not allowed.\",\n }\n\n # Validierung: Data nicht leer\n if not data:\n return {\n \"status\": \"denied\",\n \"error\": \"Data dict must not be empty.\",\n }\n\n # Validierung: Spaltennamen\n for col in data:\n if not _validate_identifier(col):\n return {\n \"status\": \"denied\",\n \"error\": f\"Invalid column name: {col}\",\n }\n\n try:\n with DatabaseConnection.get_connection(database) as conn:\n cursor = conn.cursor(buffered=True)\n\n # Build parameterized INSERT\n columns = list(data.keys())\n placeholders = \", \".join([\"%s\"] * len(columns))\n columns_str = \", \".join([f\"`{c}`\" for c in columns])\n values = tuple(data.values())\n\n query = f\"INSERT INTO `{table}` ({columns_str}) VALUES ({placeholders})\"\n cursor.execute(query, values)\n inserted_id = cursor.lastrowid\n conn.commit()\n cursor.close()\n\n duration = int((time.time() - start) * MS_PER_SECOND)\n\n # Log\n try:\n logger.log(\n LogEntry(\n timestamp=datetime.now(),\n client_name=\"mcp-db\",\n tool_name=\"db_insert\",\n request=f\"INSERT INTO {table} ({columns_str})\",\n status=\"success\",\n duration_ms=duration,\n )\n )\n except Exception:\n pass\n\n return {\n \"status\": \"success\",\n \"table\": table,\n \"inserted_id\": inserted_id,\n \"execution_ms\": duration,\n }\n\n except Exception as e:\n duration = int((time.time() - start) * MS_PER_SECOND)\n\n try:\n logger.log(\n LogEntry(\n timestamp=datetime.now(),\n client_name=\"mcp-db\",\n tool_name=\"db_insert\",\n request=f\"INSERT INTO {table}\",\n status=\"error\",\n duration_ms=duration,\n error_message=str(e)[:LOG_ENTRY_MAX_LENGTH],\n )\n )\n except Exception:\n pass\n\n return {\n \"status\": \"error\",\n \"error\": str(e)[:LOG_QUERY_MAX_LENGTH],\n \"execution_ms\": duration,\n }\n",
"structuredPatch": [
{
"oldStart": 1,
"oldLines": 17,
"newStart": 1,
"newLines": 15,
"lines": [
"-\"\"\"Insert Tool - Fügt Datensätze ein.\"\"\"",
"+\"\"\"Insert Tool - Fuegt Datensaetze ein.\"\"\"",
"+",
" import re",
"-import sys",
" import time",
"+from datetime import datetime",
" ",
"-sys.path.insert(0, \"\/var\/www\/mcp-servers\/mcp_db\")",
"-sys.path.insert(0, \"\/var\/www\/mcp-servers\/shared\")",
"-",
" from constants import LOG_ENTRY_MAX_LENGTH, LOG_QUERY_MAX_LENGTH, MS_PER_SECOND",
" ",
" from config import Config",
"-from domain.log_contract import LogEntry",
" from infrastructure.db_connection import DatabaseConnection",
"-from infrastructure.protokoll_logger import ProtokollLogger",
"+from shared.domain import LogEntry",
"+from shared.infrastructure import get_logger",
" ",
" ",
" def _validate_identifier(name: str) -> bool:"
]
},
{
"oldStart": 21,
"oldLines": 7,
"newStart": 19,
"newLines": 7,
"lines": [
" ",
" def register_insert_tool(mcp) -> None:",
" \"\"\"Registriert db_insert Tool.\"\"\"",
"- logger = ProtokollLogger()",
"+ logger = get_logger(\"mcp-db\", Config)",
" ",
" @mcp.tool()",
" def db_insert("
]
},
{
"oldStart": 30,
"oldLines": 7,
"newStart": 28,
"newLines": 7,
"lines": [
" database: str = \"ki_dev\",",
" ) -> dict:",
" \"\"\"",
"- Fügt einen Datensatz ein.",
"+ Fuegt einen Datensatz ein.",
" ",
" Args:",
" table: Zieltabelle"
]
},
{
"oldStart": 93,
"oldLines": 6,
"newStart": 91,
"newLines": 9,
"lines": [
" try:",
" logger.log(",
" LogEntry(",
"+ timestamp=datetime.now(),",
"+ client_name=\"mcp-db\",",
"+ tool_name=\"db_insert\",",
" request=f\"INSERT INTO {table} ({columns_str})\",",
" status=\"success\",",
" duration_ms=duration,"
]
},
{
"oldStart": 114,
"oldLines": 6,
"newStart": 115,
"newLines": 9,
"lines": [
" try:",
" logger.log(",
" LogEntry(",
"+ timestamp=datetime.now(),",
"+ client_name=\"mcp-db\",",
"+ tool_name=\"db_insert\",",
" request=f\"INSERT INTO {table}\",",
" status=\"error\",",
" duration_ms=duration,"
]
}
],
"originalFile": "\"\"\"Insert Tool - Fügt Datensätze ein.\"\"\"\nimport re\nimport sys\nimport time\n\nsys.path.insert(0, \"\/var\/www\/mcp-servers\/mcp_db\")\nsys.path.insert(0, \"\/var\/www\/mcp-servers\/shared\")\n\nfrom constants import LOG_ENTRY_MAX_LENGTH, LOG_QUERY_MAX_LENGTH, MS_PER_SECOND\n\nfrom config import Config\nfrom domain.log_contract import LogEntry\nfrom infrastructure.db_connection import DatabaseConnection\nfrom infrastructure.protokoll_logger import ProtokollLogger\n\n\ndef _validate_identifier(name: str) -> bool:\n \"\"\"Validiert Tabellen-\/Spaltennamen gegen SQL-Injection.\"\"\"\n return bool(re.match(r\"^[a-zA-Z0-9_]+$\", name))\n\n\ndef register_insert_tool(mcp) -> None:\n \"\"\"Registriert db_insert Tool.\"\"\"\n logger = ProtokollLogger()\n\n @mcp.tool()\n def db_insert(\n table: str,\n data: dict,\n database: str = \"ki_dev\",\n ) -> dict:\n \"\"\"\n Fügt einen Datensatz ein.\n\n Args:\n table: Zieltabelle\n data: Dict mit Spalte:Wert Paaren\n database: Zieldatenbank (ki_dev oder ki_content)\n\n Returns:\n Dict mit status, inserted_id, error\n \"\"\"\n start = time.time()\n\n # Validierung: Tabellenname\n if not _validate_identifier(table):\n return {\n \"status\": \"denied\",\n \"error\": \"Invalid table name.\",\n }\n\n # Validierung: Datenbank\n if database not in Config.ALLOWED_DATABASES:\n return {\n \"status\": \"denied\",\n \"error\": f\"Database '{database}' not allowed.\",\n }\n\n # Validierung: Data nicht leer\n if not data:\n return {\n \"status\": \"denied\",\n \"error\": \"Data dict must not be empty.\",\n }\n\n # Validierung: Spaltennamen\n for col in data:\n if not _validate_identifier(col):\n return {\n \"status\": \"denied\",\n \"error\": f\"Invalid column name: {col}\",\n }\n\n try:\n with DatabaseConnection.get_connection(database) as conn:\n cursor = conn.cursor(buffered=True)\n\n # Build parameterized INSERT\n columns = list(data.keys())\n placeholders = \", \".join([\"%s\"] * len(columns))\n columns_str = \", \".join([f\"`{c}`\" for c in columns])\n values = tuple(data.values())\n\n query = f\"INSERT INTO `{table}` ({columns_str}) VALUES ({placeholders})\"\n cursor.execute(query, values)\n inserted_id = cursor.lastrowid\n conn.commit()\n cursor.close()\n\n duration = int((time.time() - start) * MS_PER_SECOND)\n\n # Log\n try:\n logger.log(\n LogEntry(\n request=f\"INSERT INTO {table} ({columns_str})\",\n status=\"success\",\n duration_ms=duration,\n )\n )\n except Exception:\n pass\n\n return {\n \"status\": \"success\",\n \"table\": table,\n \"inserted_id\": inserted_id,\n \"execution_ms\": duration,\n }\n\n except Exception as e:\n duration = int((time.time() - start) * MS_PER_SECOND)\n\n try:\n logger.log(\n LogEntry(\n request=f\"INSERT INTO {table}\",\n status=\"error\",\n duration_ms=duration,\n error_message=str(e)[:LOG_ENTRY_MAX_LENGTH],\n )\n )\n except Exception:\n pass\n\n return {\n \"status\": \"error\",\n \"error\": str(e)[:LOG_QUERY_MAX_LENGTH],\n \"execution_ms\": duration,\n }\n"
}
}