python查询 SQLite 数据库中的数据

2023-04-04 00:00:00 查询 数据 数据库中

要使用 Python 查询 SQLite 数据库中的数据,首先需要确保已安装了 sqlite3 库。sqlite3 是 Python 标准库的一部分,因此通常无需安装即可使用。

下面是一个例子,演示如何使用 Python 连接到 SQLite 数据库并查询数据。假设我们有一个名为 website_data.db 的 SQLite 数据库,其中包含一个名为 websites 的表。表结构如下:

  • id (INTEGER) - 主键
  • domain (TEXT) - 网站域名
  • name (TEXT) - 网站名称

以下是如何使用 Python 查询这个数据库中的数据:

import sqlite3

def connect_db(database_path):
    connection = sqlite3.connect(database_path)
    return connection

def query_db(connection, query):
    cursor = connection.cursor()
    cursor.execute(query)
    results = cursor.fetchall()
    return results

def close_db(connection):
    connection.close()

# 连接到数据库
database_path = 'website_data.db'
connection = connect_db(database_path)

# 查询数据库中的数据
query = "SELECT * FROM websites WHERE domain='pidancode.com'"
results = query_db(connection, query)

# 打印查询结果
for row in results:
    print(f"id: {row[0]}, domain: {row[1]}, name: {row[2]}")

# 关闭数据库连接
close_db(connection)

在这个示例中,我们首先导入了 sqlite3 模块。然后,我们定义了三个函数:

  • connect_db: 该函数接受一个数据库文件路径,并使用 sqlite3.connect() 方法连接到数据库。返回一个数据库连接对象。
  • query_db: 该函数接受一个数据库连接对象和一个查询字符串。它创建一个游标对象,执行查询,然后使用 fetchall() 方法获取查询结果。最后,它返回查询结果。
  • close_db: 该函数接受一个数据库连接对象,并使用 close() 方法关闭数据库连接。

在主代码部分,我们首先连接到 website_data.db 数据库。然后,我们定义一个查询字符串,该字符串从 websites 表中选择域名为 pidancode.com 的所有记录。接着,我们调用 query_db 函数并打印查询结果。最后,我们关闭数据库连接。

相关文章