# -*- coding:utf-8 -*-
#import json   
import shutil
from prettytable import PrettyTable
#print json.dumps(context, encoding="UTF-8", ensure_ascii=False) 
#输入姓名或手机号码,显示详细信息  
c_cnt=0 #次数计数器 
while  c_cnt<3:  #尝试超过三次即退出
    #读取文件内容放入字典
    with open('address_list.txt','r') as f:
        context={}
        for i in f.readlines():
            line=i.strip().split()
            context[line[0]]=line[1:] #给字典赋值,列表的第一个值赋值为键,其余值赋为值
    v_cnt=0 #数据计数器
    row = PrettyTable() 
    row.field_names = ['name','job','place','office-phone','mobile','email'] #格式化输出
    v_put=raw_input("Your input is :").strip().decode('gbk').encode('utf8')#decode是解码,本例系统字符集是gbk,要转化成utf8做处理用encode(文件是utf8格式)
    #增删改功能
    if v_put=='more' or v_put=='MORE':
        print '''选择功能:
                            D:删除
                            I:插入
                            U:更新'''
        v_more=raw_input("Your input is :").strip()  
        if   v_more=='d' or    v_more=='D' :
            while c_cnt<3:
                v_del = raw_input("please input the info to delete: ").strip().decode('gbk').encode('utf8')
                if len(v_del) == 0:
                    print "empty input,try again!"
                    c_cnt+=1
                elif v_del=='exit' or v_del=='EXIT':     
                    break    
                else:
                    # 输入值与源文件比对,有则丢弃,没有则添加到新文件,最后新文件覆盖源文件,实现删除功能
                    v_dcn=0
                    with open('address_list.txt','r') as yy:
                        with open('address_list_bak.txt','a') as bb:
                            with file('address_list_new.txt','a') as nn:
                                for line in yy.readlines():
                                    if v_del in line:
                                        bb.write(line)
                                        v_dcn+=1
                                        print  "\033[32;1mdelete\033[0m"+":  "+line
                                    elif v_del not in line:
                                        nn.write(line)
                                if v_dcn==0:
                                    c_cnt+=1
                                    print "The data is not exsist!"
                                elif v_dcn>0:    
                                    break
            shutil.move('address_list_new.txt','address_list.txt')
        elif   v_more=='i' or    v_more=='I' :
            while v_cnt<4:			
                v_ins1 = raw_input("please input name: ").strip().decode('gbk').encode('utf8')
                v_ins2 = raw_input("please input job: ").strip().decode('gbk').encode('utf8')
                v_ins3 = raw_input("please input place: ").strip().decode('gbk').encode('utf8')
                v_ins4 = raw_input("please input office-phone: ").strip().decode('gbk').encode('utf8')
                v_ins5 = raw_input("please input mobile: ").strip().decode('gbk').encode('utf8')
                v_ins6 = raw_input("please input email: ").strip().decode('gbk').encode('utf8')
                if len(v_ins1) == 0 or v_ins5 == 0:
                    print "Name or mobile is empty ,try again!"
                    c_cnt+=1
                    continue
                else:
                    with open('D:\python_pro\\address_list.txt','a') as yy:
                        yy.write('\n'+v_ins1+'    '+v_ins2+'    '+v_ins3+'    '+v_ins4+'    '+v_ins5+'    '+v_ins6) 
                        break 
    elif  v_put=='exit' or v_put=='EXIT':       
        break
    else:              
        if len(v_put)==0 or (v_put[0]=='1'  and len(v_put)<4) :  #限制输入的数字,否则会泄露数据,列入输入1 ,则返回所有手机号
            c_cnt+=1
            print "Your input is not valid!"
            continue
        elif v_put[0]=='1'  and len(v_put)>=4 :
            for num,con in context.items(): #遍历所有信息,有则计数器加一,并输出
                if con[3].count(v_put)!=0: #count是列表的属性,返回element在列表某个值中出现的次数,通过次数来做模糊匹配 
                    v_cnt+=1
                    row.add_row(num.split()+con)  #num.split() 字符串+".split()",可以对字符串切片,并转换为列表
            print row
                    #使用json处理字典内部汉字乱码问题 print  json.dumps(num, encoding="UTF-8", ensure_ascii=False),json.dumps(con, encoding="UTF-8", ensure_ascii=False)
            if  v_cnt>0:           
                print 'Found %s items.'  % v_cnt 
            else:
                c_cnt+=1
                print "The number is not found !"     
        else :
            for name in context.keys(): #遍历所有key,有则计数器加一,并输出
                if name.count(v_put)>0: #字典的模糊匹配
                    v_cnt+=1
                    row.add_row(name.split()+context[name])
            if  v_cnt>0:   
                print row        
                print 'Found %s items.'  % v_cnt 
            else:
                c_cnt+=1
                print "The name is not found !"