PyMySQL模块的使用

系统管理员 2021-04-17 17:05 504阅读 0赞

# 快速使用 #

[https://github.com/PyMySQL/PyMySQL][https_github.com_PyMySQL_PyMySQL]

## 官方教程 ##

import pymysql.cursors
    
    # 连接数据库
    connection = pymysql.connect(host='localhost',
                                 user='user',
                                 password='passwd',
                                 db='db',
                                 charset='utf8mb4',
                                 cursorclass=pymysql.cursors.DictCursor)
    
    try:
        with connection.cursor() as cursor:
            # 增加一条数据
            sql = "INSERT INTO `users` (`email`, `password`) VALUES (%s, %s)"
            cursor.execute(sql, ('webmaster@python.org', 'very-secret'))
    
        connection.commit()
    
        with connection.cursor() as cursor:
            # 查询数据
            sql = "SELECT `id`, `password` FROM `users` WHERE `email`=%s"
            cursor.execute(sql, ('webmaster@python.org',))
            result = cursor.fetchone()
            print(result)
    finally:
        connection.close()

## 备查 ##

# 获取输入SQL语句的光标对象，指定返回的数据格式为字典格式
    cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
    
    # 执行sql语句
    cursor.execute(sql, (username, pwd))
    
    # 提交事务
    conn.commit()
    
    # 获取单条查询数据
    ret = cursor.fetchone()
    
    # 返回所有数据
    ret = cursor.fetchall()
    
    # 回滚
    conn.rollback()

# 基本使用 #

# 导入模块
    import pymysql
    
    username = input("请输入用户名: ")
    pwd = input("请输入密码: ")
    
    # 连接数据库
    conn = pymysql.connect(host="localhost",
                           port=3306,
                           database="userinfo",
                           user='root',
                           password="123456",
                           charset="utf8"
                           )
    
    # 获取输入SQL语句的光标对象
    cursor = conn.cursor()
    
    # 定义要执行的SQL语句
    sql = "select * from info WHERE username=%s and password=%s;"
    
    # 执行sql语句，让pymysql帮我们拼接SQL语句
    cursor.execute(sql, (username, pwd))
    
    # 关闭光标对象
    cursor.close()
    
    # 关闭数据库连接
    conn.close()

**返回字典格式的数据**

# 获取输入SQL语句的光标对象，指定返回的数据格式为字典格式
    cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)

# 增删改查操作 #

## 增 ##

import pymysql
    
    # 获取用户输入
    username = input("请输入用户名: ")
    pwd = input("请输入密码: ")
    
    # 连接数据库
    conn = pymysql.connect(host="localhost",
                           port=3306,
                           database="userinfo",
                           user='root',
                           password="123456",
                           charset="utf8"
                           )
    
    # 获取输入SQL语句的光标对象
    cursor = conn.cursor()
    
    # sql语句
    sql = "insert into info (username, password) VALUES (%s, %s);"
    
    # 执行sql语句
    cursor.execute(sql, [username, pwd])
    
    # 提交
    conn.commit()
    
    # 关闭光标对象和连接
    cursor.close()
    conn.close()

**插入数据失败回滚**  
在执行增删改操作时，如果不想提交前面的操作，可以使用 rollback() 回滚取消操作。

import pymysql
    
    # 获取用户输入
    username = input("请输入用户名: ")
    pwd = input("请输入密码: ")
    
    # 连接数据库
    conn = pymysql.connect(host="localhost",
                           port=3306,
                           database="userinfo",
                           user='root',
                           password="123456",
                           charset="utf8"
                           )
    
    # 获取输入SQL语句的光标对象
    cursor = conn.cursor()
    
    # 拼接sql语句
    sql = "insert into info (username, password) VALUES (%s, %s);"
    
    # 执行sql语句
    # 让pymysql帮我们拼接SQL语句(传入的参数可以是列表也可以是元祖，一定要是一个可迭代对象)
    
    try:
        cursor.execute(sql, [username, pwd])
        # 提交
        conn.commit()
    except Exception as e:
        print("报错啦：", str(e))
        conn.rollback() # 回滚
    
    # 关闭光标对象和连接
    cursor.close()
    conn.close()

**获取插入数据的ID(关联操作时会用到)**

