python3 操作mysql数据库（mysql.connector 和 pymysql ）

左手的ㄟ右手 2021-07-28 12:19 694阅读 0赞

1. PyMySQL 的使用

(1) 什么是 PyMySQL？

  PyMySQL 是 Python 中用于连接 MySQL 服务器的一个库，它遵循 Python 数据库 API 规范 V2.0，并包含了 pure-Python MySQL 客户端库。

(2) 安装 PyMysql:

  1. pip install PyMysql

(3) 参数说明:

  1. pymysql.Connect()参数说明
  2. host(str):      MySQL服务器地址
  3. port(int):      MySQL服务器端口号
  4. user(str):      用户名
  5. passwd(str):    密码
  6. db(str):        数据库名称
  7. charset(str):   连接编码
  9. connection对象支持的方法
  10. cursor()        使用该连接创建并返回游标
  11. commit()        提交当前事务
  12. rollback()      回滚当前事务
  13. close()         关闭连接
  15. cursor对象支持的方法
  16. execute(op)     执行一个数据库的查询命令
  17. fetchone()      取得结果集的下一行
  18. fetchmany(size) 获取结果集的下几行
  19. fetchall()      获取结果集中的所有行
  20. rowcount()      返回数据条数或影响行数
  21. close()         关闭游标对象

(4) 使用 PyMySQL:

  1. #!/usr/bin/env python
  2. # coding=utf-8
  4. import pymysql
  6. def connectdb():
  7.     print('连接到mysql服务器...')
  8.     # 打开数据库连接
  9.     # 用户名:hp, 密码:Hp12345.,用户名和密码需要改成你自己的mysql用户名和密码，并且要创建数据库TESTDB，并在TESTDB数据库中创建好表Student
  10.     db = pymysql.connect("localhost","hp","Hp12345.","TESTDB")
  11.     print('连接上了!')
  12.     return db
  14. def createtable(db):
  15.     # 使用cursor()方法获取操作游标 
  16.     cursor = db.cursor()
  18.     # 如果存在表Sutdent先删除
  19.     cursor.execute("DROP TABLE IF EXISTS Student")
  20.     sql = """CREATE TABLE Student (
  21.             ID CHAR(10) NOT NULL,
  22.             Name CHAR(8),
  23.             Grade INT )"""
  25.     # 创建Sutdent表
  26.     cursor.execute(sql)
  28. def insertdb(db):
  29.     # 使用cursor()方法获取操作游标 
  30.     cursor = db.cursor()
  32.     # SQL 插入语句
  33.     sql = """INSERT INTO Student
  34.          VALUES ('001', 'CZQ', 70),
  35.                 ('002', 'LHQ', 80),
  36.                 ('003', 'MQ', 90),
  37.                 ('004', 'WH', 80),
  38.                 ('005', 'HP', 70),
  39.                 ('006', 'YF', 66),
  40.                 ('007', 'TEST', 100)"""
  42.     #sql = "INSERT INTO Student(ID, Name, Grade) \
  43.     #    VALUES ('%s', '%s', '%d')" % \
  44.     #    ('001', 'HP', 60)
  45.     try:
  46.         # 执行sql语句
  47.         cursor.execute(sql)
  48.         # 提交到数据库执行
  49.         db.commit()
  50.     except:
  51.         # Rollback in case there is any error
  52.         print '插入数据失败!'
  53.         db.rollback()
  55. def querydb(db):
  56.     # 使用cursor()方法获取操作游标 
  57.     cursor = db.cursor()
  59.     # SQL 查询语句
  60.     #sql = "SELECT * FROM Student \
  61.     #    WHERE Grade > '%d'" % (80)
  62.     sql = "SELECT * FROM Student"
  63.     try:
  64.         # 执行SQL语句
  65.         cursor.execute(sql)
  66.         # 获取所有记录列表
  67.         results = cursor.fetchall()
  68.         for row in results:
  69.             ID = row[0]
  70.             Name = row[1]
  71.             Grade = row[2]
  72.             # 打印结果
  73.             print "ID: %s, Name: %s, Grade: %d" % \
  74.                 (ID, Name, Grade)
  75.     except:
  76.         print "Error: unable to fecth data"
  78. def deletedb(db):
  79.     # 使用cursor()方法获取操作游标 
  80.     cursor = db.cursor()
  82.     # SQL 删除语句
  83.     sql = "DELETE FROM Student WHERE Grade = '%d'" % (100)
  85.     try:
  86.        # 执行SQL语句
  87.        cursor.execute(sql)
  88.        # 提交修改
  89.        db.commit()
  90.     except:
  91.         print '删除数据失败!'
  92.         # 发生错误时回滚
  93.         db.rollback()
  95. def updatedb(db):
  96.     # 使用cursor()方法获取操作游标 
  97.     cursor = db.cursor()
  99.     # SQL 更新语句
  100.     sql = "UPDATE Student SET Grade = Grade + 3 WHERE ID = '%s'" % ('003')
  102.     try:
  103.         # 执行SQL语句
  104.         cursor.execute(sql)
  105.         # 提交到数据库执行
  106.         db.commit()
  107.     except:
  108.         print '更新数据失败!'
  109.         # 发生错误时回滚
  110.         db.rollback()
  112. def closedb(db):
  113.     db.close()
  115. def main():
  116.     db = connectdb()    # 连接MySQL数据库
  118.     createtable(db)     # 创建表
  119.     insertdb(db)        # 插入数据
  120.     print '\n插入数据后:'
  121.     querydb(db) 
  122.     deletedb(db)        # 删除数据
  123.     print '\n删除数据后:'
  124.     querydb(db)
  125.     updatedb(db)        # 更新数据
  126.     print '\n更新数据后:'
  127.     querydb(db)
  129.     closedb(db)         # 关闭数据库
  131. if __name__ == '__main__':
  132.     main()

