0.安装
pip install pymysql
1.打开数据库连接
import pymysql
db = pymysql.connect(host="数据库地址",
user="用户名",
password="密码",
port="端口",
database="数据库名",
charset='utf8')
2.创建游标
cursor = db.cursor()
3.操作数据库
3.1创建表
# 如果数据表已经存在使用execute()方法删除表。
cursor.execute("DROP TABLE IF EXISTS EMPLOYEE")
# 创建数据表SQL语句
sql = """CREATE TABLE EMPLOYEE (
id int(7) NOT NULL auto_increment,
FIRST_NAME CHAR(20) NOT NULL,
LAST_NAME CHAR(20),
AGE INT,
SEX CHAR(1),
INCOME FLOAT,primary key(id))ENGINE=InnoDB DEFAULT CHARSET=utf8"""
cursor.execute(sql)
3.2添加
# SQL 插入语句
sql = """INSERT INTO EMPLOYEE(FIRST_NAME,
LAST_NAME, AGE, SEX, INCOME)
VALUES ('Mac', 'Mohan', 20, 'M', 2000)"""
try:
cursor.execute(sql)
# 提交到数据库执行
db.commit()
except:
# 发生错误时回滚
db.rollback()
3.3查询
# SQL 查询语句
sql = "SELECT * FROM EMPLOYEE WHERE INCOME > {}".format(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={},lname={},age={},sex={},income={}".format(fname, lname, age, sex, income))
except:
print("Error: unable to fecth data")
fetchone(): 获取单条数据,返回结果行元组,若无结果,返回null;
rowcount:获取下一个查询结果集,是一个对象;
fetchall():获取多条数据元组;若无结果,返回()。
备注:在MySQL中是NULL,而在Python中则是None。
3.4修改
# SQL 更新语句
sql = "UPDATE EMPLOYEE SET AGE = AGE + 1 WHERE SEX = '{}'".format('M')
try:
cursor.execute(sql)
# 提交到数据库执行
db.commit()
except:
# 发生错误时回滚
db.rollback()
3.5删除
# SQL 刪除语句
sql = "DELETE FROM EMPLOYEE WHERE AGE > {}".format(20)
try:
cursor.execute(sql)
# 提交到数据库执行
db.commit()
except:
# 发生错误时回滚
db.rollback()
4.关闭游标与数据库连接
cursor.close()
db.close()