博客

python函数返回数据库连接和游标

  • Post category:Python

Python连接数据库需要使用相应的驱动和数据库连接信息，常用的驱动有pymysql、pyodbc等，本次攻略以pymysql库为例。

安装pymysql库

  1. 使用pip命令安装pymysql库：pip install pymysql

建立数据库连接

可以使用pymysql.connect()方法构建数据库连接。

import pymysql

# 数据库连接信息
host = 'localhost'
user = 'root'
password = '123456'
database = 'testdb'

# 建立数据库连接
conn = pymysql.connect(
    host=host,
    user=user,
    password=password,
    database=database,
    charset='utf8mb4'
)

其中，pymysql.connect()方法的参数如下：

  • host：数据库主机地址
  • user：数据库用户
  • password：数据库用户密码
  • database：要连接的数据库名
  • charset：连接字符集，一般设置为utf8mb4

获得游标

建立连接后，需要创建游标。

import pymysql

# 数据库连接信息
host = 'localhost'
user = 'root'
password = '123456'
database = 'testdb'

# 建立数据库连接
conn = pymysql.connect(
    host=host,
    user=user,
    password=password,
    database=database,
    charset='utf8mb4'
)

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

执行SQL语句

建立游标后，可以使用游标执行SQL语句。

import pymysql

# 数据库连接信息
host = 'localhost'
user = 'root'
password = '123456'
database = 'testdb'

# 建立数据库连接
conn = pymysql.connect(
    host=host,
    user=user,
    password=password,
    database=database,
    charset='utf8mb4'
)

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

# 执行SQL语句
sql = "SELECT * FROM user WHERE age>=20"
cur.execute(sql)

# 获取查询结果
result = cur.fetchall()
print(result)

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

其中，cur.execute()方法用于执行SQL语句。

关闭游标和连接

当查询结束后，需要关闭游标和连接，以释放资源。

import pymysql

# 数据库连接信息
host = 'localhost'
user = 'root'
password = '123456'
database = 'testdb'

# 建立数据库连接
conn = pymysql.connect(
    host=host,
    user=user,
    password=password,
    database=database,
    charset='utf8mb4'
)

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

# 执行SQL语句
sql = "SELECT * FROM user WHERE age>=20"
cur.execute(sql)

# 获取查询结果
result = cur.fetchall()
print(result)

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

以上就是使用pymysql连接数据库和执行SQL语句的完整攻略。

Tags: