欢迎访问宙启技术站
智能推送

如何在Python中调用get_connection()函数获取数据库连接

发布时间:2024-01-19 15:16:15

在Python中调用get_connection()函数来获取数据库连接需要安装数据库相关的Python库,并根据具体的数据库类型和配置信息进行相应的连接。以下是在Python中调用get_connection()函数获取MySQL数据库连接的示例。

首先,需要安装mysql-connector-python库,可以使用以下命令进行安装:

pip install mysql-connector-python

然后,可以使用以下代码来获取MySQL数据库连接:

import mysql.connector

def get_connection():
    # 数据库配置信息
    config = {
        'user': 'root',
        'password': 'password',
        'host': 'localhost',
        'database': 'my_database'
    }

    # 获取数据库连接
    connection = mysql.connector.connect(**config)
    return connection

# 调用get_connection()函数获取数据库连接
db_connection = get_connection()

# 使用数据库连接进行数据库操作
cursor = db_connection.cursor()
cursor.execute('SELECT * FROM my_table')
result = cursor.fetchall()

# 打印查询结果
for row in result:
    print(row)

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

在上面的代码中,首先定义了一个get_connection()函数,该函数用于获取数据库连接。在函数内部,通过配置数据库的用户名、密码、主机地址和数据库名称等信息,创建一个config配置字典。然后使用mysql.connector.connect()函数,传入该配置字典来获取数据库连接对象connection

接下来,可以使用connection对象创建一个游标cursor,通过游标执行SQL查询语句SELECT * FROM my_table,并使用cursor.fetchall()方法获取查询结果result

最后,遍历查询结果并打印每一行的数据,并调用db_connection.close()方法关闭数据库连接。

上述示例代码是以MySQL数据库为例,调用mysql-connector-python库的函数来获取数据库连接。实际上,在不同的Python库中,获取数据库连接的函数可能会有所不同,需要根据使用的具体库来调用相应的函数。