python操作mysql

python操作mysql

安装

python操作mysql数据库，主要就是通过pymysql模块

pip install pymysql

操作流程

1）建立数据库连接对象 conn

2）通过 conn 创建操作sql的 游标对象

3）编写sql交给 cursor 执行

4）如果是查询，通过 cursor对象 获取结果

5）操作完毕，端口操作与连接

代码步骤

注意

1.对记录增删改默认需要commit()

准备

import pymysql

一.建立连接

conn = pymysql.connect(user='root', passwd='root', database='t5')

二.获取游标对象

# 注：游标不设置参数，查询的结果就是数据元组，数据没有标识性
# 设置pymysql.cursors.DictCursor，查询的结果是字典，key是表的字段cursor = conn.cursor(pymysql.cursors.DictCursor)

三.增删改查

创建表

sql1="create table zx(id int)"
cursor.execute(sql1)

增

#单条添加
sql2='insert into zx values(%s)'
cursor.execute(sql2,(1,))
cursor.execute(sql2,(2,))
cursor.execute(sql2,(3,))
conn.commit()
#多条添加
cursor.executemany(sql2, [(4,), (5,)])
conn.commit()

删

sql3="delete from zx where id =%s"
cursor.execute(sql3,3)
conn.commit()

改

sql4='update zx set id=10 where id =2'
cursor.execute(sql4)
conn.commit()

查

sql5='select * from zx'
cursor.execute(sql5)

#fetchone() 偏移一条取出
r1=cursor.fetchone()
print(r1)

#fetchmany(n) 偏移n条取出
r2=cursor.fetchmany(1)
print(r2)

#fetchall() 偏移剩余全部
r3=cursor.fetchall()
print(r3)

结果

{'id': 1}
[{'id': 10}]
[{'id': 4}, {'id': 5}]

结束

#释放端口与连接
cursor.close()
conn.close()

游标其他用法

 # 操作游标
    # cursor.scroll(0, 'absolute')  # absolute绝对偏移，游标重置，从头开始偏移
    cursor.scroll(-2, 'relative')  # relative相对偏移，游标在当前位置进行左右偏移

事务

#转账案例
try:
    sql1 = 'update t2 set money=money-1 where name="tom"'
    r1 = cursor.execute(sql1)
    sql2 = 'update t2 set money=money+1 where name="ruakei"'  # 转入的人不存在
    r2 = cursor.execute(sql2)
except:
    print('转账执行异常')
    conn.rollback()
else:
    print('转账没有异常')
    if r1 == 1 and r2 == 1:
        print('转账成功')
        conn.commit()
    else:
        conn.rollback()

sql注入攻击

不要自己拼接参数，交给pymysql占位填充！

sql = 'select * from user where name=%s and password=%s'
row = cursor.execute(sql, (usr, pwd))
if row:
    print('登录成功')
else:
    print('登录失败')