

Python 基于列表实现的通讯录管理系统

码途钥匙

2024-12-12

通讯录管理系统

这是一个基于文本的界面程序，用户可以通过命令行与之交互，它使用了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()