DevOps系列之 Python操作数据库

pymysql操作mysql数据库

• 安装pymysql

pip install pymysql

pymysql操作数据库

1.连接数据库 使用Connect方法连接数据库 pymysql.Connections.Connection(host=None, user=None, password='', database=None, port=0, charset='') 参数说明： host – 数据库服务器所在的主机。 user – 登录用户名。 password – 登录用户密码。 database – 连接的数据库。 port – 数据库开放的端口。（默认: 3306） charset – 连接字符集。 返回值： 返回连接对象

例子： link = pymysql.Connect(host='localhost', port=3306, user='root', password='123456', db='shop', charset='utf8')

方法	说明
begin()	开启事务
commit()	提交事务
cursor(cursor=None)	创建一个游标用来执行sql语句
rollback()	回滚事务
close()	关闭连接
select_db(db)	选择数据库

• 连接对象方法

2.创建游标

cursor = link.cursor() print(cursor.rowcount) #打印受影响行数

方法	说明
close()	关闭游标
execute(query, args=None)	执行单条语句，传入需要执行的语句，是string类型；同时可以给查询传入参数，参数可以是tuple、list或dict。执行完成后，会返回执行语句的影响行数。
fetchone()	取一条数据
fetchmany(n)	取多条数据
fetchall()	取所有数据

3.执行sql语句

# 执行sql语句

sql = 'select * from user1'

# 执行完sql语句，返回受影响的行数 num = cursor.execute(sql)

4.获取结果集 result1 = cursor.fetchone() print(result1)

5.关闭连接 cursor.close() link.close()

pymysql中事务处理

pymysql默认是没有开启自动提交事务，所以我们如果进行增、删、改，就必须手动提交或回滚事务。

sql = 'delete from user where id=%s' % user_id
​
# 如果要执行增删改语句的时候，下面的就是固定格式
try:
 cursor.execute(sql)
 # 如果全部执行成功，提交事务
 link.commit()
 print(cursor.lastrowid) #获取最后插入记录的自增id号
except Exception as e:
 print(e)
 link.rollback()
finally:
 cursor.close()
 link.close()

案例

使用pymysql向goods表中添加一条数据：

from pymysql import *
 
​
def main():
  # 创建connection连接
  conn = connect(host='localhost', port=3306, database='shop', user='root',
          password='root', charset='utf8')
  # 获取cursor对象
  cs1 = conn.cursor()
  # 执行sql语句
  query = "insert into goods(id,name,price,num) values(%s,%s,%s,%s)"
  cs1.execute(query,(4,'蒙牛酸奶',13.9,88))
 
  # 提交之前的操作，如果之前已经执行多次的execute，那么就都进行提交
  conn.commit()
 
  # 关闭cursor对象
  cs1.close()
  # 关闭connection对象
  conn.close()
 
 
if __name__ == '__main__':
  main()

当然也可以删除、查询、修改表中的数据，但是无论是怎么操作，都需要创建连接并在结束的时候关闭连接对象。