我们经过第一阶段的学习,对python有了初步的应用。现在让我们用之前所学的做一个学生信息管理系统。
具体看到这个我们就应该想到用面向对象的方式来做这个系统。
具体的代码实现如下:
from prettytable import PrettyTable
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):
"""
学生信息管理类:
1. 学生信息录入
2. 学生信息浏览
3. 学生信息查询
4. 学生信息排序
5. 学生信息的删除
6. 学生信息的修改
"""
# 全局变量, 存储所有的学生信息;列表中的每个元素是学生对象
students = []
def __init__(self):
filename = 'student.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 find(self, name):
for student in self.students:
if student.name == name:
return student
else:
return None
def add(self):
"""
学生信息录入
:return:
"""
sid = input("学号: ")
name = input("姓名: ")
age = input("年龄: ")
gender = input("性别: ")
birth = input("出生年月: ")
address = input("地址: ")
tel = input("电话: ")
mail = input("邮箱地址: ")
s=Student(sid, name, age, gender, birth, address, tel, mail)
self.students.append(s)
def show(self):
"""
学生信息浏览
:return:
"""
print("学生信息显示".center(50, "*"))
talbe = PrettyTable()
talbe.field_names = ["学号", '姓名', '年龄', '性别', '出生年月', '地址', '电话', '邮箱']
for student in self.students:
talbe.add_row([student.sid, student.name, student.age, student.gender, student.birth,
student.address,student.tel,student.mail])
print(talbe)
def search(self, key,value):
"""
学生信息查询
:param key in ['sid','name']: 学生信息查询的关键字, 判断是学号还是姓名
:return:
"""
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):
"""
学生信息排序
:param key:
:return:
"""
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):
"""
学生信息的删除
:return:
"""
name = input("删除的学生姓名:").strip()
student = self.find(name)
if student:
if student.name == name:
self.students.remove(student)
print("删除学生成功")
else:
print("学生%s不存在" % (name))
def modify(self, key, oldvalue, newvalue):
"""
学生信息的修改
:return:
"""
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):
"""
将学生信息写入文件中
:return:
"""
filename = 'student.csv'
f = open(filename, 'w')
for student in self.students:
f.write(str(student) + '\n')
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':
manager.delete()
elif choice == '6':
key = input("修改(sid-根据学号, name-根据姓名): ")
value1 = input("旧的属性值: ")
value2 = input("新的属性值: ")
manager.modify(key, value1, value2)
elif choice == '7':
exit(0)
else:
print("请输入正确的选择!")
main()
我们来看下功能的实现:
首先我们录入几个学生的信息,然后利用浏览功能展示出来:
当然不想全部浏览的话,我们系统提供了单独查询的功能,有按照学号和按照姓名两种查询方式。
系统功能中还提供例排序功能,也有按照学号排序和按照年龄排序两种方式。
当不小心输错学生信息怎么办,别着急,系统还有我们的修改学生信息功能。
最后就是删除功能了。
当我们删除掉一个学生后在进行浏览就没有这个学生例了。
这样我们面向对象的学生信息管理系统就做好了。