运行结果:

watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3doYXRkYXk_size_16_color_FFFFFF_t_70

实例2：

  1. import pymysql.cursors
  3. # 连接数据库
  4. connect = pymysql.connect(
  5.     host='localhost',
  6.     port=3310,
  7.     user='woider',
  8.     passwd='3243',
  9.     db='python',
  10.     charset='utf8'
  11. )
  13. # 获取游标
  14. cursor = connect.cursor()
  16. # 插入数据
  17. sql = "INSERT INTO trade (name, account, saving) VALUES ( '%s', '%s', %.2f )"
  18. data = ('雷军', '13512345678', 10000)
  19. cursor.execute(sql % data)
  20. connect.commit()
  21. print('成功插入', cursor.rowcount, '条数据')
  23. # 修改数据
  24. sql = "UPDATE trade SET saving = %.2f WHERE account = '%s' "
  25. data = (8888, '13512345678')
  26. cursor.execute(sql % data)
  27. connect.commit()
  28. print('成功修改', cursor.rowcount, '条数据')
  30. # 查询数据
  31. sql = "SELECT name,saving FROM trade WHERE account = '%s' "
  32. data = ('13512345678',)
  33. cursor.execute(sql % data)
  34. for row in cursor.fetchall():
  35.     print("Name:%s\tSaving:%.2f" % row)
  36. print('共查找出', cursor.rowcount, '条数据')
  38. # 删除数据
  39. sql = "DELETE FROM trade WHERE account = '%s' LIMIT %d"
  40. data = ('13512345678', 1)
  41. cursor.execute(sql % data)
  42. connect.commit()
  43. print('成功删除', cursor.rowcount, '条数据')
  45. # 事务处理
  46. sql_1 = "UPDATE trade SET saving = saving + 1000 WHERE account = '18012345678' "
  47. sql_2 = "UPDATE trade SET expend = expend + 1000 WHERE account = '18012345678' "
  48. sql_3 = "UPDATE trade SET income = income + 2000 WHERE account = '18012345678' "
  50. try:
  51.     cursor.execute(sql_1)  # 储蓄增加1000
  52.     cursor.execute(sql_2)  # 支出增加1000
  53.     cursor.execute(sql_3)  # 收入增加2000
  54. except Exception as e:
  55.     connect.rollback()  # 事务回滚
  56.     print('事务处理失败', e)
  57. else:
  58.     connect.commit()  # 事务提交
  59.     print('事务处理成功', cursor.rowcount)
  61. # 关闭连接
  62. cursor.close()
  63. connect.close()

2. mysql.connector 的使用

(1) 什么是 mysql.connector？

  由于 MySQL 服务器以独立的进程运行，并通过网络对外服务，所以，需要支持 Python 的 MySQL 驱动来连接到 MySQL 服务器。

目前，有两个 MySQL 驱动：

mysql-connector-python：是 MySQL 官方的纯 Python 驱动；

MySQL-python ：是封装了 MySQL C驱动的 Python 驱动。

(2) 安装 mysql.connector:

  1. pip install mysql-connector-python
  2. pip install MySQL-python

