Python实现命令行通讯录实例教程

# 1. 初始化全局配置与数据存储容器
# 2. 定义 Person 类封装联系人属性（姓名、关系、电话）
# 3. 定义 Operation 类集中处理增、删、查、存、退等指令逻辑
# 4. 启动程序并检查本地数据文件 existence
# 5. 依据检测结果决定加载旧数据或创建新文件
# 6. 进入 while 主循环等待用户输入指令
# 7. 指令路由分发：
#    addperson → 添加新联系人
#    delperson → 移除指定联系人
#    search → 检索联系人详情
#    quit → 强制退出且不保存
#    sq → 写入文件后安全退出

import pickle
import os

# 全局数据存储字典与枚举分组
person_data = {}
relationship_list = ['家人', '朋友', '同事']
running = True

class Person:
    def __init__(self, name, relationship='朋友', tel='None'):
        self.name = name
        self.relationship = relationship
        self.tel = tel
        person_data[name] = {'relationship': relationship, 'tel': tel}

class Operation:
    @staticmethod
    def add_person():
        name = input('请输入姓名：')
        if name in person_data:
            print('联系人已存在。')
            return
        try:
            rel_idx = int(input('请选择分组(0:家人,1:朋友,2:同事):'))
            rel = relationship_list[rel_idx]
        except (ValueError, IndexError):
            print('输入无效，默认设置为朋友。')
            rel = '朋友'
        tel = input('请输入电话：')
        Person(name, rel, tel)
        print(f'已成功添加 {name}。')

    @staticmethod
    def del_person():
        name = input('请输入要删除的联系人姓名：')
        if name in person_data:
            del person_data[name]
            print(f'已删除 {name}。')
        else:
            print('联系人不存在。')

    @staticmethod
    def search():
        name = input('请输入要查找的联系人的姓名：')
        if name in person_data:
            info = person_data[name]
            print(f'姓名：{name}, 关系：{info["relationship"]}, 电话：{info["tel"]}')
        else:
            print('联系人不存在。')

    @staticmethod
    def save_quit():
        global running
        addressbook_file = 'addressbook.data'
        with open(addressbook_file, 'wb') as f:
            pickle.dump(person_data, f)
        print('更改已保存并退出程序。')
        running = False

    @staticmethod
    def quit_no_save():
        global running
        print('未保存更改，直接退出程序。')
        running = False

# 主程序入口
if __name__ == '__main__':
    addressbook_file = 'addressbook.data'
    if os.path.exists(addressbook_file):
        with open(addressbook_file, 'rb') as f:
            person_data = pickle.load(f)
        print('已从本地文件加载通讯录数据。')
    else:
        create_input = input('未找到通讯录文件，是否创建？yes/no ')
        if create_input.lower() == 'yes':
            with open(addressbook_file, 'wb') as f:
                pickle.dump(person_data, f)
            print('通讯录文件已创建。')
        elif create_input.lower() != 'no':
            print('输入无效，将不创建文件。')
            running = False

    while running:
        command = input('请输入指令：').strip()
        if command == 'addperson':
            Operation.add_person()
        elif command == 'delperson':
            Operation.del_person()
        elif command == 'search':
            Operation.search()
        elif command == 'quit':
            Operation.quit_no_save()
        elif command == 'sq':
            Operation.save_quit()
        else:
            print('未找到指令！')

$ python3 addressbook.py
已从本地文件加载通讯录数据。
请输入指令：addperson
请输入姓名：zhangsan
请选择分组(0:家人,1:朋友,2:同事):1
请输入电话：1234567
已成功添加 zhangsan。
请输入指令：search
请输入要查找的联系人的姓名：zhangsan
姓名：zhangsan, 关系：朋友, 电话：1234567
请输入指令：sq
更改已保存并退出程序。