Python+sqlite 一个简单的学生选课系统

# -*- coding: utf-8 -*-
# @Time    : 2020/4/23/0023
# @Author  : Yann
# @File    : t3.py
# @Software: PyCharm
import sqlite3

conn = sqlite3.connect(r'D:\test.db')
c = conn.cursor()
# 一个简易的学生选课系统
course_name = ''
course_teacher = ''


# 写一个从键盘获得课程信息的函数
def getCour():
    name = input("请输入课程名")
    teacher = input("请输入教师名")
    return [name, teacher]


# 添加课程到数据库，此为功能1
def insertCour():
    list = getCour()
    course_name = list[0]
    course_teacher = list[1]
    sql = 'INSERT INTO COURSE (course_name,course_teacher) VALUES (:name,:teacher)'
    c.execute(sql, {'name': course_name, 'teacher': course_teacher})
    conn.commit()
    print('添加成功')


# 修改课程内容，此为功能2
def updateCour():
    print('现有的课程为：')
    sql = "SELECT * FROM COURSE"
    result = c.execute(sql)
    for i in result:
        print(i)
    id = int(input('请输入要修改的课程的id'))
    new_name = input('请输入新的课程名')
    new_teacher = input('请输入新的教师名')
    sql = 'UPDATE COURSE SET course_name = (:name),course_teacher=(:teacher) WHERE id = (:id)'
    c.execute(sql, {'name': new_name, 'teacher': new_teacher, 'id': id})
    conn.commit()
    print('修改成功，现有的课程为：')
    sql = "SELECT * FROM COURSE"
    result = c.execute(sql)
    for i in result:
        print(i)


# 删除一条课程记录，此为内容3
def delCour():
    print('现有的课程为：')
    sql = "SELECT * FROM COURSE"
    result = c.execute(sql)
    for i in result:
        print(i)
    id = int(input('请输入要删除的课程的id'))
    sql = 'DELETE FROM COURSE WHERE id = (:id)'
    c.execute(sql, {'id': id})
    print('删除成功，现有的课程为：')
    sql = "SELECT * FROM COURSE"
    result = c.execute(sql)
    conn.commit()
    for i in result:
        print(i)


def indexCour():
    sql = 'SELECT * FROM COURSE'
    result = c.execute(sql)
    print('课程id\t课程名称\t授课教师')
    for i in result:
        print('\t' + str(i[0]) + '\t' + i[1] + '\t' + i[2])


if __name__ == '__main__':
    while True:
        choice = int(input('请选择功能：1-添加课程，2-修改课程，3-删除课程，4-查询所有课程'))
        if choice == 1:
            insertCour()
        elif choice == 2:
            updateCour()
        elif choice == 3:
            delCour()
        elif choice == 4:
            indexCour()
        else:
            print('错误输入')