pymysql的使用

pymysql是从Python连接到MySQL数据库服务器的接口,其官方文档为:https://pymysql.readthedocs.io/en/latest/

安装:pip install pymysql

对于数据库的操作,我们一般是这样的操作:
步骤1:连接数据库
步骤2:数据库的操作(增删改查)
步骤3:关闭数据库连接

以下是pymysql经常被用到的一些方法

方法 描述
pymysql.connect() connect()方法返回要给数据库连接对象,参数可以传入很多,常用的参数有:host、port、user、password、database、charset,connect()创建了连接对象,执行完sql操作后,必须使用close()关闭
close() 数据库连接对象的一个方法,用于关闭数据库连接
cursor() 数据库连接对象的一个方法,用于获取游标对象,游标对象的execute(sql语句)方法可以执行sql语句
execute(sql) 游标对象的一个方法,可以执行sql语句
commit() 提交到数据库,数据库连接对象的一个方法,如果对表数据有修改的时候,就需要将修改提交到数据库,否则修改没有生效
rollback() 回滚已提交的内容,,数据库连接对象的一个方法,依据事务的原子性 ,提交要么全部生效,要么全不生效,如果遇到异常,需要对已提交的内容进行回滚

数据库连接

方法 描述
pymysql.connect() connect()方法返回要给数据库连接对象,参数可以传入很多,常用的参数有:host、port、user、password、database、charset,connect()创建了连接对象,执行完sql操作后,必须使用close()关闭
close() 数据库连接对象的一个方法,用于关闭数据库连接
  • host:mysql服务器地址
  • port:mysql服务器连接端口,默认值为3306
  • user:用户名
  • password:密码
  • database:数据库名称
  • charset:编码方式,推荐使用"utf8mb4"

简单示例:

import pymysql

# 建立连接
conn = pymysql.connect(host="localhost",
                       port=3306,
                       user="root",
                       password="wen",
                       database="mysql",
                       charset="utf8mb4")

# 关闭连接
conn.close()

数据库操作

CRUD操作

mysql的操作一般归纳为数据库、表、表数据的增删改查

功能 描述
创建表 create table 表名 (column_name column_type);
删除表 drop table 表名;
查询表数据 select 字段名1,字段名2,…,字段名n from 表名 where xxx [limit n][offset m];
插入表数据 insert into 表名(字段名1,字段名2,…,字段名n) values(值1,值2,…,值n),(值21,值22,…,值2n)… ;
更新表数据 update 表名 set 字段名1=新值1,字段名2=新值2,…,字段名n=新值n where xxx;
删除表数据 delete from 表名 where xxx;

示例:

import pymysql

# 建立连接
conn = pymysql.connect(host="localhost",
                       port=3306,
                       user="root",
                       password="wen",
                       database="test",
                       charset="utf8mb4")
# 执行操作(先获取游标对象,再执行sql语句)
cursor = conn.cursor()  # 获取游标对象

cursor.execute("show tables;")
print(f"当前test库的表有(创建表前):{cursor.fetchall()}")
# 创建表
creat_table_sql = """
    create table person(
    `id` int unsigned auto_increment primary key,
    `name` varchar(40) not null,
    `birthday` date);
"""
cursor.execute(creat_table_sql)
cursor.execute("show tables;")
print(f"当前test库的表有(创建表后):{cursor.fetchall()}\n")

# 插入表数据
cursor.execute("insert into person(name, birthday) values('wen', '2001-09-23'), ('xiaoba', '1998-04-17');")
conn.commit()   # 提交记录

# 查询表数据
cursor.execute("select * from person")
print(f"person插入数据后表内容为:{cursor.fetchall()}\n")

# 更新表数据
cursor.execute("update person set birthday='2001-12-17' where name='xiaoba';")
conn.commit()
cursor.execute("select * from person")
print(f"person更新数据后表内容为:{cursor.fetchall()}\n")

# 删除表数据
cursor.execute("delete from person where name='wen'")
conn.commit()
print(f"person删除数据后表内容为:{cursor.fetchall()}\n")

# 关闭连接
conn.close()

执行结果为:

当前test库的表有(创建表前):()
当前test库的表有(创建表后):(('person',),)

person插入数据后表内容为:((1, 'wen', datetime.date(2001, 9, 23)), (2, 'xiaoba', datetime.date(1998, 4, 17)))

person更新数据后表内容为:((1, 'wen', datetime.date(2001, 9, 23)), (2, 'xiaoba', datetime.date(2001, 12, 17)))

person删除数据后表内容为:()

查询操作

游标对象提供了3种获取查询记录的方法

方法 描述
fetchone() 获取单条记录(元组形式)
fetchmany(n) 获取n条记录 (元组形式)
fetchall() 获取所有结果记录(元组形式)

示例:

import pymysql

# 建立连接
conn = pymysql.connect(host="localhost",
                       port=3306,
                       user="root",
                       password="wen",
                       database="test",
                       charset="utf8mb4")
# 执行操作(先获取游标对象,再执行sql语句)
cursor = conn.cursor()  # 获取游标对象

