脚本之家,脚本语言编程技术及教程分享平台!
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|

服务器之家 - 脚本之家 - Python - Python 实现简单的电话本功能

Python 实现简单的电话本功能

2020-07-29 12:02脚本之家 Python

这篇文章主要介绍了Python 实现简单的电话本功能的相关资料,包括添加联系人信息,查找姓名显示联系人,存储联系人到 TXT 文档等内容,十分的细致,有需要的小伙伴可以参考下

myPhoneBook2.py

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
#!/usr/bin/python
# -*- coding: utf-8 -*-
import re
class PhoneBook(object):
  '''这是一个电话簿脚本。
 
  该脚本能够实现
  AddContact:添加联系人信息
  ShowContact:查找姓名显示联系人
  SaveContacts:存储联系人到 TXT 文档(存储格式——姓名:号码/号码)
  LoadContacts:从 txt 文档中载入联系人
  '''
  def __init__(self):
    self.contactsDict = {}
 
  def AddContact(self):
    while True:
      name = raw_input('请输入姓名>>>')
      name = name.strip()
      # 姓名必须包含有效字符
      if name != '':
        break
      print '***姓名不能为空'
       
    while True:
      number = raw_input('请输入号码>>>')
      number = re.sub(r'\D', '', number)
      # 删除号码中的非数字字符
      if number != '':
        break
      print '***号码只能是数字'
 
    cover = True #若联系人已存在,是否覆盖
    if self.contactsDict.has_key(name):
      print '***联系人已存在'
      self.ShowContact(name)
      while True:
        control = raw_input('''
输入 "c":覆盖原来号码
输入 "n":保留原来号码并存储新号码
输入 "q" 退出\n>>>''')
        if control.lower() == 'n':
          cover = False
          break
        if control.lower() == 'c':
          break
        if control.lower() == 'q':
          return None
        print '***输入错误'
     
    if cover:
      self.contactsDict[name] = number
    else:
      if number in self.contactsDict[name]:
        print '***号码已经存在'
      else:
        self.contactsDict[name] = self.contactsDict[name] + '/' + number
 
  def ShowContact(self, name):
    print '++++++++++++++++++++++++'
    if self.contactsDict.has_key(name):
      print '【联系人信息】'
      print '【姓名: %s】' % name
      numberList = self.contactsDict[name].split('/')
      for num in range(len(numberList)):
        print '【号码%d: %s】' % (num+1, numberList[num])
     
    else:
      print '【找不到联系人%s】' % name
    print '++++++++++++++++++++++++'
   
  def DeleteNumber(self, name):
    if self.contactsDict.has_key(name):
      self.ShowContact(name)
      number = self.contactsDict[name].split('/')
      while True:
        print '''
请输入要删除号码的序号
或者输入 "a" 删除该联系人
或者输入 "q" 退出(不删除)
(若联系人号码被全部删除,该联系人也将被删除)'''
        control = raw_input('>>>')
        if control.lower() == 'q':
          break
        elif control.lower() == 'a':
          del self.contactsDict[name]
          break
        elif control.isdigit() and int(control) <= len(number):
          del number[int(control)-1]
          self.contactsDict[name] = '/'.join(number)
          break
        else:
          print '***输入有误'
 
  def LoadContacts(self):
    '''
    try:
      PhoneBook = open('PhoneBook.txt', 'a+')
      contacts = PhoneBook.read()
      if contacts == '':
        print '***电话簿为空'
      else:
        ContactsList = contacts.split('\n')
        for contact in ContactsList:
          if not contact == '':
            contact = contact.split(':')
            name = contact[0]
            number = contact[1]
            self.contactsDict[name] = number
    finally:
      PhoneBook.close()
    '''
    self.contactsDict = {line.split(':')[0]: line.split(':')[1] for line in open('PhoneBook.txt','a+').readlines()}
 
  def SaveContacts(self):
    try:  
      if self.contactsDict:
        PhoneBook = open('PhoneBook.txt', 'w')
        for name, number in self.contactsDict.items():
          line = name + ':' + number
          PhoneBook.write(line)
          PhoneBook.write('\n')
       
      else:
        print '***没有联系人信息'
     
    finally:
      PhoneBook.close()
 
 
