您的位置:

Python连接MySQL增删改查

MySQL是一种常见的关系型数据库,而Python是一种编程语言,两者之间通过一些特定的库进行连接,以实现对MySQL数据库的增删改查操作。在本文中,我们将详细介绍如何使用Python连接MySQL数据库,并进行增删改查操作。

一、连接MySQL数据库

在使用Python对MySQL进行操作之前,首先需要连接到数据库。Python中的MySQL连接库主要有两种,一种是MySQLdb,一种是PyMySQL。这里我们将以PyMySQL为例进行介绍,在使用之前需要先通过pip install PyMySQL命令进行安装。

import pymysql

# 建立数据库连接
conn = pymysql.connect(
    host='localhost',  # 连接地址
    port=3306,  # 端口号
    user='root',  # 数据库用户名
    password='password',  # 数据库密码
    database='test',  # 数据库名称
    charset='utf8',  # 字符编码
)

# 关闭数据库连接
conn.close()

以上代码中,我们使用pymysql.connect()方法进行连接,需要传入连接所需的参数:连接地址、端口号、数据库用户名、数据库密码、数据库名称、字符编码。建立连接后,我们可以使用conn.close()方法进行关闭。

二、查询数据

在连接MySQL数据库之后,我们可以对其中的数据进行查询。查询数据需要使用游标,Python中的pymysql库提供了两种游标:普通游标和字典游标。普通游标返回的结果是一个元组,而字典游标返回的结果是一个字典。

1. 普通游标

import pymysql

# 建立数据库连接
conn = pymysql.connect(
    host='localhost',  # 连接地址
    port=3306,  # 端口号
    user='root',  # 数据库用户名
    password='password',  # 数据库密码
    database='test',  # 数据库名称
    charset='utf8',  # 字符编码
)

# 建立游标
cur = conn.cursor()

# 执行查询
sql = "SELECT * FROM users"
cur.execute(sql)

# 获取结果集
results = cur.fetchall()
for row in results:
    id = row[0]
    name = row[1]
    age = row[2]
    print("id=%d, name=%s, age=%d" % (id, name, age))

# 关闭游标
cur.close()

# 关闭数据库连接
conn.close()

以上代码中, first we create connent and create cursor,then execute query using cursor.execute and then get the reslt retunred using cursor.fetchall.

2. 字典游标

import pymysql

# 建立数据库连接
conn = pymysql.connect(
    host='localhost',  # 连接地址
    port=3306,  # 端口号
    user='root',  # 数据库用户名
    password='password',  # 数据库密码
    database='test',  # 数据库名称
    charset='utf8',  # 字符编码
)

# 建立字典游标
cur = conn.cursor(pymysql.cursors.DictCursor)

# 执行查询
sql = "SELECT * FROM users"
cur.execute(sql)

# 获取结果集
results = cur.fetchall()
for row in results:
    id = row['id']
    name = row['name']
    age = row['age']
    print("id=%d, name=%s, age=%d" % (id, name, age))

# 关闭游标
cur.close()

# 关闭数据库连接
conn.close()

以上代码中,我们通过传入pymysql.cursors.DictCursor参数,创建了字典游标,然后执行查询,获取结果集,并对结果集进行遍历输出。

三、插入数据

在数据库中插入数据需要使用游标,首先需要创建游标,然后执行插入语句,最后提交事务。

import pymysql

# 建立数据库连接
conn = pymysql.connect(
    host='localhost',  # 连接地址
    port=3306,  # 端口号
    user='root',  # 数据库用户名
    password='password',  # 数据库密码
    database='test',  # 数据库名称
    charset='utf8',  # 字符编码
)

# 建立游标
cur = conn.cursor()

# 执行插入
sql = "INSERT INTO users(name,age) VALUES(%s,%s)"
cur.execute(sql, ('Tom', 18))

# 提交事务
conn.commit()

# 关闭游标
cur.close()

# 关闭数据库连接
conn.close()

以上代码中,我们创建游标,然后通过cur.execute()方法执行插入语句,并向其中传入需要插入的数据。最后,我们需要执行conn.commit()方法来提交事务。如果不执行conn.commit()方法,则不会对数据库产生任何修改。

四、更新数据

更新数据与插入数据类似,需要通过游标来实现。需要注意的是,更新语句需要指定更新的数据条件。

import pymysql

# 建立数据库连接
conn = pymysql.connect(
    host='localhost',  # 连接地址
    port=3306,  # 端口号
    user='root',  # 数据库用户名
    password='password',  # 数据库密码
    database='test',  # 数据库名称
    charset='utf8',  # 字符编码
)

# 建立游标
cur = conn.cursor()

# 执行更新
sql = "UPDATE users SET age=%s WHERE name=%s"
cur.execute(sql, (20, 'Tom'))

# 提交事务
conn.commit()

# 关闭游标
cur.close()

# 关闭数据库连接
conn.close()

以上代码中,我们通过游标执行更新语句,其中更新语句指定了更新数据的条件。然后需要执行conn.commit()方法提交事务。

五、删除数据

删除数据需要使用游标,同样需要指定删除数据的条件,并提交事务。

import pymysql

# 建立数据库连接
conn = pymysql.connect(
    host='localhost',  # 连接地址
    port=3306,  # 端口号
    user='root',  # 数据库用户名
    password='password',  # 数据库密码
    database='test',  # 数据库名称
    charset='utf8',  # 字符编码
)

# 建立游标
cur = conn.cursor()

# 执行删除
sql = "DELETE FROM users WHERE name=%s"
cur.execute(sql, ('Tom',))

# 提交事务
conn.commit()

# 关闭游标
cur.close()

# 关闭数据库连接
conn.close()

以上代码中,我们通过游标执行删除语句,其中删除语句指定了删除数据的条件。然后需要执行conn.commit()方法提交事务。