通讯录管理系统
这是一个基于文本的界面程序,用户可以通过命令行与之交互,它使用了CSV文件来存储和读取联系人信息,这使得数据可以持久化保存。此外,程序还提供了一些基本的输入验证,以确保输入的数据是有效的。
它包含了两个类PersonInformation
类用于创建和管理个人信息对象,而ContactList
类则用于管理这些对象的集合。
PersonInformation
类
这个类用于存储一个人的基本信息,包括姓名、性别、年龄、电话和住址。它有一个__str__
方法,用于定义当对象被转换为字符串时的表现形式。
class PersonInformation: def __init__(self, name, gender, age, telephone, address): self.name = name self.gender = gender self.age = age self.telephone = telephone self.address = address def __str__(self): return f"名字: {self.name}, 性别: {self.gender}, 年龄: {self.age}, 电话: {self.telephone}, 住址: {self.address}"
ContactList
类
这个类用于管理一个联系人列表。它提供了以下功能:
- 添加联系人(
add_contact
) - 删除联系人(
remove_contact
) - 查找联系人(
find_contact
) - 修改联系人信息(
modify_contact
) - 显示所有联系人(
show_contacts
) - 保存联系人到CSV文件(
save_contacts
) - 从CSV文件加载联系人(
load_contacts
)
class ContactList: def __init__(self): self.arr = [] def add_contact(self, person): self.arr.append(person) self.save_contacts() def is_empty(self): if not self.arr: print("当前联系人列表为空!") def remove_contact(self, name): person = self.find_contact(name) if person: self.arr.remove(person) self.save_contacts() print("删除成功!") self.is_empty() else: print("此联系人不存在!") def show_contacts(self): print('通讯录:') for person in self.arr: print(person) self.is_empty() def find_contact(self, name): for person in self.arr: if person.name == name: return person return None def modify_contact(self, old_name, new_info): person = self.find_contact(old_name) if person: person.name = new_info.name person.gender = new_info.gender person.age = new_info.age person.telephone = new_info.telephone person.address = new_info.address self.save_contacts() print("修改成功!\n ", person) else: print("此联系人不存在!") def save_contacts(self, filename="contacts.csv"): with open(filename, 'w', newline='', encoding='utf-8') as f: writer = csv.writer(f) writer.writerow(['姓名', '性别', '年龄', '电话', '住址']) for person in self.arr: writer.writerow([person.name, person.gender, person.age, person.telephone, person.address]) def load_contacts(self, filename="contacts.csv"): if not os.path.isfile(filename): with open(filename, 'w', newline='', encoding='utf-8') as f: writer = csv.writer(f) writer.writerow(['姓名', '性别', '年龄', '电话', '住址']) with open(filename, 'r', newline='', encoding='utf-8') as f: reader = csv.reader(f) next(reader) for row in reader: if len(row) == 5: name, gender, age, telephone, address = row self.add_contact(PersonInformation(name, gender, age, telephone, address)) else: print(f"跳过不完整的联系人记录: {row}")
menu
函数
这个函数用于显示程序的菜单选项,让用户可以选择执行不同的操作。
main
函数
这是程序的主入口,它首先创建一个ContactList
实例,然后进入一个无限循环,不断显示菜单并根据用户的选择调用相应的方法。当用户选择退出时,程序会结束。
程序的运行流程
- 程序启动后,首先加载已有的联系人信息。
- 显示菜单,等待用户输入选择。
- 根据用户的选择执行相应的操作:
- 添加联系人:输入新的联系人信息并添加到列表中。
- 删除联系人:输入要删除的联系人姓名,然后从列表中删除。
- 查找联系人:输入要查找的联系人姓名,并显示其信息。
- 修改联系人:输入要修改的联系人姓名,然后输入新的信息进行修改。
- 显示所有联系人:列出当前所有的联系人信息。
- 退出通讯录:退出程序。
完整代码
import os, csv class PersonInformation: def __init__(self, name, gender, age, telephone, address): self.name = name self.gender = gender self.age = age self.telephone = telephone self.address = address def __str__(self): return f"名字: {self.name}, 性别: {self.gender}, 年龄: {self.age}, 电话: {self.telephone}, 住址: {self.address}" class ContactList: def __init__(self): self.arr = [] def add_contact(self, person): self.arr.append(person) self.save_contacts() def is_empty(self): if not self.arr: print("当前联系人列表为空!") def remove_contact(self, name): person = self.find_contact(name) if person: self.arr.remove(person) self.save_contacts() print("删除成功!") self.is_empty() else: print("此联系人不存在!") def show_contacts(self): print('通讯录:') for person in self.arr: print(person) self.is_empty() def find_contact(self, name): for person in self.arr: if person.name == name: return person return None def modify_contact(self, old_name, new_info): person = self.find_contact(old_name) if person: person.name = new_info.name person.gender = new_info.gender person.age = new_info.age person.telephone = new_info.telephone person.address = new_info.address self.save_contacts() print("修改成功!\n ", person) else: print("此联系人不存在!") def save_contacts(self, filename="contacts.csv"): with open(filename, 'w', newline='', encoding='utf-8') as f: writer = csv.writer(f) writer.writerow(['姓名', '性别', '年龄', '电话', '住址']) for person in self.arr: writer.writerow([person.name, person.gender, person.age, person.telephone, person.address]) def load_contacts(self, filename="contacts.csv"): if not os.path.isfile(filename): with open(filename, 'w', newline='', encoding='utf-8') as f: writer = csv.writer(f) writer.writerow(['姓名', '性别', '年龄', '电话', '住址']) with open(filename, 'r', newline='', encoding='utf-8') as f: reader = csv.reader(f) next(reader) for row in reader: if len(row) == 5: name, gender, age, telephone, address = row self.add_contact(PersonInformation(name, gender, age, telephone, address)) else: print(f"跳过不完整的联系人记录: {row}") def menu(contact_list): print() print("*********************************") print("** 1、增加联系人 2、删除联系人 **") print("** 3、查找联系人 4、修改联系人 **") print("** 5、展示联系人 0、退出通讯录 **") print("*********************************") def main(): contact_list = ContactList() contact_list.load_contacts() while True: menu(contact_list) name, gender, age, telephone, address = [""]*5 input_choice = input("请输入您的选择:") if input_choice == '1': while len(name)<2: name = input("请输入您要添加的联系人的姓名:\n") if contact_list.find_contact(name): print(f'此姓名[{name}]已存在!') break while gender not in ('男','女','1','0'): gender = input("请输入性别(1-男, 0-女):\n") if gender=='1': gender = '男' elif gender=='0': gender = '女' while not age.isnumeric(): age = input("请输入年龄:\n") while telephone is None or len(telephone)!=11 or not telephone.startswith('1'): telephone = input("请输入手机号:\n") while len(address)<6: address = input("请输入住址(字数不少于6):\n") contact_list.add_contact(PersonInformation(name, gender, age, telephone, address)) print("联系人添加成功!") elif input_choice == '2': name = input("请输入您要删除的联系人的名字:\n") contact_list.remove_contact(name) elif input_choice == '3': name = input("请输入您要查找的联系人的名字:\n") person = contact_list.find_contact(name) print(person if person else "此联系人不存在!") elif input_choice == '4': old_name = input("请输入您要修改的联系人的名字:\n") new_info = PersonInformation(*([""]*5)) person = contact_list.find_contact(old_name) if person: print('待修改联系人:(直接回车保留原字段)\n ', person) while len(new_info.name)<2: new_info.name = input("请输入新的名字:\n") if new_info.name=="": new_info.name = person.name elif contact_list.find_contact(name): print(f'此姓名[{name}]已存在,退出修改!') break while new_info.gender not in ('男','女','1','0'): new_info.gender = input("请输入性别(1-男, 0-女):\n") if new_info.gender=='1': new_info.gender = '男' elif new_info.gender=='0': new_info.gender = '女' elif new_info.gender=="": new_info.gender = person.gender while not new_info.age.isnumeric(): new_info.age = input("请输入年龄:\n") if new_info.age=="": new_info.age = person.age while len(new_info.telephone)!=11 or not new_info.telephone.startswith('1'): new_info.telephone = input("请输入新的手机号:\n") if new_info.telephone=="": new_info.telephone = person.telephone while len(new_info.address)<6: new_info.address = input("请输入新的住址(字数不少于6):\n") if new_info.address=="": new_info.address = person.address contact_list.modify_contact(old_name, new_info) elif input_choice == '5': contact_list.show_contacts() elif input_choice == '0': print("成功退出通讯录!") break else: print("输入错误,请重新选择!") if __name__ == "__main__": main()
运行示例
********************************* ** 1、增加联系人 2、删除联系人 ** ** 3、查找联系人 4、修改联系人 ** ** 5、展示联系人 0、退出通讯录 ** ********************************* 请输入您的选择:5 通讯录: 当前联系人列表为空! ********************************* ** 1、增加联系人 2、删除联系人 ** ** 3、查找联系人 4、修改联系人 ** ** 5、展示联系人 0、退出通讯录 ** ********************************* 请输入您的选择:1 请输入您要添加的联系人的姓名: Hann 请输入性别(1-男, 0-女): 1 请输入年龄: 51 请输入手机号: 13962600000 请输入住址(字数不少于6): 江苏省昆山市 联系人添加成功! ********************************* ** 1、增加联系人 2、删除联系人 ** ** 3、查找联系人 4、修改联系人 ** ** 5、展示联系人 0、退出通讯录 ** ********************************* 请输入您的选择:5 通讯录: 名字: Hann, 性别: 男, 年龄: 51, 电话: 13962600000, 住址: 江苏省昆山市 ********************************* ** 1、增加联系人 2、删除联系人 ** ** 3、查找联系人 4、修改联系人 ** ** 5、展示联系人 0、退出通讯录 ** ********************************* 请输入您的选择:1 请输入您要添加的联系人的姓名: Hann 此姓名[Hann]已存在! ********************************* ** 1、增加联系人 2、删除联系人 ** ** 3、查找联系人 4、修改联系人 ** ** 5、展示联系人 0、退出通讯录 ** ********************************* 请输入您的选择:1 请输入您要添加的联系人的姓名: YangCheng 请输入性别(1-男, 0-女): 1 请输入年龄: 22 请输入手机号: 13962611111 请输入住址(字数不少于6): 江苏省昆山市 联系人添加成功! ********************************* ** 1、增加联系人 2、删除联系人 ** ** 3、查找联系人 4、修改联系人 ** ** 5、展示联系人 0、退出通讯录 ** ********************************* 请输入您的选择:5 通讯录: 名字: Hann, 性别: 男, 年龄: 51, 电话: 13962600000, 住址: 江苏省昆山市 名字: YangCheng, 性别: 男, 年龄: 22, 电话: 13962611111, 住址: 江苏省昆山市 ********************************* ** 1、增加联系人 2、删除联系人 ** ** 3、查找联系人 4、修改联系人 ** ** 5、展示联系人 0、退出通讯录 ** ********************************* 请输入您的选择:4 请输入您要修改的联系人的名字: Hann 待修改联系人:(直接回车保留原字段) 名字: Hann, 性别: 男, 年龄: 51, 电话: 13962600000, 住址: 江苏省昆山市 请输入新的名字: HannYang 请输入性别(1-男, 0-女): 请输入年龄: 请输入新的手机号: 请输入新的住址(字数不少于6): 修改成功! 名字: HannYang, 性别: 男, 年龄: 51, 电话: 13962600000, 住址: 江苏省昆山市 ********************************* ** 1、增加联系人 2、删除联系人 ** ** 3、查找联系人 4、修改联系人 ** ** 5、展示联系人 0、退出通讯录 ** ********************************* 请输入您的选择:5 通讯录: 名字: HannYang, 性别: 男, 年龄: 51, 电话: 13962600000, 住址: 江苏省昆山市 名字: YangCheng, 性别: 男, 年龄: 22, 电话: 13962611111, 住址: 江苏省昆山市 ********************************* ** 1、增加联系人 2、删除联系人 ** ** 3、查找联系人 4、修改联系人 ** ** 5、展示联系人 0、退出通讯录 ** ********************************* 请输入您的选择:3 请输入您要查找的联系人的名字: hannyang 此联系人不存在! ********************************* ** 1、增加联系人 2、删除联系人 ** ** 3、查找联系人 4、修改联系人 ** ** 5、展示联系人 0、退出通讯录 ** ********************************* 请输入您的选择:3 请输入您要查找的联系人的名字: HannYang 名字: HannYang, 性别: 男, 年龄: 51, 电话: 13962600000, 住址: 江苏省昆山市 ********************************* ** 1、增加联系人 2、删除联系人 ** ** 3、查找联系人 4、修改联系人 ** ** 5、展示联系人 0、退出通讯录 ** ********************************* 请输入您的选择:2 请输入您要删除的联系人的名字: hann 此联系人不存在! ********************************* ** 1、增加联系人 2、删除联系人 ** ** 3、查找联系人 4、修改联系人 ** ** 5、展示联系人 0、退出通讯录 ** ********************************* 请输入您的选择:2 请输入您要删除的联系人的名字: HannYang 删除成功! ********************************* ** 1、增加联系人 2、删除联系人 ** ** 3、查找联系人 4、修改联系人 ** ** 5、展示联系人 0、退出通讯录 ** ********************************* 请输入您的选择:5 通讯录: 名字: YangCheng, 性别: 男, 年龄: 22, 电话: 13962611111, 住址: 江苏省昆山市 ********************************* ** 1、增加联系人 2、删除联系人 ** ** 3、查找联系人 4、修改联系人 ** ** 5、展示联系人 0、退出通讯录 ** ********************************* 请输入您的选择:0 成功退出通讯录!