python 之操作mysql 数据库实例

时间:2021-06-01 19:17:56

对于python操作mysql 数据库,具体的步骤应为:

1. 连接上mysql host 端口号 数据库 账号 密码
2. 建立游标
3. 执行sql(注意,如果是update,insert,delete 需要进行提交才能生效。)
4. 获取结果
5. 关闭连接、关闭游标

一、默认获取的结果是元祖

 conn = pymysql.connect(host='localhost',user='root',passwd='',port=3306,db='sakila',charset='utf8')
#charset 必须写utf8, 不能写utf-8; port 要写int 类型,不能加上引号
cur = conn.cursor() # 建立游标,游标认为你是仓库管理员
cur.execute('select * from user limit 10;')
res = cur.fetchone() # 只获取一条结果,它的结果是一个一维元祖,第一次获取到第一条
##print('fetchone',cur.fetchone()) # 第二次获取到的是第二行;
print(res)
print('fetchall',cur.fetchall())# 获取第二条开始的所有数据,获取sql语句执行的结果, 是个二维元祖,它把结果放到一个元祖里,每一条数据也是一个元祖
print(res)
print(res[0][1])
print('fetchone',cur.fetchone()) # #被获取完了,所以获取到的是None
# 如果确定只有一条数据的用fetchone, 超过一条数据的那就用fetchall

二、将获取的结果转字典,方便使用

 conn = pymysql.connect(host='localhost',user='root',passwd='',port=3306,db='sakila',charset='utf8')
cur = conn.cursor(cursor=pymysql.cursors.DictCursor) # 将元祖转为字典
sql = "select * from user where name='aa'"
cur.execute(sql)
print(cur.fetchone()) #获取字典
res = cur.fetchall()
print(res) #list 里存字典
cur.close()#关闭游标
conn.close()#关闭连接

在python 中会经常用到mysql,因此可以写了一个函数用于mysql的操作,要使用时调用以下即可。

 import pymysql
def op_mysql(host,user,password,db,sql,port=3306,charset='utf8'):
conn = pymysql.connect(host=host,user=user,
password=password,
db=db,
port=port,
charset=charset)
cur = conn.cursor(cursor=pymysql.cursors.DictCursor)
cur.execute(sql)
# select, update, delete, insert 长度都是6位
# SELECT
sql_start = sql[:6].upper() # sql 的前6位字符串,判断它是什么类型的sql语句
if sql_start == 'SELECT':
res = cur.fetchall()
else:
conn.commit()
res = 'ok'
cur.close()
conn.close()
return res
sql = 'select * from user limit 5;'
op_mysql(
host='localhost',
user='root',
password='',
port=3306,
db='sakila',
charset='utf8',
sql=sql)