pymysql对操作MySQL DDL数据库模式定义语言提供了很好的支持。
连接MySQL数据库后,可以使用cursor()方法创建一个游标对象。游标对象用于执行MySQL语句并返回结果。
基本用法
创建数据库、表:
import pymysql
connection = None
try:
# 建立数据库连接
connection = pymysql.connect(
host='localhost',
user='root',
password='root',
database='test_db',
)
#游标对象
cursor = connection.cursor()
#创建数据库
create_db_sql = "CREATE DATABASE IF NOT EXISTS mydatabase"
cursor.execute(create_db_sql)
print("数据库创建成功")
# 选择数据库
cursor.execute("USE mydatabase")
# 创建表
create_table_sql = """
CREATE TABLE IF NOT EXISTS users (
id INT AUTO_INCREMENT PRIMARY KEY,
username VARCHAR(50) NOT NULL UNIQUE,
email VARCHAR(100) NOT NULL,
age INT,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
INDEX idx_username (username),
INDEX idx_email (email)
)
"""
cursor.execute(create_table_sql)
print("表创建成功")
except Exception as e:
print(e)
finally:
if connection:
connection.close()
修改表结构:
#向表中添加字段:alter table 表名 add [column] 字段名 字段类型:
alter table users add phone VARCHAR(20);
#删除表中的字段:alter table 表名 drop [column] 字段名:
alter table users drop phone;
#修改字段的名字:alter table 表名 change [column] 旧字段名 新字段名 字段类型:
alter table users change column username username2 varchar(20);
#修改字段的类型:alter table 表名 modify [column] 字段名 字段类型;
alter table users modify id varchar(20);
#修改表的名字:alter table 旧表名 rename to 新表名:
alter table users rename to users2;
增加字段代码示例:
#游标对象
cursor = connection.cursor()
alter_table_sql = """
ALTER TABLE users
ADD COLUMN phone VARCHAR(20)
"""
cursor.execute(alter_table_sql)
print("增加字段成功")