5 分钟快速掌握在 Python 使用 SQLite 数据库

2021-08-12 10:19:50 浏览数 (1)

去年,州的先生曾经发表过一篇《小巧、稳定、快速!我为什么喜欢用 SQLite》的文章,里面对 SQLite 数据库极尽推崇。

小巧、稳定、快速!我为什么喜欢用SQLite

毕竟,对于大多数人和大多数业务来说,SQLite 都是一个极佳的数据库。

这个低开销、跨平台、自包含、零配置、事务性 SQL 数据库引擎让你一方面可以摆脱磁盘文件存储的痛苦(关系型数据存储),一方面可以让你摆脱各个大型数据库安装和运维的痛苦(单文件的存在,非常适合于关系型数据库初学者的学习和使用)。

在 Python 中,直接有一个内置库提供了对 SQLite 数据库的支持,所以我们可以在 Python 中直接使用 SQLite 数据库。

这可以让我们直接将 SQLite 数据库作为数据存储载体应用在我们的 Python 程序中,比如图形界面程序(PyQt5、Kivy、Tkinter)中的数据存储、Web 应用程序中的数据存储(Django 使用 SQLite 作为默认的数据库后端)

下面,我们就来了解一下 SQLite 在 Python 中的使用。

引入模块

Python 的内置库 sqlite3 提供了对 SQLite 数据库的支持。我们在 Python 代码中引入这个模块,即可拥有操作 SQLite 数据库的能力。

代码语言:javascript复制
import sqlite3

创建和连接数据库

因为 SQLite 是一个文件型的数据库,所以我们不需要像其他数据库那样配置 URL、端口、账号和密码,直接对 SQLite 数据库文件进行连接即可。

代码语言:javascript复制
# 创建或连接数据库
conn = sqlite3.connect("test.db")

如果本地不存在这个 test.db 数据库文件,则会自动创建。

创建数据表

创建或连接好数据库之后,就可以在数据库中创建数据表了。

大部分朋友,对数据表应该都很熟悉了。如果对数据库的概念还不是了解,那么可以用 Excel 来类比数据库,以快速对数据库有个基本的概念:

  • 一个 Excel 文件就是一个数据库;
  • 一个工作簿就是一个数据表;
  • 工作簿里面的列就是数据表里面的列名;
  • 工作簿里面的行就是数据库里面的行数据;

创建数据表,我们需要使用到 SQL 中的数据定义语言(DDL)。

SQL 的数据表创建语法为:

代码语言:javascript复制
CREATE TABLE 表名称
(
列名称 1 数据类型,
列名称 2 数据类型,
列名称 3 数据类型,
....
)

假设我们需要创建的表如下图所示:

那么,我们的表创建语句应该为:

代码语言:javascript复制
CREATE TABLE user(
	user_id int,
	user_name text,
	password text
)

我们在 Python 中通过实例化 SQLite 连接的 execute() 方法来执行 SQL 语句。

代码语言:javascript复制
# 创建 user 数据库
conn.execute('''
CREATE TABLE user(
	user_id int,
	user_name text,
	password text
)
''')

下面的大部分 SQL 操作,我们也都是使用这个方法来执行。

写入数据

数据表创建之后,我们可以使用 SQL 的数据操作语言来对数据进行增删改查了。

也就是:

  • SELECT - 从数据库表中获取数据
  • UPDATE - 更新数据库表中的数据
  • DELETE - 从数据库表中删除数据
  • INSERT INTO - 向数据库表中插入数据

下面,我们往数据表中写入一些数据,同样是使用 execute() 方法:

代码语言:javascript复制
# 写入数据
conn.execute("INSERT INTO user (user_id,user_name,password) VALUES(1,'zmister','123456')")
conn.execute("INSERT INTO user (user_id,user_name,password) VALUES(2,'mrdoc','234323')")
conn.execute("INSERT INTO user (user_id,user_name,password) VALUES(2,'python','hahaheheh')")

最后,再调用连接的 commit() 方法以提交这些语句。

代码语言:javascript复制
conn.commit()

这样我们就往 test.db 数据库的 user 数据表中插入了 3 条数据。

插入、更新和删除语句都需要调用 commit()方法。

查看数据库

除了在程序的代码中对数据库进行操作外,日常我们一般使用图形化的数据库管理工具对数据库进行管理。

在此州的先生推荐 sqlite expert 这个软件,免费、小巧、没有试用期,不用到处找破解方法。

你可以在其官方下载页面进行下载安装:http://www.sqliteexpert.com/download.html

安装之后,我们直接选择本地的 SQLite 数据库文件就可以打开:

在这里面可以进行数据库的各种管理操作和各种数据操作,在此暂且不表,自己把玩吧。

查询表数据

如果我们需要查询表数据,那么需要使用到 SELECT 语句,比如查询我们刚刚创建的 user 表内的数据:

代码语言:javascript复制
SELECT * FROM user

我们在 Python 代码中继续实现它:

代码语言:javascript复制
# coding:utf-8

import sqlite3

# 创建或连接数据库
conn = sqlite3.connect("test.db")

# 查询数据
cursor = conn.execute("SELECT * FROM user")
for row in cursor.fetchall():
    print(row)

conn.close()

这将逐行在控制台打印出数据:

可以看到,返回的数据是以元组的形式呈现的,如果我们需要以字典的形式进行程序,可以指定 row_factory 以改变游标的可调用对象。

代码语言:javascript复制
# coding:utf-8

import sqlite3

def dict_factory(cursor, row):
    d = {}
    for idx, col in enumerate(cursor.description):
        d[col[0]] = row[idx]
    return d

# 创建或连接数据库
conn = sqlite3.connect("test.db")
conn.row_factory = dict_factory

# 查询数据
cursor = conn.execute("SELECT * FROM user")
for row in cursor.fetchall():
    print(row)

conn.close()

这返回的就会是字典形式的数据:

更新数据

如果需要更新数据,那么就得使用 UPDATE 语句:

代码语言:javascript复制
UPDATE 表名称 SET 列名称 = 新值 WHERE 列名称 = 某值

例如,我们要把 user_id 为 1 的数据的 password 字段更新为 888888:

代码语言:javascript复制
UPDATE user SET password = '888888' WHERE user_id = 1;

放在 Python 中执行的代码为:

代码语言:javascript复制
# coding:utf-8

import sqlite3

# 创建或连接数据库
conn = sqlite3.connect("test.db")

# 更新数据
conn.execute("UPDATE user SET password = '888888' WHERE user_id = 1;")
conn.commit()

# 查询数据
cursor = conn.execute("SELECT * FROM user")
for row in cursor.fetchall():
    print(row)

conn.close()

返回查询的数据表示数据已更新:

删除数据

删除数据需要使用 DELETE 语句,其语法为:

代码语言:javascript复制
DELETE FROM 表名称 WHERE 列名称 = 值

例如,我们要把 user_name 为 python 的数据删除,其语句为:

代码语言:javascript复制
DELETE FROM user WHERE user_name = 'python'

放在 Python 中执行的代码为:

代码语言:javascript复制
# coding:utf-8

import sqlite3

# 创建或连接数据库
conn = sqlite3.connect("test.db")

# 删除数据
conn.execute("DELETE FROM user WHERE user_name = 'python'")
conn.commit()

# 查询数据
cursor = conn.execute("SELECT * FROM user")
for row in cursor.fetchall():
    print(row)

conn.close()

返回查询的数据表示数据已删除:

最后

SQLite 作为一个小巧强悍的数据库,有足够的优势值得你在自己的项目和程序中作为数据存储的载体。

而且 Python 内置库原生提供了对 SQLite 的支持,如果你困于文本文件存储的性能和大型数据库的难以维护,就来试试 SQLite 吧。

0 人点赞