# -*- coding:utf-8 -*- 
# 練習2： 
# 模擬手機通訊錄搜索功能 
class PERSON(object): 
  def __init__(self): 
    self.name_list = ['zhangsan', 'zhangsanfeng', 'zhaoliu', 'wangwu', 'wulong', 'wanda', 'lisi', 'liuyang','luoxiaoxiao', 'linmeimei', 'luhan', 'gaohairui', 'guojing', 'huangrong', 'hanhan'] 
    #聲明空字典 
    self.ss_dict = {} 
  def total(self): 
    #for循環(huán)遍歷列表，取出每一個姓名 
    for name in self.name_list: 
      #取出首字母 
      s =name[0] 
      #判斷字典中是否有s這個key 
      if s not in self.ss_dict: 
        #如果沒有，根據(jù)key將存放name的列表放入字典 
        # s_list = [] 
        # s_list.append(name) 
        # self.ss_dict[s] = s_list 
        self.ss_dict[s] = [name] 
      else: 
        #如果有，根據(jù)key取出列表，將這個name添加到列表 
        s_list = self.ss_dict[s] 
        s_list.append(name) 
    print(self.ss_dict) 
  def add_person(self): 
    name = input('請輸入姓名：') 
    if name not in self.name_list: 
      self.name_list.append(name) 
      s = name[0] 
      if s not in self.ss_dict: 
        self.ss_dict[s] = [name] 
      else: 
        s_list = self.ss_dict[s] 
        s_list.append(name) 
    else: 
      print('已存在！') 
  def query_person(self): 
    s = input('請輸入要查找聯(lián)系人名字的首字母：') 
    # s就是要查找的字典的key 
    if s not in self.ss_dict: 
      print('不存在') 
    else: 
      # 根據(jù)這個key取出字典中對應的值（列表） 
      s_list = self.ss_dict[s] 
      for index,name in enumerate(s_list): 
        print('序號：%s 姓名：%s' % (index,name)) 
      # for x in range(0,len(s_list)): 
      #   name = s_list[x] 
      #   print('序號：%s 姓名：%s'%(x,name)) 
    return s 
  def alter_person(self): 
    s= self.query_person() 
    index = int(input('請輸入要修改聯(lián)系人的序號:')) 
    new_name = input('輸入更改后的名字（%s）：'%self.ss_dict[s][index]) 
    del self.ss_dict[s][index] 
    s = new_name[0] 
    if s not in self.ss_dict: 
      self.ss_dict[s] = [new_name] 
    else: 
      s_list = self.ss_dict[s] 
      s_list.append(new_name) 
  def delete_person(self): 
    while len(self.ss_dict) == '0': 
      print('信息為空！') 
    print('1.根據(jù)名字刪除') 
    print('2.全部刪除') 
    print('0.退出') 
    num = int(input('輸入序號：')) 
    while num not in range(0,3 ): 
      num = int(input('選項不存在，請重輸:')) 
    if num == 1: 
      s = self.query_person() 
      index = int(input('請輸入要刪除聯(lián)系人的序號:')) 
      rs = input('是否真的要刪除,(y/n):') 
      if rs == 'y': 
        self.ss_dict[s].pop(index) 
        print('刪除成功') 
      else: 
        print('取消刪除') 
 
    elif num == 2: 
      rs = input('是否真的要刪除,(y/n):') 
      if rs == 'y': 
        self.ss_dict.clear() 
        print('刪除成功') 
      else: 
        print('取消刪除') 
    else: 
      return 
  def start(self): 
    self.total() 
    while 1: 
      print('1.添加聯(lián)系人') 
      print('2.查詢聯(lián)系人') 
      print('3.修改聯(lián)系人') 
      print('4.刪除聯(lián)系人') 
      print('0.退出') 
      num = int(input('輸入序號：')) 
      while num not in range(0,5): 
        num = int(input('選項不存在，請重輸:')) 
      if num == 1: 
        self.add_person() 
      elif num == 2: 
        self.query_person() 
      elif num == 3: 
        self.alter_person() 
      elif num == 4: 
        self.delete_person() 
      else: 
        break 
 
p = PERSON() 
p.start()