提到数据库,大家第一时间想到的可能是 sql 数据库,这种数据库非常好用,但是对于新手就不是很容易上手,需要熟悉一段时间才可以大概掌握。这种数据库在大型的项目开发过程中用到的地方不会很多,但是一些普通的项目还是很实用的,大大减少了代码量。
sqlite3 数据库是 Python 自带的数据库,甚至不需要额外安装模块,而且操作简单。
Python + Mysql = SQLite
但是这种数据库在网上的教程很少,因为我也是最近才知道,所以找了很长时间的资料,
今天终于找齐了,来总结一下。真的找了好长时间啊
1 |
import sqlite3 |
首先打开我们的编译器(推荐使用vscode,因为数据库文件后缀是 .db,vscode里显示比较清楚)
先放上代码
1 |
conn = sqlite3.connect('test.db') |
使用 sqlite3 的 connect 函数可以创建数据库或者连接数据库,
如果这个数据库存在,就连接这个数据库,
如果这个库不存在,就创建数据库。
右边小括号里的是数据库名。
我们把这个保存在变量 conn 里,这个变量名可以自己起
公式:
变量名 = sqlite3.connect( '你要的数据库名.db' )
还是先放代码
1 |
cur = conn.cursor() |
在 2.1 中,我们创建了与数据库的连接,我们现在还需要一个游标来执行 sql 命令,
所以我们要用 conn 的 cursor 函数创建一个游标。
conn 就是刚才 2.1 中创建的保存数据库的变量,你要用你定义的变量名,
再定义一个变量表示这个游标。
公式:
变量 = 数据库变量.cursor()
先放代码
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import sqlite3 def check(db_name,table_name): conn = sqlite3.connect(db_name) cursor = conn.cursor() sql = '''SELECT tbl_name FROM sqlite_master WHERE type = 'table' ''' cursor.execute(sql) values = cursor.fetchall() tables = [] for v in values: tables.append(v[0]) if table_name not in tables: return False # 可以建表 else: return True # 不能建表 conn = sqlite3.connect('数据库名.db') #创建一个游标 cursor cur = conn.cursor() if (check("数据库名.db","Table名") == False): sql_text_1 = '''CREATE TABLE 你的table名 (xx XX, xx XX);''' # 执行sql语句 cur.execute(sql_text_1) |
代码比较长,因为我把 2.1 和 2.2 的两行代码也加了上去。
我们需要 table(表)来存数据,创建 table 的代码如下:
1 2 3 4 5 |
变量名 = '''CREATE TABLE 你的table名 (xx XX, xx XX);''' # 执行sql语句 cur.execute(上面的变量名) |
上面是用一个变量保存了创建 table 的 sql 语句,
下面使用 cur(就是刚才那个游标).execute() 函数执行创建 table 的语句。
我们也可以使用 cur.executemany 函数来同时执行多条 sql 语句。
sql 语句的内容,比如创建表的格式是上面这样的
格式:' ' 'CREATE TABLE 你的 table 名 (xx XX, xx XX);' ' '
这行代码里小写的 xx 就是你要的属性名,比如你的数据库是这样
姓名 | 班级 |
张三 | 1 |
其中,属性名就是 “姓名” 和 “班级”,
小写的 xx 就应该分别写姓名和班级(注意,不带引号)
后面的大写的XX就是这个属性所接受的数据的类型,
就相当于Python中的 int 类型和 str 类型。
只不过,我们在 sql 语句中,把 int 类型改成了 NUMBER,把 str 类型改成了 TEXT。
当我们运行这个代码,我们的文件夹目录里会多出来一个 你的数据库名.db 文件
当我们再次运行,会发现程序报错了。
报错信息的大概意思是:table 已经存在了。
这是因为我们第一次运行时已经创建了 table ,我们再次运行时,
程序会再次创建同名 table,就会报错。
所以,我们创建 table 之前要判断一下这个table存不存在,如果存在就不创建,如果不存在就创建
这个判断我把它写成了一个函数,就是我上面代码那个 check 函数。
这一步也是我想了好长时间,还找资料找了好长时间才知道的
先放代码
1 2 |
cur.executemany('INSERT INTO 你的table名 VALUES (?,?)', data) conn.commit() |
其中第一行代码中 executemany() 函数的意思就是同时执行多个 sql 语句。
这个函数的括号里写的逗号前面就是插入数据的 sql 语句,后面 data 可以是一个列表或者元组。注意,如果是列表的话,必须是列表里面有若干个元组的形式。
插入数据的 sql 语句的使用:
INSERT INTO 你的table名 VALUES (若干个逗号,用逗号分割)
这里我们要插入 data 这些数据,所以在括号里我们使用问号 “?” 来代替这个元素。
大家可以回去看一下 2.3 创建table 的讲解,在2.3中,我们创建了两个属性,分别是 “姓名” 和 “班级”。因为我们有两个属性,所以要有两个问号。
先放代码
1 2 3 4 |
def find_tb(): cur.execute("select * from 你的table名") # 提取查询到的数据 return cur.fetchall() |
这个就很简单了,我写这个函数使用时可以把你那个table里的所有数据都取出来。
第一行是查找table的 sql 语句,格式是:
select * from 你的table名
下一行再用 fetchall() 函数把数据提取出来,直接 return 即可。
好了,前面的东西大家应该也都看完了,来点 demo ?
顺便说一下,我这个 demo 的灵感来源是最近在网上刷到很多高考查分的视频,恰好最近在做这个数据库,所以说我这个 demo 受考试的启发,就做了一个学生分数系统,其实这个特别爽,可以把自己的分数改成全部满分!虽然实际没啥用,但是还是很爽的
代码里没有我没讲过的部分,大家可以对照上面的讲解看代码,VScode无报错运行。
对了,如果有看不懂的可以私信我,不出意外的话一天之内就能回复。
代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
import sqlite3 import os def check(db_name,table_name): conn = sqlite3.connect(db_name) cursor = conn.cursor() sql = '''SELECT tbl_name FROM sqlite_master WHERE type = 'table' ''' cursor.execute(sql) values = cursor.fetchall() tables = [] for v in values: tables.append(v[0]) if table_name not in tables: return False # 可以建表 else: return True # 不能建表 def find_tb(): cur.execute("select * from scores") # 提取查询到的数据 return cur.fetchall() def zcd(): os.system('cls') print("学生分数管理系统") print("1.增加学生分数信息") print("2.查看全部学生分数") print("3.查询分数段内学生分数") print("4.退出")
if __name__ == '__main__': # 创建与数据库的连接 conn = sqlite3.connect('stuents_scores.db') #创建一个游标 cursor cur = conn.cursor() # 如果没有表则执行建表的sql语句 if (check("stuents_scores.db","scores") == False): sql_text_1 = '''CREATE TABLE scores (姓名 TEXT, 班级 TEXT, 性别 TEXT, 语文 NUMBER, 数学 NUMBER, 英语 NUMBER, 总分 NUMBER);''' # 执行sql语句 cur.execute(sql_text_1) zcd() while True: op = int(input("请输入:")) if op == 1: S_name = input("请输入要添加的学生的姓名(如:张三):") S_class = input("请输入要添加的学生的班级(如:一班):") S_xb = input("请输入该学生性别:") S_Chinese = int(input("请输入该学生语文成绩(只输入一个数字,如:82):")) S_Maths = int(input("请输入该学生数学成绩(只输入一个数字,如:95):")) S_English = int(input("请输入该学生英语成绩(只输入一个数字,如:98):")) S_gj = S_Maths+S_Chinese+S_English # 总分 data = [(S_name, S_class, S_xb, S_Chinese, S_Maths, S_English,S_gj)] cur.executemany('INSERT INTO scores VALUES (?,?,?,?,?,?,?)', data) conn.commit() # cur.close() # conn.close() print("成功!") os.system('pause') os.system('cls') zcd() elif op == 2: info_list = find_tb() print("全部学生信息(排名不分前后):") for i in range(len(info_list)): print("第"+str(i+1)+"个:") print("学生姓名:"+str(info_list[i][0])) print("学生班级:"+str(info_list[i][1])) print("学生性别:"+str(info_list[i][2])) print("学生语文成绩:"+str(info_list[i][3])) print("学生数学成绩:"+str(info_list[i][4])) print("学生英语成绩:"+str(info_list[i][5])) print("学生总成绩:"+str(info_list[i][6])) os.system('pause') os.system('cls') zcd() elif op == 3: info_list = find_tb() fen = int(input("你要要查询总成绩高于n分的学生, 请输入n:")) for i in range(len(info_list)): if info_list[i][6] >= fen: print("查询结果:") print("第"+str(i+1)+"个:") print("学生总成绩:"+str(info_list[i][6])) os.system('pause') os.system('cls') zcd() elif op == 4: os.system('cls') break |