import pymysql
    
    # 获取用户输入
    username = input("请输入用户名: ")
    pwd = input("请输入密码: ")
    
    # 连接数据库
    conn = pymysql.connect(host="localhost",
                           port=3306,
                           database="userinfo",
                           user='root',
                           password="123456",
                           charset="utf8"
                           )
    
    # 获取输入SQL语句的光标对象
    cursor = conn.cursor()
    
    # 拼接sql语句
    sql = "insert into info (username, password) VALUES (%s,%s);"
    
    # 执行sql语句
    cursor.execute(sql, [username, pwd])
    
    # 提交
    conn.commit()
    
    # 提交之后，获取刚插入的数据的ID
    last_id = cursor.lastrowid
    
    # 关闭光标对象和连接
    cursor.close()
    conn.close()

**批量执行**

import pymysql
    
    # 连接数据库
    conn = pymysql.connect(host="localhost",
                           port=3306,
                           database="userinfo",
                           user='root',
                           password="123456",
                           charset="utf8"
                           )
    
    # 获取光标对象
    cursor = conn.cursor()
    
    data = [("奔驰哥", 123456), ("小宝宝", 123456), ("教主", 123456)]
    
    # sql语句
    sql = "insert into info(username, password) VALUES (%s, %s)"
    
    # 批量插入
    cursor.executemany(sql, data)
    
    # 提交
    conn.commit()
    
    # 关闭光标对象和连接
    cursor.close()
    conn.close()

## 删 ##

import pymysql
    
    # 连接数据库
    conn = pymysql.connect(host="localhost",
                           port=3306,
                           database="userinfo",
                           user='root',
                           password="123456",
                           charset="utf8"
                           )
    
    # 获取光标对象
    cursor = conn.cursor()
    
    # sql语句
    sql = "delete from info where id=%s"
    
    # 执行SQL语句
    cursor.execute(sql, [1])
    
    # 提交事务
    conn.commit()
    
    # 关闭光标对象和连接
    cursor.close()
    conn.close()

## 改 ##

import pymysql
    
    # 连接数据库
    conn = pymysql.connect(host="localhost",
                           port=3306,
                           database="userinfo",
                           user='root',
                           password="123456",
                           charset="utf8"
                           )
    
    # 获取输入SQL语句的光标对象
    cursor = conn.cursor()
    
    # sql语句
    sql = "update info set password=%s where id=%s"
    
    password = "654321"
    id = 2
    
    cursor.execute(sql, [password, id])
    conn.commit()
    
    cursor.close()
    conn.close()

## 查 ##

**查询单条数据**

import pymysql
    
    # 连接数据库
    conn = pymysql.connect(host="localhost",
                           port=3306,
                           database="userinfo",
                           user='root',
                           password="123456",
                           charset="utf8"
                           )
    
    # 获取输入SQL语句的光标对象，指定返回的数据格式为字典格式
    cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
    
    sql = "select * from info"
    
    # 提交事务
    cursor.execute(sql)
    
    # 获取单条查询数据
    ret = cursor.fetchone()
    
    # 返回所有数据
    ret = cursor.fetchall() 
    
    # 关闭光标和连接
    cursor.close()
    conn.close()

# 进阶用法 #

查询后光标就会移动，下次查询就会再上次查询的数据之后开始新的查询

# 可以获取指定数量的数据
    cursor.fetchmany(3)
    
    # 光标按绝对位置移动1
    cursor.scroll(1, mode="absolute")
    
    # 光标按照相对位置(当前位置)移动1
    cursor.scroll(1, mode="relative")

# 错误处理 #

| 异常              | 描述                                                         |
| ----------------- | ------------------------------------------------------------ |
| Warning           | 当有严重警告时触发，例如插入数据是被截断等等。               |
| Error             | 警告以外所有其他错误类。                                     |
| InterfaceError    | 当有数据库接口模块本身的错误（而不是数据库的错误）发生时触发。 |
| DatabaseError     | 和数据库有关的错误发生时触发。                               |
| DataError         | 当有数据处理时的错误发生时触发，例如：除零错误，数据超范围等等。 |
| OperationalError  | 指非用户控制的，而是操作数据库时发生的错误。例如：连接意外断开、 数据库名未找到、事务处理失败、内存分配错误等等操作数据库是发生的错误。 |
| IntegrityError    | 完整性相关的错误，例如外键检查失败等。                       |
| InternalError     | 数据库的内部错误，例如游标（cursor）失效了、事务同步失败等等。 |
| ProgrammingError  | 程序错误，例如数据表（table）没找到或已存在、SQL语句语法错误、 参数数量错误等等。 |
| NotSupportedError | 不支持错误，指使用了数据库不支持的函数或API等。例如在连接对象上 使用.rollback()函数，然而数据库并不支持事务或者事务已关闭。 |

[https_github.com_PyMySQL_PyMySQL]: https://github.com/PyMySQL/PyMySQL