if __name__ == '__main__':
  myPhoneBook = PhoneBook()
  myPhoneBook.LoadContacts()
  try:
    while True:
      raw_input('按回车键继续')
      print '''
--------------------------------
  输入 a:添加联系人
  输入 s:显示联系人信息
  输入 d:删除联系人
  输入 q:退出
--------------------------------'''
      control = raw_input('>>>')
      if control.lower() == 'a':
        myPhoneBook.AddContact()
      elif control.lower() == 's':
        name = raw_input('请输入要查找的联系人姓名\n>>>')
        myPhoneBook.ShowContact(name)
      elif control.lower() == 'd':
        name = raw_input('请输入要删除的联系人姓名\n>>>')
        myPhoneBook.DeleteNumber(name)
      elif control.lower() == 'q':
        break
      else:
        print '***输入有误'
       
  finally:
    myPhoneBook.SaveContacts()

花了一个下午和半个晚上写了这个简单通讯录:
哈哈,第一次写这么长的Python代码,自认为结构还是挺合理的。
代码如下:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
#-*- coding:utf-8 -*-
# file :addrList.py
# date :2011-10-24 15:40:13
# 设计一个基本的通讯录管理程序,使其具有添加,删除,编辑,查找等功能。
# 要求使用C/C++,java,javascript,python中任意一种语言实现。字符界面即可。
# 不需要有GUI(图形界面)
import sys
import os
import string
import re
from datetime import datetime
 
QUIT_FLAG = False
ADDRS_LIST = "addr_list.txt"
_addrs_dict = {} 
_addrs_count = 0
DEBUG=2
def info(message):
  global DEBUG
  if DEBUG > 0:
    print message
 
def debug(message):
  global DEBUG
  if DEBUG > 1:
    print message
 
def warn(message):
  global DEBUG
  if DEBUG > 0:
    print message
   
def error(message):
  print message
 
def help():
  print "用法:输入菜单中的命令即可执行相应操作!"
  print "可用菜单命令如下:"
  showMenu()
 
def showMenu():
  print "+******************操作菜单***********************+"
  print "|查看所有联系人(all) | 查找联系人(find)      |"
  print "|添加联系人(add)   | 删除联系人(remove)     |"
  print "|编辑联系人(edit)  | 保存并退出(save)      |"
  print "|使用帮助(help)   | 退出但不保存(quit)     |"
  print "+_________________________________________________+"
 
def showError(info):
  print info
 
 
def doSwitch(op):
  if op == "all":
    doAll()
  elif op == "find":
    doFind()
  elif op == "add":
    doAdd()
  elif op == "remove":
    doRemove()
  elif op == "edit":
    doEdit()
  elif op == "save":
    doSave()
  elif op == "help":
    help()
  elif op == "quit":
    doQuit()
  else:
    showError("错误:您输入的命令有误,请重新输入。需要帮助请输入help!")
 
def verifyInput(items):
  _flag = True
  _rPhone = re.compile(r'1[0-9]{10}')
  _rQQ  = re.compile(r'[1-9][0-9]{4,9}')
  if len(items[0]) > 10:
    _flag = False
    print "姓名太长了!"
  if not _rPhone.match(items[1]):
    _flag = False
    print "手机号码格式不正确"
  if not _rQQ.match(items[2]):
    _flag = False
    print "QQ号码输入有误"
  return _flag
   
 
def buildAddr(addr):
  _addr={}
  items=addr.split()
  if len(items) < 3:
    print "您输入的信息太少了"
    return None
  if not verifyInput(items):
    return None
 
  _addr['name']=items[0]
  _addr['phone'] = items[1]
  _addr['QQ'] = items[2]
  return _addr
 
def addAddr(addr):
  global _addrs_count,_addrs_dict
  _addrs_count+=1
  _addr=buildAddr(addr)
  if not _addr:
    return None
  _addrs_dict[_addrs_count]=_addr
 
def init():
  if not os.path.isfile(ADDRS_LIST):
    return None
 
  faddr=open(ADDRS_LIST,"r")
  for line in faddr: 
    if len(line) == 0:
      continue
    addAddr(line)
  faddr.close()
 
 
def save():
  global _addrs_dict
  faddr=open(ADDRS_LIST,"w+")
  for addr in _addrs_dict.values():
    faddr.write("{0}\t{1}\t{2}\n".format(addr['name'],addr['phone'],addr['QQ']))
  faddr.flush()
  faddr.close()
 
 
def doAll():
  global _addrs_dict
  if len(_addrs_dict) < 1:
    print "联系人当中暂无记录!"
    return None
  printHead()
  for key,addr in _addrs_dict.items():
    printAddr(key,addr)
 
