Python--操作mysql

import pymysql

# 1.连接mysql ip 端口号 密码 账号 数据库 charset='utf8'
# 2.建立游标
# 3.执行sql
# 4.获取结果
# 5.关闭连接，关闭游标

# charset 必须填写utf8  port一定要写int类型
conn = pymysql.connect(host='192.168.1.20', port=3306, user='root', passwd='123456', db='test', charset='utf8')
# cur = conn.cursor()  # 建立游标，游标认为是仓库管理员
cur = conn.cursor(cursor=pymysql.cursors.DictCursor)  # 需要指定游标的类型，字典类型
# cur.execute('select * from student;')  # 执行sql语句
cur.execute('select * from student;')
# update delete insert 需要提交才会生效
# cur.execute('insert into student (username, password) values (%s,%s)', ('wangsilei3', '123456'))
# conn.commit()  # 提交
# res = cur.fetchall()  # 获取sql语句执行结果,他把结果放到一个元组里面，每一条数据也是一个元组
res = cur.fetchone()  # 只获取一条结果，它的结果是一个一维的元组
print(res)

# cur.scroll(0, mode='absolute')  # 移动游标
# cur.scroll(1, mode='relative')  # 相对当前位置移动 -1往前移动
# cur.scroll(2, mode='absolute')  # 相对绝对位置移动
# res = cur.fetchone()  # 获取第二条数据
# print(res)
cur.close()  # 关闭游标
conn.close()  # 关闭连接


封装操作数据库方法

import pymysql


def op_mysql(host, user, password, db, sql, port=3306, charset='uft8'):
    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)
    sql_start = sql[:6].upper()  # 取sql前六个字符串，判断他是什么类型的sql语句
    if sql_start == 'SELECT':
        res = cur.fetchall()
    else:
        conn.commit()
        res = 'OK'
    cur.close()
    conn.close()
    return res


# 别人导入该py文件时，下面代码不会被执行
# 一般自己测试的时候写
if __name__ == '__main__':
    sql1 = 'select * from bt_stu limit 5'
    res_sql = op_mysql(
        host='211.149.218.16',
        user='jxz', password='123456',  # port这里一定要写int类型
        port=3306, db='jxz', charset='utf8', sql=sql1)
    print(res_sql)