Python基础（二十九、pymsql）

news2024/9/30 11:36:03

文章目录

一、安装pymysql库
二、代码实践
- 1.连接MySQL数据库
- 2.创建表格
- 3.插入数据
- 4.查询数据
- 5.更新数据
- 6.删除数据
三、完整代码示例
四、结论

使用Python的pymysql库可以实现数据存储，这是一种连接MySQL数据库的方式。在本篇文章中，将详细介绍如何使用pymysql库连接MySQL数据库、创建表格、插入数据以及查询数据等操作。
准备工作，本地要安装好mysql数据库，可通过官方网站下载 Mysql数据库

一、安装pymysql库

使用Python的pymysql库需要先安装该库。可以通过pip安装：

pip install pymysql

在这里插入图片描述

二、代码实践

1.连接MySQL数据库

连接到MySQL数据库之前，需要知道MySQL服务器的相关信息，例如服务器地址、用户名、密码和数据库名称等。这些信息可以根据具体情况进行修改。

使用以下代码连接到MySQL数据库：

import pymysql

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

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

# 关闭连接
conn.close()

上述代码中，我们使用connect()方法创建一个连接对象，该方法需要传递MySQL服务器的相关信息，例如服务器地址、用户名、密码和数据库名称等。成功建立连接后，我们可以使用cursor()方法创建一个游标对象。最后，我们使用close()方法关闭连接。

2.创建表格

连接到MySQL数据库后，可以使用以下代码创建一个表格：

# 创建表格
create_table_query = """CREATE TABLE users (
    id INT(11) NOT NULL AUTO_INCREMENT,
    name VARCHAR(255) NOT NULL,
    age INT(11) NOT NULL,
    PRIMARY KEY (id)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci"""

cursor.execute(create_table_query)

上述代码中，我们使用SQL语句创建了名为“users”的表格，并定义了三个字段：id、name和age。同时，我们还为id字段添加了主键约束。最后，我们使用execute()方法执行SQL语句。

3.插入数据

创建表格后，可以使用以下代码向表格中插入数据：

# 插入数据
insert_query = """INSERT INTO users (name, age) VALUES ('Tom', 25)"""
cursor.execute(insert_query)

上述代码中，我们使用SQL语句向“users”表格中插入了一行数据，其中name字段为“Tom”，age字段为25。最后，我们使用execute()方法执行SQL语句。

4.查询数据

可以使用以下代码查询表格中的数据：

# 查询数据
select_query = "SELECT * FROM users"
cursor.execute(select_query)
results = cursor.fetchall()

for row in results:
    id = row[0]
    name = row[1]
    age = row[2]
    print("id=%d,name=%s,age=%d" % (id, name, age))

上述代码中，我们使用SQL语句查询“users”表格中的所有数据，并使用fetchall()方法获取结果集。最后，我们遍历结果集并输出每一行数据。

5.更新数据

可以使用以下代码更新表格中的数据：

# 更新数据
update_query = "UPDATE users SET age=26 WHERE name='Tom'"
cursor.execute(update_query)
conn.commit()

上述代码中，我们使用SQL语句更新“users”表格中名为“Tom”的行的age字段为26。最后，我们使用commit()方法提交更改。

6.删除数据

可以使用以下代码从表格中删除数据：

# 删除数据
delete_query = "DELETE FROM users WHERE name='Tom'"
cursor.execute(delete_query)
conn.commit()

上述代码中，我们使用SQL语句从“users”表格中删除名为“Tom”的行。最后，我们使用commit()方法提交更改。

三、完整代码示例

下面是完整的Python代码示例：

import pymysql

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

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

# 创建表格
create_table_query = """CREATE TABLE users (
    id INT(11) NOT NULL AUTO_INCREMENT,
    name VARCHAR(255) NOT NULL,
    age INT(11) NOT NULL,
    PRIMARY KEY (id)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci"""

cursor.execute(create_table_query)

# 插入数据
insert_query = """INSERT INTO users (name, age) VALUES ('Tom', 25)"""
cursor.execute(insert_query)

# 查询数据
select_query = "SELECT * FROM users"
cursor.execute(select_query)
results = cursor.fetchall()

for row in results:
    id = row[0]
    name = row[1]
    age = row[2]
    print("id=%d,name=%s,age=%d" % (id, name, age))

# 更新数据
update_query = "UPDATE users SET age=26 WHERE name='Tom'"
cursor.execute(update_query)
conn.commit()

# 删除数据
delete_query = "DELETE FROM users WHERE name='Tom'"
cursor.execute(delete_query)
conn.commit()

# 关闭连接
conn.close()