mcp-dbutils 1.0.1__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('`"[]')
@@ -625,6 +629,9 @@ class ConnectionServer:
             # 没有细粒度权限控制，默认允许所有写操作
             return
 
+        # 将表名转换为小写，用于大小写不敏感的比较
+        table_name_lower = table_name.lower()
+
         # 检查表级权限
         tables = write_permissions.get("tables", {})
         if not tables:
@@ -638,9 +645,12 @@ class ConnectionServer:
                     operation=operation_type, table=table_name
                 ))
 
-        # 检查特定表的权限
-        if table_name in tables:
-            table_config = tables[table_name]
+        # 创建表名到配置的映射，支持大小写不敏感的比较
+        tables_lower = {k.lower(): v for k, v in tables.items()}
+
+        # 检查特定表的权限（大小写不敏感）
+        if table_name_lower in tables_lower:
+            table_config = tables_lower[table_name_lower]
             operations = table_config.get("operations", ["INSERT", "UPDATE", "DELETE"])
             if operation_type in operations:
                 return

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

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

{mcp_dbutils-1.0.1.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=y82luhDlBZbs1HC4V8jqULAl_LLaXwlMagW45Pwdi2M,61598
+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.1.dist-info/METADATA,sha256=TwdVxwgEQgqSsMsKg0HglvScfwjlw5o6gRpN7HPu_4I,8440
-mcp_dbutils-1.0.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mcp_dbutils-1.0.1.dist-info/entry_points.txt,sha256=XTjt0QmYRgKOJQT6skR9bp1EMUfIrgpHeZJPZ3CJffs,49
-mcp_dbutils-1.0.1.dist-info/licenses/LICENSE,sha256=1A_CwpWVlbjrKdVEYO77vYfnXlW7oxcilZ8FpA_BzCI,1065
-mcp_dbutils-1.0.1.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,,