Python 如何操作 SQLite 数据库

让我们来详细讲解下 Python 如何操作 SQLite 数据库的完整攻略。

简介

SQLite 是一种轻量级的关系型数据库，可以嵌入到各种应用程序中，支持多种操作系统，是移动应用和桌面应用的理想选择。Python 代码就可以操作 SQLite 数据库，无需像 MySQL, PostgreSQL 这样的数据库一样需要一个服务进程。本文将讲解如何通过 Python 操作 SQLite 数据库。

安装

使用 Python 操作 SQLite 需要先安装 SQLite3 这个模块，可以通过 pip 安装。

pip install pysqlite3

连接SQLite

首先，需要连接到一个 SQLite 数据库。SQLite 的文件类型是 .db 文件。我们首先需要在 SQLite 中创建一个数据库。

import sqlite3

conn = sqlite3.connect('example.db')

在连接 SQLite 数据库时，可以指定需要连接的数据库的路径。如果不存在该文件，则会自动创建该文件。可以使用 connect() 方法创建一个数据库连接，如果需要不能修改数据库则需要打开方式为 sqlite3.connect('example.db', isolation_level=None)。

创建表

准备数据库和表格，可以使用 execute() 方法创建表。SQLite 中没有任何限制，可以将整数、实数、字符串甚至 BLOB（二进制数据）类型的数据存储到任何列中。我们将创建一个具有名称、年龄和地址的人员数据表。

import sqlite3

conn = sqlite3.connect('example.db')
print("Opened database successfully")

conn.execute('''CREATE TABLE COMPANY
       (ID INT PRIMARY KEY NOT NULL,
       NAME TEXT NOT NULL,
       AGE INT NOT NULL,
       ADDRESS CHAR(50),
       SALARY REAL);''')
print("Table created successfully")

conn.close()

插入数据

现在，我们已经创建了 COMPANY 表，可以在其中插入数据。我们将使用 execute() 方法插入 4 条数据。以下是插入数据的 Python 代码。

import sqlite3

conn = sqlite3.connect('example.db')
print("Opened database successfully")

conn.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
      VALUES (1, 'Paul', 32, 'California', 20000.00)")

conn.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
      VALUES (2, 'Allen', 25, 'Texas', 15000.00)")

conn.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
      VALUES (3, 'Teddy', 23, 'Norway', 20000.00)")

conn.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
      VALUES (4, 'Mark', 25, 'Rich-Mond', 65000.00)")

conn.commit()
print("Records created successfully")
conn.close()

查询数据

使用 SELECT 语句查询数据。我们可以使用 cursor 对象来查询数据。以下是完整的 Python 代码：

import sqlite3

conn = sqlite3.connect('example.db')
print("Opened database successfully")

cursor = conn.execute("SELECT id, name, address, salary from COMPANY")
for row in cursor:
   print("ID = ", row[0])
   print("NAME = ", row[1])
   print("ADDRESS = ", row[2])
   print("SALARY = ", row[3], "\n")

print("Operation done successfully")
conn.close()

此处，我们只是简单地进行选择动作。您可以使用 WHERE 语句添加更多筛选条件。

以上就是 Python 操作 SQLite 的完整攻略，希望对您有所帮助。