用python實現學生資訊管理系統

filename = 'student.txt' #用於儲存學生資訊的txt檔名
import os #匯入os模組，用於判斷路徑下檔案是否存在
#定義主函數，主函數的作用是通過選擇不同的選項進入不同的功能
def main():
    while True:
        menu()
        choice = int(input('請選擇'))
        if choice in [0, 1, 2, 3, 4, 5, 6, 7]:
            if choice == 0:
                answer = input('您確定要退出系統嗎?y/n')
                if answer == 'y' or answer == 'Y':
                    print('謝謝您的使用.')
                    break
                else:
                    continue
            elif choice == 1: insert()
            elif choice == 2: search()
            elif choice == 3: delete()
            elif choice == 4: modify()
            elif choice == 5: sort()
            elif choice == 6: total()
            elif choice == 7: show()
#定義選單函數，沒什麼實際作用的函數，主要是為了讓使用者看明白應該怎麼操作
def menu():
    print('==========學生資訊管理系統==========')
    print('-------------功能選單-------------')
    print('1.錄入學生資訊')
    print('2.查詢學生資訊')
    print('3.刪除學生資訊')
    print('4.修改學生資訊')
    print('5.排序')
    print('6.統計學生資訊')
    print('7.顯示所有資訊')
    print('0.退出')
    print('--------------------------------')
#定義學生資訊新增函數
def insert():
    student_lst = [] #用於儲存學生資訊的列表
    while True:
        id = input('請輸入ID(如1001):')
        if not id: break #如果id為空值，就停止
        name = input('請輸入姓名:')
        if not name: break
        try:
            english = int(input('請輸入英語成績:'))
            python = int(input('請輸入Python成績:'))
            java = int(input('請輸入Java成績:'))
        except: #意思是如果上面的成績不是int型別，就會要求重新輸入
            print('輸入無效,不是整數型別,請重新輸入.')
        student = {'ID': id, 'Name': name, 'English': english, 'Python': python, 'Java': java} #將新的學生資訊生成字典
        student_lst.append(student) #將單條學生資訊新增到列表後面
        save(student_lst) #save函數在後面有定義
        answer = input('是否繼續新增學生資訊?y/nn')
        if answer == 'y': continue
        else:
            break
            print('學生資訊錄入完畢.')
#定義學生資訊儲存函數
def save(lst):
    try:
        stu_txt = open(filename, 'a', encoding = 'utf_8')
    except:
        stu_txt = open(filename, 'w', encoding='utf_8')
    for item in lst:
        stu_txt.write(str(item) + 'n')
    stu_txt.close()
#定義學生資訊搜尋函數，可以實現按照ID或者姓名搜尋
def search():
    while True:
        Method = int(input('請輸入查詢方法,1表示按ID查詢,2表示按姓名查詢.'))
        if Method != 1 and Method != 2:
            print('不是預設的查詢方法,請重新輸入.')
            search() #如果輸入值不是1和2，就會要求重新輸入，可以通過直接再次執行search()函數實現
        Inf = input('請按照查詢方法輸入要查詢的學生的資訊:') #輸入ID或者姓名
        with open(filename, 'r', encoding = 'utf-8') as sfile: #開啟儲存學生資訊的檔案，實際上在這之前應該有一步判斷檔案是否存在，懶得補了，前面套一層if os.path.exists(filename)就行，後面的函數中有類似的部分
            stu_ifo = sfile.readlines()
            d = {}
            flag = True #用於判斷最後是否找到學生資訊
            if Inf != '':
                for item in stu_ifo:
                    d = dict(eval(item)) #將檔案中的每一行資訊轉化為一個字典
                    if Method == 1 and d['ID'] == Inf:
                        show_student(d) #show_student函數在後面有定義，這裡也可以之際print(d)，但是輸出會很醜，就是直接把一個字典列印出來，不美觀
                        flag = False
                    elif Method == 2 and d['Name'] == Inf:
                        show_student(d)
                        flag = False
            else:
                print('沒有輸入正確的學生資訊,請重新輸入.')
                search()
        if flag: #flag的作用在這裡體現
            print('未查到學生資訊.')
        answer = input('是否繼續查詢學生資訊?y/nn')
        if answer == 'y': continue #continue會重新返回這個函數
        else: break #break會返回主函數
#輸出學生資訊的一個小函數，作用就是使輸出更美觀
def show_student(lst):
    if len(lst) == 0:
        print('沒有查詢到學生資訊,無資料顯示.')
        return
    #定義標題顯示格式
    format_title = '{:^6}t{:^12}t{:^8}t{:^10}t{:^10}t{:^8}'
    print(format_title.format('ID', 'Name', 'English', 'Python', 'Java', 'Sum'))
    #定義內容顯示格式
    format_data = '{:^6}t{:^12}t{:^8}t{:^10}t{:^10}t{:^8}'
    print(format_data.format(lst['ID'],
                             lst['Name'],
                             lst['English'],
                             lst['Python'],
                             lst['Java'],
                             int(lst['English']) + int(lst['Python']) + int(lst['Java'])))
