使用python脚本，批量修改mysql数据库表字段名称

前提：搭建python相关情况
目的：

1. <strong>    将驼峰式命名或混合大小写的字段名转换为下划线分隔的小写字段名。
2.     例如：UserName -> user_name, userName -> user_name</strong>

复制代码

一、新建脚本 update_table_column_names.py

1. import pymysql
2. import re
3. import sys
5. def to_snake_case(name):
6.     s1 = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', name)
7.     return re.sub('([a-z0-9])([A-Z])', r'\1_\2', s1).lower()
9. # 获取命令行参数，如果存在则使用指定表名，否则为空
10. specified_table = sys.argv[1] if len(sys.argv) > 1 else None
12. # 连接数据库
13. conn = pymysql.connect(host='192.168.1.1', port=3306, user='root', password='123456', db='test')
14. cursor = conn.cursor()
16. if specified_table:
17.     tables = [(specified_table,)]
18. else:
19.     # 获取所有表名
20.     cursor.execute("SHOW TABLES")
21.     tables = cursor.fetchall
23. for table in tables:
24.     original_table_name = table[0]
25.     new_table_name = to_snake_case(original_table_name)
26.     if original_table_name != new_table_name:
27.         # 使用 ALTER TABLE 语句修改表名
28.         cursor.execute(f"ALTER TABLE `{original_table_name}` RENAME TO `{new_table_name}`")
30.     # 获取表的所有列及其详细信息，包括注释
31.     cursor.execute(f"SHOW FULL COLUMNS FROM `{new_table_name}`")
32.     columns = cursor.fetchall()
33.     for column in columns:
34.         column_name = column[0]
35.         column_type = column[1]
36.         is_nullable = "NULL" if column[3] == "YES" else "NOT NULL"
37.         default_value = f"DEFAULT {column[5]}" if column[5] else ""
38.         extra = column[6]
39.         column_comment = column[8]
40.         
41.         new_column_name = to_snake_case(column_name)
42.         if column_name != new_column_name:
43.             # 保留注释、默认值等元数据
44.             cursor.execute(f"""ALTER TABLE `{new_table_name}`
45.                                CHANGE COLUMN `{column_name}` `{new_column_name}` {column_type}
46.                                {is_nullable} {default_value} {extra} COMMENT '{column_comment}'""")
48. conn.commit()
49. cursor.close()
50. conn.close()

复制代码

二、执行脚本命令

1. 注意：默认不写表名称，则更新所有的表
2. py .\update_table_column_names.py table_name

复制代码

 

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。