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('錯誤輸入')