import pymysql.cursors
# 连接数据库
connect = pymysql.Connect(
host='localhost',
port=3306,
user='root',
passwd='magicimage',
db='student',
charset='utf8'
)
# 获取游标
cursor = connect.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 = "INSERT INTO student (name, sex, class_id) VALUES ( '%s', '%s', '%s' )"
data = ('雷军', 0, 1)
try:
# 执行sql语句
cursor.execute(sql % data)
connect.commit()
print('成功插入', cursor.rowcount, '条数据')
except:
connect.rollback()
'''
显示结果
成功插入 1 条数据
'''
'''
数据库查询操作
fetchone() 方法获取单条数据,
fetchall() 方法获取多条数据。
rowcount: 这是一个只读属性,并返回执行execute()方法后影响的行数。
'''
# SQL 查询语句
sql = "SELECT * FROM student \
WHERE class_id > %s" % (1)
try:
# 执行SQL语句
cursor.execute(sql)
# 获取所有记录列表
results = cursor.fetchall()
for row in results:
id = row[0]
name = row[1]
sex = row[2]
class_id = row[3]
# 打印结果
print ("id=%s,name=%s,sex=%s,class_id=%s" % \
(id, name, sex,class_id ))
except:
print ("Error: unable to fetch data")
'''
结果
id=2,name=雪,sex=1,class_id=2
id=3,name=于,sex=0,class_id=3
id=5,name=孙,sex=0,class_id=2
'''
'''
数据库更新操作
'''
# SQL 更新语句
sql = "UPDATE student SET name = '雷' WHERE id = '%s'" % (7)
try:
# 执行SQL语句
cursor.execute(sql)
# 提交到数据库执行
connect.commit()
print('成功更新', cursor.rowcount, '条数据')
except:
# 发生错误时回滚
connect.rollback()
'''
删除操作
'''
# SQL 删除语句
sql = "DELETE FROM student WHERE id = '%s'" % (7)
try:
# 执行SQL语句
cursor.execute(sql)
# 提交修改
connect.commit()
print('成功删除', cursor.rowcount, '条数据')
except:
# 发生错误时回滚
connect.rollback()
# 关闭数据库连接
connect.close()