python 之操作mysql 数据库实例

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

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

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

conn = pymysql.connect(host='localhost',user='root',passwd='123456',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='123456',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='123456',
        port=3306,
        db='sakila',
        charset='utf8',
        sql=sql)