PyMySQL 是在 Python3.x 版本中用于连接 MySQL 服务器的一个库。

安装:

pip install PyMySQL
数据库增删改查流程差不多,主要在于sql语句。

#!/usr/bin/env python3
#-*- coding:utf-8 -*-
import pymysql
#创建一个数据库链接
db = pymysql.connect("localhost",'root','drcom','TESTDB')
#创建一个游标对象
#游标是系统为用户开设的一个数据缓冲区,存放sql语句的执行结果
cursor = db.cursor()
#执行sql查询,返回值是查询到的数据条目数
cursor.execute("select version()")
#获取查询数据
#fetchone每次只返回一行,也就是结果集的下一行
#fetchall() 得到结果集中剩下的所有行
data = cursor.fetchone()
print("database version:%s"%data)
#执行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插入语句
insert_db = '''INSERT INTO EMPLOYEE(FIRST_NAME,
		LAST_NAME,AGE,SEX,INCOME)
		VALUES('Mac','Mohan',20,'M',2000)'''
try:
	cursor.execute(insert_db)
#提交到数据库执行,不执行这一步,数据是不会被真正的写入到数据库的
	db.commit()
except:
#如果发生错误则回滚
	db.rollback()

cursor.execute("SELECT * FROM TESTDB.EMPLOYEE")
result = cursor.fetchall()
print(result)
#关闭数据库

db.close()