Python實現簡單的學生信息管理系統

import os
 
class Student(object):
    def __init__(self, sid, name, age=None, gender=None,
                 birth=None, address=None, tel=None, mail=None):
        self.sid = sid
        self.name = name
        self.age = age
        self.gender = gender
        self.birth = birth
        self.address = address
        self.tel = tel
        self.mail = mail
 
    def __str__(self):
        return "%s：%s：%s：%s：%s：%s：%s：%s" %(self.sid, self.name, self.age, self.gender,
                                    self.birth, self.address, self.tel, self.mail)
 
class StudentManage(object):
    students = []
    def __init__(self):
        filename = 'students.csv'
        if os.path.exists(filename):
            with open(filename) as f:
                for line in f:
                    info = line.split(':')
                    studentObj = Student(*info)
                    self.students.append(studentObj)
            print("加載緩存的學生信息成功")
        else:
            print("無緩存的學生信息！")
 
    def add(self):
        sid = input("學號： ")
        name = input("姓名： ")
        age = input("年齡： ")
        gender = input("性別（1-男 2-女）： ")
        birth = input("出生日期（年-月-日）： ")
        address = input("地址： ")
        tel = input("聯系電話： ")
        mail = input("電子郵箱： ")
        stu = Student(sid, name, age, gender, birth, address, tel, mail)
        self.students.append(stu)
        print("錄入學生[%s]信息成功" %(name))
 
    def show(self):
        print("學生信息展示".center(50, '*'))
        for student in self.students:
            print('-' * 50)
            print(str(student))
 
    def search(self, key, value):
        if key in ['sid', 'name']:
            for student in self.students:
                if getattr(student, key) == value:
                    print(str(student))
            else:
                print("未找到符合條件的學生信息！")
        else:
            print("系統隻支持根據[學號]和[姓名]查詢！")
 
    def sort(self, key):
        # 根據指定的key值進行排序
        sorted_students = sorted(self.students, key=lambda x: int(getattr(x, key)) if x else None)
        for student in sorted_students:
            print(student)
 
    def delete(self, key, value):
        if key in ['sid', 'name']:
            for student in self.students:
                if getattr(student, key) == value:
                    self.students.remove(student)
                    print("刪除屬性值[%s]為[%s]的學生信息" % (key, value))
            else:
                print("未找到符合條件的學生信息!")
        else:
            print("當前系統隻支持根據 [學號] 和 [姓名] 刪除!")
 
    def modify(self, key, oldvalue, newvalue):
        if key in ['sid', 'name']:
            for student in self.students:
                if getattr(student, key) == oldvalue:
                    setattr(student, key, newvalue)
                    print("修改屬性值[%s]為[%s]的學生信息為[%s]成功" % (key, oldvalue, newvalue))
            else:
                print("未找到符合條件的學生信息!")
        else:
            print("當前系統隻支持根據 [學號] 和 [姓名] 修改!")
 
    def save(self):
        filename = 'students.csv'
        f = open(filename, 'w')
        for student in self.students:
            f.write(str(student) + '\n')
        print("寫入學生信息成功!")
 
        f.close()
 
def main():
    manager = StudentManage()
    while True:
        menu = """
                        學生信息管理系統
            1. 學生信息錄入
            2. 學生信息瀏覽
            3. 學生信息查詢
            4. 學生信息排序
            5. 學生信息的刪除
            6. 學生信息的修改
            7. 退出
        請輸入正確的選擇:    """
 
        choice = input(menu)
        if choice == '1':
            manager.add()
            manager.save()
        elif choice == '2':
            manager.show()
        elif choice == '3':
            key = input("查詢(sid-根據學號, name-根據姓名): ")
            value = input("查詢值: ")
            manager.search(key, value)
        elif choice == '4':
            key = input("排序(sid-根據學號, age-根據年齡): ")
            manager.sort(key)
        elif choice == '5':
            key = input("刪除(sid-根據學號, name-根據姓名): ")
            value = input("刪除的屬性值: ")
            manager.delete(key, value)
        elif choice == '6':
            key = input("修改(sid-根據學號, name-根據姓名): ")
            value1 = input("舊的屬性值: ")
            value2 = input("新的屬性值: ")
            manager.modify(key, value1, value2)
        elif choice == '7':
            exit(0)
        else:
            print("請輸入正確的選擇!")
 
main()