mcp-dbutils 1.0.2__py3-none-any.whl → 1.0.3__py3-none-any.whl

mcp_dbutils/base.py

@@ -569,6 +569,7 @@ class ConnectionServer:
         """Extract table name from SQL statement
 
         This is a simple implementation that works for basic SQL statements.
+        Handles multi-line SQL statements by normalizing whitespace.
 
         Args:
             sql: SQL statement
@@ -577,23 +578,26 @@ class ConnectionServer:
             str: Table name
         """
         sql_type = self._get_sql_type(sql)
-        sql = sql.strip()
+
+        # 预处理SQL：规范化空白字符，将多行SQL转换为单行
+        # 将所有连续的空白字符（包括换行符、制表符等）替换为单个空格
+        normalized_sql = " ".join(sql.split())
 
         if sql_type == "INSERT":
             # INSERT INTO table_name ...
-            match = sql.upper().split("INTO", 1)
+            match = normalized_sql.upper().split("INTO", 1)
             if len(match) > 1:
                 table_part = match[1].strip().split(" ", 1)[0]
                 return table_part.strip('`"[]')
         elif sql_type == "UPDATE":
             # UPDATE table_name ...
-            match = sql.upper().split("UPDATE", 1)
+            match = normalized_sql.upper().split("UPDATE", 1)
             if len(match) > 1:
                 table_part = match[1].strip().split(" ", 1)[0]
                 return table_part.strip('`"[]')
         elif sql_type == "DELETE":
             # DELETE FROM table_name ...
-            match = sql.upper().split("FROM", 1)
+            match = normalized_sql.upper().split("FROM", 1)
             if len(match) > 1:
                 table_part = match[1].strip().split(" ", 1)[0]
                 return table_part.strip('`"[]')

{mcp_dbutils-1.0.2.dist-info → mcp_dbutils-1.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp-dbutils
-Version: 1.0.2
+Version: 1.0.3
 Summary: MCP Database Utilities Service
 Author: Dong Hao
 License-Expression: MIT

{mcp_dbutils-1.0.2.dist-info → mcp_dbutils-1.0.3.dist-info}/RECORD

@@ -1,6 +1,6 @@
 mcp_dbutils/__init__.py,sha256=6LLccQv7je2L4IpY_I3OzSJZcK32VUDJv2IY31y6eYg,1900
 mcp_dbutils/audit.py,sha256=U-Fd511HxOQH4BxfCXgd4KyaBLESEBnGsPkUNviSTwc,7294
-mcp_dbutils/base.py,sha256=2KhuovVvSUHV664ppLy1LO4a-CfN2rBF-BrP6eHzIb4,61907
+mcp_dbutils/base.py,sha256=oOoKdBbHLB3PfuhQ_fZXZz-4G8df4q5HeZPB4U45HKs,62206
 mcp_dbutils/config.py,sha256=zwN9yPKv4WvEPG3WIRT6uBVZSRxFniSmN2kEog7KPcI,5921
 mcp_dbutils/log.py,sha256=mqxi6I_IL-MF1F_pxBtnYZQKOHbGBJ74gsvZHVelr1w,823
 mcp_dbutils/stats.py,sha256=wMqWPfGnEOg9v5YBtTsARV-1YsFUMM_pKdzitzSU9x4,7137
@@ -16,8 +16,8 @@ mcp_dbutils/sqlite/__init__.py,sha256=fK_3-WylCBYpBAzwuopi8hlwoIGJm2TPAlwcPWG46I
 mcp_dbutils/sqlite/config.py,sha256=rsfAE8yaCVZC39ziXssqsi0EXUOEWA-MtKHvrO-6jG4,4933
 mcp_dbutils/sqlite/handler.py,sha256=vpkyCow26hpBqigNUNW0VGyWhsTz8uFflssM7K-FJi4,21882
 mcp_dbutils/sqlite/server.py,sha256=EBKNKz_wTvChwg6BZlvZIBA1H5mmE2NiNEMOgu_CMy4,7373
-mcp_dbutils-1.0.2.dist-info/METADATA,sha256=hZm7hYfZKEzyNJmYCFNPvdXq0JNkJVOj4IJDbfOuihE,8440
-mcp_dbutils-1.0.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mcp_dbutils-1.0.2.dist-info/entry_points.txt,sha256=XTjt0QmYRgKOJQT6skR9bp1EMUfIrgpHeZJPZ3CJffs,49
-mcp_dbutils-1.0.2.dist-info/licenses/LICENSE,sha256=1A_CwpWVlbjrKdVEYO77vYfnXlW7oxcilZ8FpA_BzCI,1065
-mcp_dbutils-1.0.2.dist-info/RECORD,,
+mcp_dbutils-1.0.3.dist-info/METADATA,sha256=xvp1x0QBA1tHw04kx__QfK9wpsahjMNoO_6knPsXaBg,8440
+mcp_dbutils-1.0.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_dbutils-1.0.3.dist-info/entry_points.txt,sha256=XTjt0QmYRgKOJQT6skR9bp1EMUfIrgpHeZJPZ3CJffs,49
+mcp_dbutils-1.0.3.dist-info/licenses/LICENSE,sha256=1A_CwpWVlbjrKdVEYO77vYfnXlW7oxcilZ8FpA_BzCI,1065
+mcp_dbutils-1.0.3.dist-info/RECORD,,