利用Python操作MySQL数据库

上一篇 / 下一篇  2020-02-27 13:20:30

前言
  在工作中,我们需要经常对数据库进行操作,比如OracleMySQLSQLSever 等,今天我们就学习如何利用Python来操作 MySQL 数据库。
  本人环境:Python 3.7.0 、MySQL 5.7
  安装第三方库
  使用Python来操作MySQL,需要用到 PyMySQL 这个第三方库,具体安装如下:
  pip install PyMySQL
  我这里安装的版本是 0.9.3。
  
  基本操作-增删改查
  Python操作MySQL的步骤如下:
  建立数据库连接
  通过 cursor() 创建游标对象(pymysql通过游标来执行sql和获取结果)
  使用 execute() 执行sql
  获取结果(查询)/提交事务(增删改)
  关闭游标
  关闭连接
  注意:用Python进行增删改查操作之前,最好先检查下sql语句是否正确,确保没有错误之后,再放到Python代码中。
 
  查询操作
   import pymysql
  def select_db(select_sql):
  """查询"""
  # 建立数据库连接
  db = pymysql.connect(
  host="192.168.89.128",
  port=3306,
  user="root",
  passwd="123456",
  db="test2020"
  )
  # 通过 cursor() 创建游标对象,并让查询结果以字典格式输出
  cur = db.cursor(cursor=pymysql.cursors.DictCursor)
  # 使用 execute() 执行sql
  cur.execute(select_sql)
  # 使用 fetchall() 获取所有查询结果
  data = cur.fetchall()
  # 关闭游标
  cur.close()
  # 关闭数据库连接
  db.close()
  return data
  select_sql = 'SELECT * FROM user WHERE username="张三"'
  print(select_db(select_sql))
  在上面,创建游标时使用了 cursor=pymysql.cursors.DictCursor ,目的是为了让查询结果以 列表嵌套字典 的格式输出,以方便后续操作。如果不使用该参数,那么返回的查询结果将是 元组嵌套元组 的形式。
  更新操作
   import pymysql
  def update_db(update_sql):
  """更新"""
  # 建立数据库连接
  db = pymysql.connect(
  host="192.168.89.128",
  port=3306,
  user="root",
  passwd="123456",
  db="test2020"
  )
  # 通过 cursor() 创建游标对象
  cur = db.cursor()
  try:
  # 使用 execute() 执行sql
  cur.execute(update_sql)
  # 提交事务
  db.commit()
  except Exception as e:
  print("操作出现错误:{}".format(e))
  # 回滚所有更改
  db.rollback()
  finally:
  # 关闭游标
  cur.close()
  # 关闭数据库连接
  db.close()
  update_sql = 'UPDATE user SET username = "张三2" WHERE id = 1'
  update_db(update_sql)
  在执行更新操作sql时,需要使用事务处理,即使用创建的连接进行提交事务 db.commit() ,使当前更改生效。当执行sql语句出现错误时,还应该支持事务回滚操作 db.rollback() 。
  新增操作
   import pymysql
  def insert_db(insert_sql):
  """插入"""
  # 建立数据库连接
  db = pymysql.connect(
  host="192.168.89.128",
  port=3306,
  user="root",
  passwd="123456",
  db="test2020"
  )
  # 通过 cursor() 创建游标对象
  cur = db.cursor()
  try:
  # 使用 execute() 执行sql
  cur.execute(insert_sql)
  # 提交事务
  db.commit()
  except Exception as e:
  print("操作出现错误:{}".format(e))
  # 回滚所有更改
  db.rollback()
  finally:
  # 关闭游标
  cur.close()
  # 关闭数据库连接
  db.close()
  insert_sql = 'INSERT INTO user(id, username, password) VALUES(11, "王五", "333333")'
  insert_db(insert_sql)
  在执行新增操作sql时,需要像 更新操作 一样,使用事务来处理。
  删除操作
   import pymysql
  def delete_db(delete_sql):
  """删除"""
  # 建立数据库连接
  db = pymysql.connect(
  host="192.168.89.128",
  port=3306,
  user="root",
  passwd="123456",
  db="test2020"
  )
  # 通过 cursor() 创建游标对象
  cur = db.cursor()
  try:
  # 使用 execute() 执行sql
  cur.execute(delete_sql)
  # 提交事务
  db.commit()
  except Exception as e:
  print("操作出现错误:{}".format(e))
  # 回滚所有更改
  db.rollback()
  finally:
  # 关闭游标
  cur.close()
  # 关闭数据库连接
  db.close()
  delete_sql = 'DELETE FROM user WHERE id = 11'
  delete_db(delete_sql)
  在执行删除操作sql时,需要像 更新操作、插入操作 一样,使用事务来处理。
  简单封装数据库操作
  从上面的代码中,我们可以发现存在不少问题,比如增删改操作的代码除了执行的sql不同外,其他代码都是一样的;同时每个操作时都写了一次建立数据库连接的代码,也就是说每次操作都会建立和删除连接的操作。
  针对上面的问题,我们完全可以对上面的增删改查操作进行简单的封装,把Python操作MySQL的代码封装到一个类中,只在创建对象实例的时候建立数据库连接,只在对象删除时关闭连接。
  以下是对数据库操作的简单封装:
   import pymysql
  class MysqlDb():
  def __init__(self, host, port, user, passwd, db):
  # 建立数据库连接
  self.conn = pymysql.connect(
  host=host,
  port=port,
  user=user,
  passwd=passwd,
  db=db
  )
  # 通过 cursor() 创建游标对象,并让查询结果以字典格式输出
  self.cur = self.conn.cursor(cursor=pymysql.cursors.DictCursor)
  def __del__(self): # 对象资源被释放时触发,在对象即将被删除时的最后操作
  # 关闭游标
  self.cur.close()
  # 关闭数据库连接
  self.conn.close()
  def select_db(self, sql):
  """查询"""
  # 使用 execute() 执行sql
  self.cur.execute(sql)
  # 使用 fetchall() 获取查询结果
  data = self.cur.fetchall()
  return data
  def execute_db(self, sql):
  """更新/插入/删除"""
  try:
  # 使用 execute() 执行sql
  self.cur.execute(sql)
  # 提交事务
  self.conn.commit()
  except Exception as e:
  print("操作出现错误:{}".format(e))
  # 回滚所有更改
  self.conn.rollback()
  if __name__ == '__main__':
  db = MysqlDb("192.168.89.128", 3306, "root", "123456", "test2020")
  select_sql = 'SELECT * FROM user WHERE username="张三2"'
  update_sql = 'UPDATE user SET username = "张三2" WHERE id = 1'
  insert_sql = 'INSERT INTO user(id, username, password) VALUES(11, "王五", "333333")'
  delete_sql = 'DELETE FROM user WHERE id = 11'
  data = db.select_db(select_sql)
  print(data)
  db.execute_db(update_sql)
  db.execute_db(insert_sql)
  db.execute_db(delete_sql)


TAG:

 

评分:0

我来说两句

Open Toolbar