(3) 使用 mysql.connector:

  1. #!/usr/bin/env python
  2. # coding=utf-8
  4. import mysql.connector
  6. def connectdb():
  7.     print('连接到mysql服务器...')
  8.     # 打开数据库连接
  9.     # 用户名:hp, 密码:Hp12345.,用户名和密码需要改成你自己的mysql用户名和密码，并且要创建数据库TESTDB，并在TESTDB数据库中创建好表Student
  10.     db = mysql.connector.connect(user="hp", passwd="Hp12345.", database="TESTDB", use_unicode=True)
  11.     print('连接上了!')
  12.     return db
  14. def createtable(db):
  15.     # 使用cursor()方法获取操作游标 
  16.     cursor = db.cursor()
  18.     # 如果存在表Sutdent先删除
  19.     cursor.execute("DROP TABLE IF EXISTS Student")
  20.     sql = """CREATE TABLE Student (
  21.             ID CHAR(10) NOT NULL,
  22.             Name CHAR(8),
  23.             Grade INT )"""
  25.     # 创建Sutdent表
  26.     cursor.execute(sql)
  28. def insertdb(db):
  29.     # 使用cursor()方法获取操作游标 
  30.     cursor = db.cursor()
  32.     # SQL 插入语句
  33.     sql = """INSERT INTO Student
  34.          VALUES ('001', 'CZQ', 70),
  35.                 ('002', 'LHQ', 80),
  36.                 ('003', 'MQ', 90),
  37.                 ('004', 'WH', 80),
  38.                 ('005', 'HP', 70),
  39.                 ('006', 'YF', 66),
  40.                 ('007', 'TEST', 100)"""
  42.     #sql = "INSERT INTO Student(ID, Name, Grade) \
  43.     #    VALUES ('%s', '%s', '%d')" % \
  44.     #    ('001', 'HP', 60)
  45.     try:
  46.         # 执行sql语句
  47.         cursor.execute(sql)
  48.         # 提交到数据库执行
  49.         db.commit()
  50.     except:
  51.         # Rollback in case there is any error
  52.         print '插入数据失败!'
  53.         db.rollback()
  55. def querydb(db):
  56.     # 使用cursor()方法获取操作游标 
  57.     cursor = db.cursor()
  59.     # SQL 查询语句
  60.     #sql = "SELECT * FROM Student \
  61.     #    WHERE Grade > '%d'" % (80)
  62.     sql = "SELECT * FROM Student"
  63.     try:
  64.         # 执行SQL语句
  65.         cursor.execute(sql)
  66.         # 获取所有记录列表
  67.         results = cursor.fetchall()
  68.         for row in results:
  69.             ID = row[0]
  70.             Name = row[1]
  71.             Grade = row[2]
  72.             # 打印结果
  73.             print "ID: %s, Name: %s, Grade: %d" % \
  74.                 (ID, Name, Grade)
  75.     except:
  76.         print "Error: unable to fecth data"
  78. def deletedb(db):
  79.     # 使用cursor()方法获取操作游标 
  80.     cursor = db.cursor()
  82.     # SQL 删除语句
  83.     sql = "DELETE FROM Student WHERE Grade = '%d'" % (100)
  85.     try:
  86.        # 执行SQL语句
  87.        cursor.execute(sql)
  88.        # 提交修改
  89.        db.commit()
  90.     except:
  91.         print '删除数据失败!'
  92.         # 发生错误时回滚
  93.         db.rollback()
  95. def updatedb(db):
  96.     # 使用cursor()方法获取操作游标 
  97.     cursor = db.cursor()
  99.     # SQL 更新语句
  100.     sql = "UPDATE Student SET Grade = Grade + 3 WHERE ID = '%s'" % ('003')
  102.     try:
  103.         # 执行SQL语句
  104.         cursor.execute(sql)
  105.         # 提交到数据库执行
  106.         db.commit()
  107.     except:
  108.         print '更新数据失败!'
  109.         # 发生错误时回滚
  110.         db.rollback()
  112. def closedb(db):
  113.     db.close()
  115. def main():
  116.     db = connectdb()    # 连接MySQL数据库
  118.     createtable(db)     # 创建表
  119.     insertdb(db)        # 插入数据
  120.     print '\n插入数据后:'
  121.     querydb(db) 
  122.     deletedb(db)        # 删除数据
  123.     print '\n删除数据后:'
  124.     querydb(db)
  125.     updatedb(db)        # 更新数据
  126.     print '\n更新数据后:'
  127.     querydb(db)
  129.     closedb(db)         # 关闭数据库
  131. if __name__ == '__main__':
  132.     main()

运行结果:

watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3doYXRkYXk_size_16_color_FFFFFF_t_70 1

发表评论

表情：
评论列表 （有 0 条评论，694人围观）

还没有评论，来说两句吧...

相关阅读