博客

Python中PyMySQL的基本操作

  • Post category:Python

关于Python中PyMySQL的基本操作，我们可以分为以下几个方面进行讲解。

环境配置

首先，我们需要在本地安装了Python和PyMySQL模块。如果没有安装，可以使用pip指令进行安装：

pip install pymysql

连接数据库

连接数据库是使用PyMySQL的第一步。使用pymysql.connect()函数可以连接到指定的数据库。

import pymysql

#连接数据库
conn = pymysql.connect(host='localhost', user='root', password='password', database='test', charset='utf8')

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

以上代码中的host、user、password、database和charset分别表示服务器地址、用户名、密码、数据库名和编码方式，需要根据实际情况进行修改。

执行SQL语句

连接成功后，就可以执行SQL语句了。我们使用execute()函数来执行SQL语句，使用commit()函数来提交修改。

sql = "INSERT INTO users (username, password) VALUES ('admin', '123456')"
cursor.execute(sql)
conn.commit()

以上代码向users表中插入一条数据。execute()函数可以执行select、insert、update和delete等操作。

查询数据

使用fetchall()函数从游标中获取所有记录，使用fetchone()函数从游标中获取一条记录。

cursor.execute("SELECT * FROM users")
rows = cursor.fetchall()
for row in rows:
    print(row)

以上代码从users表中获取所有记录，并将每一行打印输出。

代码示例

下面是一条完整的代码示例，演示如何连接到MySQL数据库、创建表、插入数据和查询数据。

import pymysql

#连接数据库
conn = pymysql.connect(host='localhost', user='root', password='password', database='test', charset='utf8')

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

#创建表
sql = '''CREATE TABLE IF NOT EXISTS users (
         id INT(11) NOT NULL AUTO_INCREMENT,
         username VARCHAR(20) NOT NULL,
         password VARCHAR(20) NOT NULL,
         PRIMARY KEY (id)
         )ENGINE=InnoDB DEFAULT CHARSET=utf8'''
cursor.execute(sql)

#插入数据
sql = "INSERT INTO users (username, password) VALUES ('admin', '123456')"
cursor.execute(sql)
conn.commit()

#查询数据
cursor.execute("SELECT * FROM users")
rows = cursor.fetchall()
for row in rows:
    print(row)

#关闭连接
conn.close()

以上代码演示了如何使用PyMySQL模块连接到MySQL数据库、创建表、插入数据和查询数据，希望对你有帮助！

Tags: