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

如何在Python中实现数据库连接和数据查询函数?

发布时间:2023-09-10 20:15:45

要在Python中实现数据库连接和数据查询函数,可以使用Python中的标准库sqlite3来连接和操作SQLite数据库,也可以使用第三方库如pymysql来连接和操作MySQL数据库。

首先,我们需要安装相应的库。可以使用以下命令来安装sqlite3pymysql库:

pip install sqlite3
pip install pymysql

接下来,我们可以使用以下代码实现数据库连接和数据查询函数。

### SQLite数据库连接和数据查询函数

import sqlite3

# 连接到SQLite数据库
def connect_db():
    conn = sqlite3.connect('database.db')
    return conn

# 关闭数据库连接
def close_db(conn):
    conn.close()

# 执行SQL查询语句
def execute_query(conn, query):
    cursor = conn.cursor()
    cursor.execute(query)
    result = cursor.fetchall()
    return result

# 示例:查询表中的所有数据
def get_all_data():
    conn = connect_db()
    query = "SELECT * FROM table_name"
    result = execute_query(conn, query)
    close_db(conn)
    return result

### MySQL数据库连接和数据查询函数

import pymysql

# 连接到MySQL数据库
def connect_db():
    conn = pymysql.connect(
        host='localhost',
        user='username',
        password='password',
        db='database_name',
        charset='utf8mb4',
        cursorclass=pymysql.cursors.DictCursor
    )
    return conn

# 关闭数据库连接
def close_db(conn):
    conn.close()

# 执行SQL查询语句
def execute_query(conn, query):
    cursor = conn.cursor()
    cursor.execute(query)
    result = cursor.fetchall()
    return result

# 示例:查询表中的所有数据
def get_all_data():
    conn = connect_db()
    query = "SELECT * FROM table_name"
    result = execute_query(conn, query)
    close_db(conn)
    return result

上述代码中的database.db是SQLite数据库文件的名称,table_name是要查询的表的名称,usernamepassword分别是MySQL数据库的用户名和密码,database_name是要连接的MySQL数据库的名称。

使用以上代码示例,您可以根据具体需求编写更多的数据库连接和数据查询函数,来实现在Python中连接和操作数据库的功能。