cursor.execute("select * from person where birthday>='2000-01-01'")
result_one = cursor.fetchone()
print(type(result_one))
print(f"fetchone()查询到的内容:{result_one}\n")

cursor.execute("select * from person where birthday>='2000-01-01'")
result_3 = cursor.fetchmany(3)
print(type(result_3))
print(f"fetchmany(3)查询到的内容:{result_3}\n")

cursor.execute("select * from person where birthday>='2000-01-01'")
result_all = cursor.fetchall()
print(type(result_all))
print(f"fetchall()查询到的内容:{result_all}")

# 关闭连接
conn.close()

表内容为:
pymysql的使用_第1张图片

执行结果为:


fetchone()查询到的内容:(1, 'wen', datetime.date(2001, 9, 23))


fetchmany(3)查询到的内容:((1, 'wen', datetime.date(2001, 9, 23)), (3, '张三', datetime.date(2012, 7, 9)), (5, 'harry', datetime.date(2013, 7, 9)))


fetchall()查询到的内容:((1, 'wen', datetime.date(2001, 9, 23)), (3, '张三', datetime.date(2012, 7, 9)), (5, 'harry', datetime.date(2013, 7, 9)), (6, 'heng李', datetime.date(2004, 11, 26)))

执行事务

在增加、修改、删除表数据的时候,并不是说实时更新数据库的,当前连接对数据库的操作,要么全做,要么全不做。
pymysql提供了commit()和rollback()这2个方法

功能 描述
commit() 将游标的所有更新操作进行提交
rollback() 回滚当前游标的所有操作

以下是没有执行commit()方法的示例:

import pymysql

# 建立连接
conn = pymysql.connect(host="localhost",
                       port=3306,
                       user="root",
                       password="wen",
                       database="test",
                       charset="utf8mb4")
# 执行操作(先获取游标对象,再执行sql语句)
cursor = conn.cursor()  # 获取游标对象

# 插入表数据
cursor.execute("insert into person(name, birthday) values('harry', '2004-11-23'), ('heng李', '1983-04-28');")

# 更新表数据
cursor.execute("update person set birthday='2001-12-17' where name='xiaoba';")

# 删除表数据
cursor.execute("delete from person where name='wen'")

# 查询表数据
cursor.execute("select * from person")
print(f"person表内容为:{cursor.fetchall()}\n")

# 关闭连接
conn.close()

控制台打印结果为:

person表内容为:((2, 'xiaoba', datetime.date(2001, 12, 17)), (3, '张三', datetime.date(2012, 7, 9)), (9, 'harry', datetime.date(2004, 11, 23)), (10, 'heng李', datetime.date(1983, 4, 28)))

test数据库的person表在脚本执行前后如图:
pymysql的使用_第2张图片
我们看到,在该没有进行commit的脚本中,控制台打印的数据是更新后的数据,但是实际去mysql服务器查询的时候,脚本执行前后,数据并没有变化,可以判断出:游标对象对表数据的修改并没有在表中实际生效

使用了commit()方法

import pymysql

# 建立连接
conn = pymysql.connect(host="localhost",
                       port=3306,
                       user="root",
                       password="wen",
                       database="test",
                       charset="utf8mb4")
# 执行操作(先获取游标对象,再执行sql语句)
cursor = conn.cursor()  # 获取游标对象

# 插入表数据
cursor.execute("insert into person(name, birthday) values('harry', '2004-11-23'), ('heng李', '1983-04-28');")

# 更新表数据
cursor.execute("update person set birthday='2001-12-17' where name='xiaoba';")

# 删除表数据
cursor.execute("delete from person where name='wen'")
# 将修改内容提交到数据库
conn.commit()

# 查询表数据
cursor.execute("select * from person")
print(f"person表内容为:{cursor.fetchall()}\n")

# 关闭连接
conn.close()

执行后:
控制台打印内容为:

person表内容为:((2, 'xiaoba', datetime.date(2001, 12, 17)), (3, '张三', datetime.date(2012, 7, 9)), (11, 'harry', datetime.date(2004, 11, 23)), (12, 'heng李', datetime.date(1983, 4, 28)))

mysql查询:
pymysql的使用_第3张图片
对比未使用commit()的场景,mysql查询person表的时候,person表内容更新了
rollback()方法一般是用于执行表操作错误后,对已更新的内容进行回滚(如果已经commit()成功的内容,就不回滚,一般来说服,数据库操作均使用try-catch-finally去兼容异常

import pymysql

try:
    # 建立连接
    conn = pymysql.connect(host="localhost",
                           port=3306,
                           user="root",
                           password="wen",
                           database="test",
                           charset="utf8mb4")
    # 执行操作(先获取游标对象,再执行sql语句)
    cursor = conn.cursor()  # 获取游标对象

    # 更新表数据
    cursor.execute("update person set birthdaye='2001-12-17' where name='张三';")
    # 提交修改
    conn.commit()
except Exception as e:
    # 回滚事务
    print(e)
    print("即将对数据进行回滚")
    conn.rollback()
finally:
    # 关闭连接
    conn.close()

执行结果为:

(1054, "Unknown column 'birthdaye' in 'field list'")
即将对数据进行回滚

你可能感兴趣的:(python自动化,mysql)