PyMySQL 是在 Python3.x 版本中用于连接 MySQL 服务器的一个库。
安装:
代码语言:javascript复制pip install PyMySQL
代码语言:javascript复制数据库增删改查流程差不多,主要在于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"�ta)
#执行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()