博客

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

  • Post category:Python

获得数据库连接和游标对象是Python连接到数据库最关键的步骤，下面就详细说明如何使用Python函数完成这一过程。

步骤 1：导入模块

首先，需要导入Python连接数据库所需模块。Python中连接数据库的模块有很多，比如MySQLdb、pymysql、psycopg2、sqlite3等等。这里以MySQL为例，使用pymysql模块实现。

import pymysql

步骤 2: 连接数据库

连接到数据库需要在代码中指定主机名、用户名、密码和数据库名称等必要信息。连接时需要指定MySQL的地址、端口号、用户名和密码等信息。 

def connect_mysql():
    # 建立数据库连接
    conn = pymysql.connect(
        host='localhost',  # 数据库地址
        port=3306,  # 端口号，一般为 3306
        user='root',  # 用户名
        password='123456',  # 密码
        database='test',  # 数据库名
        charset='utf8'  # 编码方式
    )
    return conn

步骤 3: 创建游标

建立连接后，需要创建游标对象，游标是对数据库对象的操作的句柄，用于执行SQL语句。

def get_cursor(conn):
    # 创建游标对象
    cursor = conn.cursor()
    return cursor

步骤 4: 查询数据库

完成了连接和游标的获取后，就可以开始对数据库进行操作，比如查询数据。

def query_data(cursor):
    # SQL查询语句
    sql = "SELECT * FROM table_name"
    # 执行SQL语句
    cursor.execute(sql)
    # 获取查询结果
    result = cursor.fetchall()
    return result

步骤 5：关闭连接

当数据库操作完成后，需要关闭连接。

def close_conn(conn, cursor):
    # 关闭游标和数据库连接
    cursor.close()
    conn.close()

完整函数代码示例

以下是完整的函数代码示例，包含了连接数据库、获取游标、查询数据库、关闭连接等步骤。

import pymysql

# 连接数据库
def connect_mysql():
    # 建立数据库连接
    conn = pymysql.connect(
        host='localhost',  # 数据库地址
        port=3306,  # 端口号，一般为 3306
        user='root',  # 用户名
        password='123456',  # 密码
        database='test',  # 数据库名
        charset='utf8'  # 编码方式
    )
    return conn

# 创建游标
def get_cursor(conn):
    # 创建游标对象
    cursor = conn.cursor()
    return cursor

# 查询数据
def query_data(cursor):
    # SQL查询语句
    sql = "SELECT * FROM table_name"
    # 执行SQL语句
    cursor.execute(sql)
    # 获取查询结果
    result = cursor.fetchall()
    return result

# 关闭连接
def close_conn(conn, cursor):
    # 关闭游标和数据库连接
    cursor.close()
    conn.close()

# 测试连接
if __name__ == '__main__':
    # 连接数据库
    conn = connect_mysql()
    # 获取游标
    cursor = get_cursor(conn)
    # 查询数据
    result = query_data(cursor)
    print(result)
    # 关闭连接
    close_conn(conn, cursor)

以上就是Python函数返回数据库连接和游标的完整攻略，其中包含了连接数据库、获取游标、查询数据库、关闭连接等步骤，并提供了完整的代码示例。

Tags: