脚本之家,脚本语言编程技术及教程分享平台!
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|shell|

服务器之家 - 脚本之家 - Python - 几行代码教你Python操作小型数据库 SQLite

几行代码教你Python操作小型数据库 SQLite

2024-03-08 14:17Python都知道 Python

本文我们使用 Python 自带的 sqlite3 模块来演示如何使用 Python 操作 SQLite 数据库。

SQLite介绍

SQLite 是一款轻型、小巧、高效的关系型数据库管理系统,它可以在任何操作系统中运行。而且它的驱动程序也非常丰富,各种语言都有对应的驱动。

这里我们使用 Python 自带的 sqlite3 模块来演示如何使用 Python 操作 SQLite 数据库。

几行代码教你Python操作小型数据库 SQLite

引入模块

使用 Python 操作 SQLite 需要先引入 sqlite3 模块。

import sqlite3

连接数据库

连接 SQLite 数据库可以使用 connect() 方法,它有一个必需参数是数据库文件名,如果文件不存在则会自动创建,如果文件存在则会连接该文件。

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

connect() 方法还可以有其他参数,例如 timeout,它表示在等待某个锁的时间内如果没有成功,则会抛出 sqlite3.BusyError 异常。

创建表

创建表可以执行 SQL 语句,也可以使用 Python 操作数据库的 API。

# 创建表
conn.execute('''
CREATE TABLE if not exists company
(ID INT PRIMARY KEY NOT NULL,
NAME TEXT NOT NULL,
AGE INT NOT NULL,
ADDRESS CHAR(50),
SALARY REAL);
''')

插入数据

插入数据需要使用 INSERT 语句,语法为:

conn.execute(f"INSERT INTO company (ID,NAME,AGE,ADDRESS,SALARY)
              VALUES ({id}, '{name}', {age}, '{address}', {salary})")

或者使用参数化 SQL 语句,语法为:

conn.execute("INSERT INTO company (ID,NAME,AGE,ADDRESS,SALARY) VALUES (?, ?, ?, ?, ?)", 
             (id, name, age, address, salary))

其中 ? 表示占位符,后面的元组中为需要插入的数据。

查询数据

查询数据需要使用 SELECT 语句,语法为:

cursor = conn.execute("SELECT ID, NAME, AGE, ADDRESS, SALARY FROM company")
for row in cursor:
    print(f"ID = {row[0]}, NAME = {row[1]}, AGE = {row[2]}, ADDRESS = {row[3]}, SALARY = {row[4]}")

更新数据

更新数据需要使用 UPDATE 语句,语法为:

conn.execute(f"UPDATE company SET SALARY = {salary} WHERE ID = {id}")

删除数据

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

conn.execute(f"DELETE from company where ID = {id}")

提交事务

Python 操作 SQLite 数据库自动开启一个事务,如果要提交事务,需要使用 commit() 方法。

conn.commit()

关闭数据库连接

不需要访问数据库时,需要关闭连接。

conn.close()

总结

通过上述代码可以看出,使用 Python 操作 SQLite 数据库非常简单,只需要使用一些简单的 SQL 语句或者 Python API 就可以完成增删改查等操作。SQLite 数据库的优点是体积小、运行速度快,尤其适合一些嵌入式设备或需要大量读写的场景。

原文地址:https://mp.weixin.qq.com/s?__biz=MzkwMDE4NzE5OQ==&mid=2247502957&idx=1&sn=94c5511fcf4f0dcd7a750186103c0653

延伸 · 阅读

精彩推荐