def doFind():
  _flag=False
  flag1=flag2=flag3=False
  cond=raw_input("请输入查询信息:>")
  debug("DEBUG:{0}".format(cond))
  if len(cond) == 0:
    return None
  if cond.isdigit():
    flag1=findById(int(cond,10))
    flag2=findByPhone(cond)
    flag3=findByQQ(cond)
  else:
    flag1=findByName(cond)
   
  _flag = flag1 or flag2 or flag3 
  if not _flag:
    print "没有查找到任何联系人!"
 
def doAdd():
  line = raw_input("请依次输入联系人的姓名,手机号码,QQ号码>")
  if len(line) == 0:
    return None
  addAddr(line)  
 
def existsId(_id):
  global _addrs_dict
  return _addrs_dict.has_key(_id)
#  if _id > _addrs_count or _id < 1:
#    return False 
#  else:
#    return True
 
def doRemove():
  FLAG = True
  while FLAG:
    key=raw_input("请输入要删除的联系人的编号(取消请输入#)")
    if key == '#':
      FLAG = False
      continue
    if not existsId(int(key,10)):
      print "不存在您所输入编号的联系人。请确认"
      continue
   
    print "编号为 {0} 的联系人信息如下:".format(key)
    printById(int(key,10))
    yesOrNo=raw_input("您确定要删除上述联系人吗?(y/n)")
    if yesOrNo in "yY":
      removeById(int(key,10))
      print "删除成功!"
    yesOrNo=raw_input("您还需要继续删除联系人吗?(y/n)")
    if not yesOrNo in "yY":
      FLAG = False
 
 
 
def doEdit():
  FLAG = True
  while FLAG:
    key=raw_input("请输入要编辑的联系人的编号(取消请输入#)")
    print "DEBUG:key:{0}".format(key)
    if key == '#':
      FLAG = False
      continue
    if not existsId(int(key,10)):
      print "不存在您所输入编号的联系人。请确认"
      continue
   
    print "编号为 {0} 的联系人信息如下:".format(key)
    printById(int(key,10))
    updateById(int(key,10))
    FLAG = False
 
 
def doSave():
  save()
  doQuit()
 
def doQuit():
  global QUIT_FLAG
  QUIT_FLAG = True
  print "正在退出……"
#  exit(0)
 
def printHead():
  print "+-----+----------+---------------+---------------+"
  print "|编号 |  姓名  | 手机号码   |   QQ号码  |"
  print "+-----+----------+---------------+---------------+"
 
def printAddr(key,addr):
#  print "+-----+----------+---------------+---------------+"
  print "|{0:^5}|{1:^10}|{2:^15}|{3:^15}|".format(key,addr['name'],addr['phone'],addr['QQ'])
  print "+-----+----------+---------------+---------------+"
 
 
 
def printById(_id):
  global _addrs_dict
  printHead()
  printAddr(_id,_addrs_dict[_id])
 
def removeById(_id):
  global _addrs_dict
  _addrs_dict.pop(_id)
 
def updateById(_id):
  global _addrs_dict
  _addr= _addrs_dict.get(_id)
 
  print "请输入该联系人的新信息,相应留空则保留原有信息"
  name=raw_input("请输入新姓名:>")
  if len(name) > 0:
    _addr['name']=name
 
  phone=raw_input("请输入新手机号码:>")
  if len(phone) > 0:
    _addr['phone']=phone
 
  qq=raw_input("请输入新的QQ号码:>")
  if len(qq) > 0 :
    _addr['QQ']=qq
 
  _addrs_dict[_id]=_addr
  print "更新成功!"
  print "联系人新的信息如下:"
  printById(_id)
 
def findById(_id):
  if existsId(_id):
    printById(_id)
    return True
  else:
    return False
 
def findByField(cond,field='name'):
  global _addrs_dict
  _flag = False
  for key,addr in _addrs_dict.items():
    if addr[field].find(cond) != -1:
      printAddr(key,addr)
      _flag=True
  return _flag
 
def findByName(name):
  return findByField(name,'name')
 
def findByPhone(phone):
  return findByField(phone,'phone')
 
def findByQQ(qq):
  return findByField(qq,'QQ')
 
 
 
def main():
  init()
  showMenu()
  while(not QUIT_FLAG):
    operation=raw_input("请在此处输入菜单命令>")
    doSwitch(operation)
 
 
if __name__=='__main__':
  main()
  ## do something
##----------------------------------------------------

延伸 · 阅读

精彩推荐