使用PyMySql连接mysql
上篇文章尝试了使用mysql-connector模块连接mysql，此次尝试PyMySql模块。
也是需要先在Pycharm里下载此模块。
一、新建表
我没有关闭数据库连接，代码如下：
import pymysql
db=pymysql.connect( #创建连接mysql
host='localhost', #自己电脑的ip地址
user='root',
passwd='123456',
database="runoob_db" #连接名为runoob_db的数据库
)
print(db)
cursor = db.cursor()
# 使用 **execute()** 方法执行 SQL，如果表存在则删除
cursor.execute("DROP TABLE IF EXISTS EMPLOYEE")
# 使用预处理语句创建表
sql = """CREATE TABLE EMPLOYEE (
FIRST_NAME CHAR(20) NOT NULL,
LAST_NAME CHAR(20),
AGE INT,
SEX CHAR(1),
INCOME FLOAT )"""
cursor.execute(sql)
# 关闭数据库连接
#db.close()
navicat显示如下：
二、插入数据
这里需要注意的是不能把新建表的语句给删了，删了之后新建的数据表会不存在，因此插入记录的代码要接着往下写。
注意：
当你只有一条插入数据的代码时，假如在下面代码中把VALUES ('Mac', 'Mohan', 20, 'M', 2000)运行后改成VALUES ('ff', 'Mohan', 20, 'M', 2000)再运行。那么在navicat上原先的Mac会变成ff而不是又增加一条数据。
import mysql.connector
db=mysql.connector.connect( #创建连接mysql
host='localhost', #自己电脑的ip地址
user='root',
passwd='123456',
database="runoob_db" #连接名为runoob_db的数据库
)
print(db)
cursor = db.cursor()
# 使用 execute() 方法执行 SQL，如果表存在则删除
cursor.execute("DROP TABLE IF EXISTS EMPLOYEE")
# 使用预处理语句创建表
sql = """CREATE TABLE EMPLOYEE (
FIRST_NAME CHAR(20) NOT NULL,
LAST_NAME CHAR(20),
AGE INT,
SEX CHAR(1),
INCOME FLOAT )"""
cursor.execute(sql)
# SQL 插入语句
sql1 = """INSERT INTO EMPLOYEE(FIRST_NAME,
LAST_NAME, AGE, SEX, INCOME)
VALUES ('Mac', 'Mohan', 20, 'M', 2000)"""
try:
# 执行sql语句
cursor.execute(sql1)
# 提交到数据库执行
db.commit()
except:
# 如果发生错误则回滚
db.rollback()
同时sql1也能写成这样：
sql1 = "INSERT INTO EMPLOYEE(FIRST_NAME, \
LAST_NAME, AGE, SEX, INCOME) \
VALUES ('%s', '%s', %s, '%s', %s)" % \
('Mac', 'Mohan', 20, 'M', 2000)
navicat显示如下：
尝试多批量插入数据，目前成功的只有一个笨方法，代码如下：
# SQL 插入语句
sql1 = """INSERT INTO EMPLOYEE(FIRST_NAME,
LAST_NAME, AGE, SEX, INCOME)
VALUES ('Mac', 'Mohan', 20, 'M', 2000)
"""
sql2="""INSERT INTO EMPLOYEE(FIRST_NAME,
LAST_NAME, AGE, SEX, INCOME)
VALUES ('ff', 'Mohan', 20, 'M', 2000)
"""
try:
# 执行sql语句
cursor.execute(sql1)
cursor.execute(sql2)
# 提交到数据库执行
db.commit()
except:
# 如果发生错误则回滚
db.rollback()
三、查询操作
这里对几个方法做个补充。
fetchone(): 该方法获取下一个查询结果集。结果集是一个对象
fetchall(): 接收全部的返回结果行.
rowcount: 这是一个只读属性，并返回执行execute()方法后影响的行数。
查询employee表中income大于1000的数据，代码如下：
# SQL 查询语句
sql = "SELECT * FROM EMPLOYEE \
WHERE INCOME > %s" % (1000)
try:
# 执行SQL语句
cursor.execute(sql)
# 获取所有记录列表
results = cursor.fetchall()
for row in results:
fname = row[0]
lname = row[1]
age = row[2]
sex = row[3]
income = row[4]
# 打印结果
print ("fname=%s,lname=%s,age=%s,sex=%s,income=%s" % \
(fname, lname, age, sex, income ))
except:
print ("Error: unable to fetch data")
Pycharm运行显示如下：
四、更新操作
将age修改为23，代码如下：
# SQL 更新语句
sql3 = "UPDATE EMPLOYEE SET AGE = 23 WHERE SEX = '%c'" % ('M')
try:
# 执行SQL语句
cursor.execute(sql3)
# 提交到数据库执行
db.commit()
except:
# 发生错误时回滚
db.rollback()
navicat修改前：
navicat修改后：
五、删除操作
删除数据表 EMPLOYEE 中 INCOME 小于 2000 的所有数据：
# SQL 删除语句
sql4 = "DELETE FROM EMPLOYEE WHERE INCOME <= %s" % (2000)
try:
# 执行SQL语句
cursor.execute(sql4)
# 提交修改
db.commit()
except:
# 发生错误时回滚
db.rollback()
navicat修改前：
navicat修改后：
六、执行事务
介绍两个方法 commit 或 rollback。
commit()方法游标的所有更新操作，rollback（）方法回滚当前游标的所有操作。每一个方法都开始了一个新的事务。
展示全部代码：
import pymysql
db=pymysql.connect( #创建连接mysql
host='localhost', #自己电脑的ip地址
user='root',
passwd='123456',
database="runoob_db" #连接名为runoob_db的数据库
)
print(db)
cursor = db.cursor()
# 使用 execute() 方法执行 SQL，如果表存在则删除
cursor.execute("DROP TABLE IF EXISTS EMPLOYEE")
# 使用预处理语句创建表
sql = """CREATE TABLE EMPLOYEE (
FIRST_NAME CHAR(20) NOT NULL,
LAST_NAME CHAR(20),
AGE INT,
SEX CHAR(1),
INCOME FLOAT )"""
cursor.execute(sql)
# SQL 插入语句
sql1 = """INSERT INTO EMPLOYEE(FIRST_NAME,
LAST_NAME, AGE, SEX, INCOME)
VALUES ('Mac', 'Mohan', 20, 'M', 2000)
"""
sql2="""INSERT INTO EMPLOYEE(FIRST_NAME,
LAST_NAME, AGE, SEX, INCOME)
VALUES ('ff', 'Mohan', 20, 'M', 2001)
"""
try:
# 执行sql语句
cursor.execute(sql1)
cursor.execute(sql2)
# 提交到数据库执行
db.commit()
except:
# 如果发生错误则回滚
db.rollback()
# SQL 查询语句
sql = "SELECT * FROM EMPLOYEE \
WHERE INCOME > %s" % (1000)
try:
# 执行SQL语句
cursor.execute(sql)
# 获取所有记录列表
results = cursor.fetchall()
for row in results:
fname = row[0]
lname = row[1]
age = row[2]
sex = row[3]
income = row[4]
# 打印结果
print ("fname=%s,lname=%s,age=%s,sex=%s,income=%s" % \
(fname, lname, age, sex, income ))
except:
print ("Error: unable to fetch data")
# SQL 更新语句
sql3 = "UPDATE EMPLOYEE SET AGE = 23 WHERE SEX = '%c'" % ('M')
try:
# 执行SQL语句
cursor.execute(sql3)
# 提交到数据库执行
db.commit()
except:
# 发生错误时回滚
db.rollback()
# SQL 删除语句
sql4 = "DELETE FROM EMPLOYEE WHERE INCOME <= %s" % (2000)
try:
# 执行SQL语句
cursor.execute(sql4)
# 提交修改
db.commit()
except:
# 发生错误时回滚
db.rollback()
# 关闭数据库连接
#db.close()
注意：
找到代码要一直往下写的原因，是因为开头代码有一句删除employee表导致。因此重点掌握各个操作语句，mysql.connector同样适用这些语句，只用在开头改一下模块名即可。
import mysql.connector
db=mysql.connector.connect( #创建连接mysql
host='localhost', #自己电脑的ip地址
user='root',
passwd='123456',
database="runoob_db" #连接名为runoob_db的数据库
)