3.4 定义系统功能函数
3.4.1 添加功能
# 添加学员函数内部需要创建学员对象,故先导入student模块
from student import *
class StudentManager(object):
......
# 2.2 添加学员
def add_student(self):
# 1. 用户输入姓名、性别、手机号
name = input('请输入您的姓名:')
gender = input('请输入您的性别:')
tel = input('请输入您的手机号:')
# 2. 创建学员对象:先导入学员模块,再创建对象
student = Student(name, gender, tel)
# 3. 将该学员对象添加到列表
self.student_list.append(student)
# 打印信息
print(self.student_list)
print(student)
3.4.2 删除学员
- 需求:用户输入目标学员姓名,如果学员存在则删除该学员。
- 步骤
- 用户输入目标学员姓名
- 遍历学员数据列表,如果用户输入的学员姓名存在则删除,否则则提示该学员不存在。
- 代码
# 2.3 删除学员:删除指定姓名的学员
def del_student(self):
# 1. 用户输入目标学员姓名
del_name = input('请输入要删除的学员姓名:')
# 2. 如果用户输入的目标学员存在则删除,否则提示学员不存在
for i in self.student_list:
if i.name == del_name:
self.student_list.remove(i)
break
else:
print('查无此人!')
# 打印学员列表,验证删除功能
print(self.student_list)
3.4.3 修改学员信息
- 需求:用户输入目标学员姓名,如果学员存在则修改该学员信息。
- 步骤
- 用户输入目标学员姓名;
- 遍历学员数据列表,如果用户输入的学员姓名存在则修改学员的姓名、性别、手机号数据,否则则提示该学员不存在。
- 代码
# 2.4 修改学员信息
def modify_student(self):
# 1. 用户输入目标学员姓名
modify_name = input('请输入要修改的学员的姓名:')
# 2. 如果用户输入的目标学员存在则修改姓名、性别、手机号等数据,否则提示学员不存在
for i in self.student_list:
if i.name == modify_name:
i.name = input('请输入学员姓名:')
i.gender = input('请输入学员性别:')
i.tel = input('请输入学员手机号:')
print(f'修改该学员信息成功,姓名{i.name},性别{i.gender}, 手机号{i.tel}')
break
else:
print('查无此人!')
3.4.5 查询学员信息
- 需求:用户输入目标学员姓名,如果学员存在则打印该学员信息
- 步骤
- 用户输入目标学员姓名
- 遍历学员数据列表,如果用户输入的学员姓名存在则打印学员信息,否则提示该学员不存在。
- 代码
# 2.5 查询学员信息
def search_student(self):
# 1. 用户输入目标学员姓名
search_name = input('请输入要查询的学员的姓名:')
# 2. 如果用户输入的目标学员存在,则打印学员信息,否则提示学员不存在
for i in self.student_list:
if i.name == search_name:
print(f'姓名{i.name},性别{i.gender}, 手机号{i.tel}')
break
else:
print('查无此人!')
3.4.6 显示所有学员信息
# 2.6 显示所有学员信息
def show_student(self):
print('姓名\t性别\t手机号')
for i in self.student_list:
print(f'{i.name}\t{i.gender}\t{i.tel}')
3.4.7 保存学员信息
- 需求:将修改后的学员数据保存到存储数据的文件。
- 步骤
思考
- 文件写入的数据是学员对象的内存地址吗?
- 文件内数据要求的数据类型是什么?
class A(object):
a = 0
def __init__(self):
self.b = 1
aa = A()
# 返回类内部所有属性和方法对应的字典
print(A.__dict__)
# 返回实例属性和值组成的字典
print(aa.__dict__)
在Python中
# 2.7 保存学员信息
def save_student(self):
# 1. 打开文件
f = open('student.data', 'w')
# 2. 文件写入学员数据
# 注意1:文件写入的数据不能是学员对象的内存地址,需要把学员数据转换成列表字典数据再做存储
new_list = [i.__dict__ for i in self.student_list]
# [{'name': 'aa', 'gender': 'nv', 'tel': '111'}]
print(new_list)
# 注意2:文件内数据要求为字符串类型,故需要先转换数据类型为字符串才能文件写入数据
f.write(str(new_list))
# 3. 关闭文件
f.close()
3.4.8 加载学员信息
-
需求:每次进入系统后,修改的数据是文件里面的数据
-
步骤
- 尝试以
"r"
模式打开学员数据文件,如果文件不存在则以 "w"
模式打开文件
- 如果文件存在则读取数据并存储数据
- 读取数据
- 转换数据类型为列表并转换列表内的字典为对象
- 存储学员数据到学员列表
- 关闭文件
-
代码
# 2.8 加载学员信息
def load_student(self):
# 尝试以"r"模式打开数据文件,文件不存在则提示用户;文件存在(没有异常)则读取数据
try:
f = open('student.data', 'r')
except:
f = open('student.data', 'w')
else:
# 1. 读取数据
data = f.read()
# 2. 文件中读取的数据都是字符串且字符串内部为字典数据,故需要转换数据类型再转换字典为对象后存储到学员列表
new_list = eval(data)
self.student_list = [Student(i['name'], i['gender'], i['tel']) for i in new_list]
finally:
# 3. 关闭文件
f.close()
四、总结
- 函数
- 面向对象
- 定义类
- 创建对象
- 定义和调用实例属性
- 定义和调用实例方法
- 数据类型
- 文件操作