python 使用pymysql连接Mysql方法

调用如下方法传入sql,即可得到返回数据

链接

import pymysql
from pymysql.cursors import DictCursor
def getData(sql):
    # 1、连接数据库
    conn = pymysql.connect(
        host='数据库Host',
        port=端口,
        user='用户名',
        password='密码',
        database='数据库名称'
    )
    # 目标：执行 sql 语句
    # 2、创建游标
    cursor = conn.cursor(cursor=DictCursor)     # 字典游标 => 获得的结果是 字典
                                                # 原本获取到的查询结果是 元组

    # 接下来就可以用游标去执行各种操作了
    # 4. 执行sql
    result = cursor.execute(sql)
    print(result)

    # 注意: 查询操作, 返回数据存放在 cursor 里

    # # fetchone() 一次拿一条记录
    # one = cursor.fetchone()
    # print(one)
    # # fetchmany(3) 一次拿三条数据
    # ret = cursor.fetchmany(3)
    # print(ret)

    # fetchall() 全拿
    send_data = cursor.fetchall()

    if cursor:
        cursor.close()
    if conn:
        conn.close()  # 一定要记得关闭链接
    return  send_data

增删改查

下面是使用 pymysql 进行增删改查操作的代码示例：

python复制代码

import pymysql

# 建立连接
conn = pymysql.connect(host='localhost', user='root', password='123456', database='test', port=3306)

# 创建游标对象
cursor = conn.cursor()

# 查询操作
sql = "SELECT * FROM users"
cursor.execute(sql)
results = cursor.fetchall()
print(results)

# 插入操作
sql = "INSERT INTO users(username, age, gender) VALUES (%s, %s, %s)"
data = ("Tom", 20, 1)
cursor.execute(sql, data)
conn.commit()
print("插入成功！")

# 更新操作
sql = "UPDATE users SET age=%s WHERE id=%s"
data = (21, 2)
cursor.execute(sql, data)
conn.commit()
print("更新成功！")

# 删除操作
sql = "DELETE FROM users WHERE id=%s"
data = (2,)
cursor.execute(sql, data)
conn.commit()
print("删除完成！")

# 关闭游标和连接
cursor.close()
conn.close()`

以上代码对 test 数据库中的 users 表执行了查询、插入、更新和删除操作。execute() 方法用于执行 SQL 语句，其中 %s 表示占位符，可以使用元组或列表等数据结构进行数据传参。调用 commit() 方法可以提交刚才执行的操作，close() 方法用于关闭游标对象和数据库连接。

批量插入

Python 批量插入 MySQL，可以使用 executemany() 方法实现，示例如下：

import mysql.connector

# 连接 MySQL 数据库
cnx = mysql.connector.connect(user='username', password='password',
                              host='127.0.0.1',
                              database='database_name')
cursor = cnx.cursor()

# 执行批量插入操作
insert_query = "INSERT INTO table_name (column1, column2, column3) VALUES (%s, %s, %s)"
data = [('value11', 'value21', 'value31'), ('value12', 'value22', 'value32')]
cursor.executemany(insert_query, data)

# 提交操作
cnx.commit()

# 关闭连接
cursor.close()
cnx.close()