Python如何操作SQLite3增删改查数据

SQLite3是轻量级的关系型数据库，它的数据库文件是一个独立的文件，可以方便地进行备份和传输

import sqlite3

# 创建数据库和表
def create_table():
    conn = sqlite3.connect('student.db')
    cursor = conn.cursor()
    cursor.execute('''CREATE TABLE IF NOT EXISTS student
                      (id INTEGER PRIMARY KEY AUTOINCREMENT,
                       name TEXT NOT NULL,
                       age INTEGER NOT NULL)''')
    conn.commit()
    conn.close()

# 添加学生信息
def add_student():
    name = input('请输入学生姓名：')
    age = input('请输入学生年龄：')
    conn = sqlite3.connect('student.db')
    cursor = conn.cursor()
    cursor.execute('INSERT INTO student (name, age) VALUES (?, ?)', (name, age))
    conn.commit()
    conn.close()
    print('添加成功！')

# 删除学生信息
def delete_student():
    name = input('请输入要删除的学生姓名：')
    conn = sqlite3.connect('student.db')
    cursor = conn.cursor()
    cursor.execute('DELETE FROM student WHERE name = ?', (name,))
    conn.commit()
    conn.close()
    print('删除成功！')

# 修改学生信息
def modify_student():
    name = input('请输入要修改的学生姓名：')
    age = input('请输入学生年龄：')
    conn = sqlite3.connect('student.db')
    cursor = conn.cursor()
    cursor.execute('UPDATE student SET age = ? WHERE name = ?', (age, name))
    conn.commit()
    conn.close()
    print('修改成功！')

# 查询学生信息
def query_student():
    name = input('请输入要查询的学生姓名：')
    conn = sqlite3.connect('student.db')
    cursor = conn.cursor()
    cursor.execute('SELECT * FROM student WHERE name = ?', (name,))
    result = cursor.fetchone()
    conn.close()
    if result:
        print('姓名：', result[1])
        print('年龄：', result[2])
    else:
        print('未找到该学生！')

# 主函数
def main():
    create_table()
    while True:
        print('1. 添加学生信息')
        print('2. 删除学生信息')
        print('3. 修改学生信息')
        print('4. 查询学生信息')
        print('5. 退出程序')
        choice = input('请输入您的选择：')
        if choice == '1':
            add_student()
        elif choice == '2':
            delete_student()
        elif choice == '3':
            modify_student()
        elif choice == '4':
            query_student()
        elif choice == '5':
            break
        else:
            print('输入有误，请重新输入！')

if __name__ == '__main__':
    main()