Python如何使用sqlite3数据库？

sqlite3数据库

sqlite3 数据库是 Python 自带的数据库，甚至不需要额外安装模块，而且操作简单。

Python + Mysql = SQLite

1.需要的模块（只有一个）

import sqlite3

2.模块的使用

首先打开我们的编译器（推荐使用vscode，因为数据库文件后缀是 .db，vscode里显示比较清楚）

2.1创建与数据库的连接

先放上代码

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

使用 sqlite3 的 connect 函数可以创建数据库或者连接数据库，

如果这个数据库存在，就连接这个数据库，

如果这个库不存在，就创建数据库。

右边小括号里的是数据库名。

我们把这个保存在变量 conn 里，这个变量名可以自己起

公式：

变量名 = sqlite3.connect( '你要的数据库名.db' )

2.2创建游标

还是先放代码

cur = conn.cursor()

在 2.1 中，我们创建了与数据库的连接，我们现在还需要一个游标来执行 sql 命令，

所以我们要用 conn 的 cursor 函数创建一个游标。

conn 就是刚才 2.1 中创建的保存数据库的变量，你要用你定义的变量名，

再定义一个变量表示这个游标。

公式：

变量 = 数据库变量.cursor()

2.3创建 table

先放代码

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 的代码如下：

变量名 = '''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 就是你要的属性名，比如你的数据库是这样

其中，属性名就是 “姓名” 和 “班级”，

小写的 xx 就应该分别写姓名和班级（注意，不带引号）

后面的大写的XX就是这个属性所接受的数据的类型，

就相当于Python中的 int 类型和 str 类型。

只不过，我们在 sql 语句中，把 int 类型改成了 NUMBER，把 str 类型改成了 TEXT。

当我们运行这个代码，我们的文件夹目录里会多出来一个 你的数据库名.db 文件

当我们再次运行，会发现程序报错了。

报错信息的大概意思是：table 已经存在了。

这是因为我们第一次运行时已经创建了 table ，我们再次运行时，

程序会再次创建同名 table，就会报错。

所以，我们创建 table 之前要判断一下这个table存不存在，如果存在就不创建，如果不存在就创建

这个判断我把它写成了一个函数，就是我上面代码那个 check 函数。

这一步也是我想了好长时间，还找资料找了好长时间才知道的

2.4插入数据

先放代码

cur.executemany('INSERT INTO 你的table名 VALUES (?,?)', data)
conn.commit()

其中第一行代码中 executemany() 函数的意思就是同时执行多个 sql 语句。

这个函数的括号里写的逗号前面就是插入数据的 sql 语句，后面 data 可以是一个列表或者元组。注意，如果是列表的话，必须是列表里面有若干个元组的形式。

插入数据的 sql 语句的使用：

INSERT INTO 你的table名 VALUES (若干个逗号，用逗号分割)

这里我们要插入 data 这些数据，所以在括号里我们使用问号 “?” 来代替这个元素。

大家可以回去看一下 2.3 创建table 的讲解，在2.3中，我们创建了两个属性，分别是 “姓名” 和 “班级”。因为我们有两个属性，所以要有两个问号。

2.5查找数据

先放代码

def find_tb():
    cur.execute("select * from 你的table名")
    # 提取查询到的数据
    return cur.fetchall()

这个就很简单了，我写这个函数使用时可以把你那个table里的所有数据都取出来。

第一行是查找table的 sql 语句，格式是：

select * from 你的table名

下一行再用 fetchall() 函数把数据提取出来，直接 return 即可。

代码：

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