Python —— pymysql模块的使用

pymysql模块的使用

大部分是实操代码，没有过多的文字说明

下载及引用模块

下载 pymysql 模块

pip install pymysql

导入 pymysql 模块

import pymysql

连接数据库

# 连接数据库
def connect_sql():
    # 打开数据库连接
    db = pymysql.connect(
        host='localhost',
        user='数据库用户名',
        password='数据库密码',
        database='需要连接的数据库名'
    )
    cursor = db.cursor()  # 创建游标对象

    cursor.execute("SELECT VERSION()")  # 执行SQL查询（测试获取数据库版本）

    datas = cursor.fetchone()  # 获取单条数据
    print(datas)  # 测试输出
    cursor.close()  # 关闭游标对象
    db.close()  # 关闭数据库连接

connect_sql()

创建数据表

# 创建数据表
def create_table():
    # 打开数据库连接
    db = pymysql.connect(
        host='localhost',
        user='数据库用户名',
        password='数据库密码',
        database='需要连接的数据库名'
    )
    cursor = db.cursor()  # 创建游标对象

    # sql语句
    sql = """CREATE TABLE DEMO (
        USER VARCHAR(50),
        AGE INT)
        """
    cursor.execute(sql)		# 执行SQL语句
    cursor.close()  # 关闭游标对象
    db.close()  # 关闭数据库连接

# 运行方法
create_table()

添加数据

（1）直接在SQL语句中写入数据

# 插入数据
def insert_data_01():
    # 打开数据库连接
    db = pymysql.connect(
        host='localhost',
        user='数据库用户名',
        password='数据库密码',
        database='需要连接的数据库名'
    )
    cursor = db.cursor()  # 创建游标对象

    # sql语句
    sql = """INSERT INTO DEMO(USER, AGE) VALUES ('JSON', 21)"""

    try:
        cursor.execute(sql)  # 执行SQL语句
        db.commit()  # 提交到数据库中执行
        print("数据写入成功！")
    except:
        # db.rollback()  # 如果发生错误则回滚
        print("写入数据发生错误！")

    cursor.close()      # 关闭游标对象
    db.close()          # 关闭数据库

# 运行方法
insert_data_01()

（2）使用占位符添加数据

一般用于添加多条数据的情况

# 插入数据
def insert_data_02():
    # 打开数据库连接
    db = pymysql.connect(
        host='localhost',
        user='数据库用户名',
        password='数据库密码',
        database='需要连接的数据库名'
    )
    cursor = db.cursor()  # 创建游标对象

    user = '张三'
    age = 23

    # sql语句
    sql = "INSERT INTO DEMO(USER, AGE) VALUES ('%s', '%s')" % (user, age)

    try:
        cursor.execute(sql)  # 执行SQL语句
        db.commit()  # 提交到数据库中执行
        print("数据02写入成功！")
    except:
        # db.rollback()  # 如果发生错误则回滚
        print("写入数据发生错误！")

    cursor.close()      # 关闭游标对象
    db.close()          # 关闭数据库

# 运行方法
insert_data_02()

查询数据

# 查询
def selcet_data():
    # 打开数据库连接
    db = pymysql.connect(
        host='localhost',
        user='数据库用户名',
        password='数据库密码',
        database='需要连接的数据库名'
    )
    cursor = db.cursor()        # 创建游标对象
    sql = "SELECT * FROM DEMO"  # sql语句
    cursor.execute(sql)         # 执行SQL语句
    results = cursor.fetchall()     # fetchall()：获取多条数据；fetchone()：获取单条数据
    print(results)      # 输出结果

    cursor.close()  # 关闭游标对象
    db.close()      # 关闭数据库

# 运行方法
selcet_data()

更新数据

# 更新
def update_data():
    db = pymysql.connect(
        host='localhost',
        user='数据库用户名',
        password='数据库密码',
        database='需要连接的数据库名'
    )
    cursor = db.cursor()  # 创建游标对象
    sql = "UPDATE DEMO SET USER = 'JAKSON' WHERE USER = 'JSON'"  # sql语句

    try:
        cursor.execute(sql)  # 执行SQL语句
        db.commit()     # 提交到数据库执行
        print("数据更新成功！")
    except:
        print("数据更新异常！")

    cursor.close()  # 关闭游标对象
    db.close()  # 关闭数据库

# 运行方法
update_data()

删除数据

def delete_data():
    db = pymysql.connect(
        host='localhost',
        user='数据库用户名',
        password='数据库密码',
        database='需要连接的数据库名'
    )
    cursor = db.cursor()  # 创建游标对象
    sql = "DELETE FROM DEMO WHERE AGE = '26'"  # sql语句

    try:
        cursor.execute(sql)  # 执行SQL语句
        db.commit()  # 提交到数据库执行
        print("数据删除成功！")
    except:
        print("数据删除异常！")

    cursor.close()  # 关闭游标对象
    db.close()  # 关闭数据库

# 运行方法
delete_data()