#定義學生資訊刪除函數
def delete():
    while True:
        student_id = input('請輸入要刪除學生的ID:')
        if student_id != '':
            if os.path.exists(filename): #判斷檔案是否存在，如果檔案都不存在，那肯定是不能夠刪除的
                with open(filename, 'r', encoding = 'utf-8') as stu_file: #開啟檔案
                    student_old = stu_file.readlines()
            else:
                student_old = []
            flag = False #flag作用與上面的函數相同
            if student_old:
                with open(filename, 'w', encoding = 'utf-8') as wfile: #上面的stu_file是一個唯讀開啟方式，這裡的wfile是可以寫入的開啟方式，目的是儲存刪除後的學生資訊
                    d = {}
                    for item in student_old:
                        d = dict(eval(item))
                        if d['ID'] != student_id:
                            wfile.write(str(d) + 'n')
                        else:
                            flag = True
                    if flag:
                        print(f'ID為{student_id}的學生資訊已刪除.')
                    else:
                        print(f'沒有找到ID為{student_id}的學生.')
            else:
                print('沒有學生資訊.')
                break
            show() #show()函數在下面有定義，意思是展示所有學生資訊
            answer = input('是否繼續刪除?y/nn')
            if answer == 'y': continue
            else: break
#定義學生資訊修改函數
def modify():
    show()
    if os.path.exists(filename): #同上
        with open(filename, 'r', encoding = 'utf-8') as rfile: #唯讀方式開啟
            student_old = rfile.readlines()
    else:
        return #如果沒有學生資訊的檔案，就回到主函數
    student_id = input('請輸入要修改學生的ID:')
    with open(filename, 'w', encoding = 'utf-8') as wfile: #寫入方式開啟
        for item in student_old:
            d = dict(eval(item))
            flag = False #作用同上
            if d['ID'] == student_id:
                print('找到學生資訊,可以修改了.')
                try:
                    d['Name'] = input('請輸入新的姓名:')
                    d['English'] = int(input('請輸入新的English成績:'))
                    d['Python'] = int(input('請輸入新的Python成績:'))
                    d['Java'] = int(input('請輸入新的Java成績:'))
                except:
                    print('輸入有誤,請重新輸入.')
                wfile.write(str(d) + 'n')
            else:
                wfile.write(str(d) + 'n')
                flag = True
            if flag:
                print(f'未找到ID為{student_id}的學生資訊.')
        answer = input('是否繼續修改學生資訊?y/nn')
        if answer == 'y': modify()
#定義學生按成績排序函數
def sort():
    if os.path.exists(filename): #同上
        show()
        with open(filename, 'r', encoding = 'utf-8') as rfile: #唯讀方式開啟
            students = rfile.readlines()
        student_new = []
        for item in students:
            d = dict(eval(item))
            student_new.append(d)
    else:
        return
    aord = int(input('請選擇:0表示升序,1表示降序.'))
    if aord == 0:
        flag = False
    elif aord == 1:
        flag = True
    else:
        print('輸入有誤,請重新輸入.')
        sort()
    mode = int(input('請選擇排序方式:1英語,2Python,3Java,4Sum'))
    if mode == 1: #藉助lambda函數實現排序
        student_new.sort(key = lambda x :int(x['English']), reverse = flag)
    elif mode == 2:
        student_new.sort(key = lambda x: int(x['Python']), reverse = flag)
    elif mode == 3:
        student_new.sort(key = lambda x: int(x['Java']), reverse = flag)
    elif mode == 4:
        student_new.sort(key = lambda x: int(x['English'] + x['Python'] + x['Java']), reverse = flag)
    else:
        print('不是預設方法,請重新輸入.')
        sort()
    #下面是為了使輸出更加美觀定義的輸出函數
    format_title = '{:^6}t{:^12}t{:^8}t{:^10}t{:^10}t{:^8}'
    print(format_title.format('ID', 'Name', 'English', 'Python', 'Java', 'Sum'))
    # 定義內容顯示格式
    format_data = '{:^6}t{:^12}t{:^8}t{:^10}t{:^10}t{:^8}'
    for lst in student_new:
        print(format_data.format(lst['ID'],
                                 lst['Name'],
                                 lst['English'],
                                 lst['Python'],
                                 lst['Java'],
                                 int(lst['English']) + int(lst['Python']) + int(lst['Java'])))
#定義學生數量統計函數，不多解釋了，比較簡單
def total():
    if os.path.exists(filename):
        with open(filename, 'r', encoding = 'utf-8') as rfile:
            students = rfile.readlines()
            if students:
                print('一共有{}名學生.'.format(len(students)))
            else:
                print('還沒有錄入學生資訊.')
    else:
        print('暫未儲存資料資訊.')
#定義學生資訊展示函數
def show():
    if os.path.exists(filename):
        with open(filename, 'r', encoding = 'utf-8') as rfile:
            students = rfile.readlines()
            format_title = '{:^6}t{:^12}t{:^8}t{:^10}t{:^10}t{:^8}'
            format_data = '{:^6}t{:^12}t{:^8}t{:^10}t{:^10}t{:^8}'
            print(format_title.format('ID', 'Name', 'English', 'Python', 'Java', 'Sum'))
            d = {}
            if len(students) != 0:
                for item in students:
                    d = dict(eval(item))
                    print(format_data.format(d['ID'],
                                             d['Name'],
                                             d['English'],
                                             d['Python'],
                                             d['Java'],
                                             int(d['English']) + int(d['Python']) + int(d['Java'])))
            else:
                print('尚未錄入學生資訊.')
    else:
        print('尚未儲存學生資訊檔案.')