Python 实现简单的电话本功能


Posted in Python onAugust 09, 2015

myPhoneBook2.py

#!/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代码,自认为结构还是挺合理的。
代码如下:

#-*- 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 
##----------------------------------------------------
Python 相关文章推荐
全面了解Python的getattr(),setattr(),delattr(),hasattr()
Jun 14 Python
python中利用队列asyncio.Queue进行通讯详解
Sep 10 Python
Python实现矩阵加法和乘法的方法分析
Dec 19 Python
浅谈Scrapy框架普通反爬虫机制的应对策略
Dec 28 Python
Python语言描述随机梯度下降法
Jan 04 Python
python中sys.argv函数精简概括
Jul 08 Python
Python 将Matrix、Dict保存到文件的方法
Oct 30 Python
利用Python正则表达式过滤敏感词的方法
Jan 21 Python
Python3.5文件读与写操作经典实例详解
May 01 Python
如何在python中写hive脚本
Nov 08 Python
Python判断字符串是否为合法标示符操作
Sep 03 Python
python tqdm用法及实例详解
Jun 16 Python
python批量提取word内信息
Aug 09 #Python
python实现下载指定网址所有图片的方法
Aug 08 #Python
Python实现多线程抓取妹子图
Aug 08 #Python
通过Python来使用七牛云存储的方法详解
Aug 07 #Python
Python爬虫框架Scrapy实战之批量抓取招聘信息
Aug 07 #Python
深入理解Python中命名空间的查找规则LEGB
Aug 06 #Python
举例详解Python中yield生成器的用法
Aug 05 #Python
You might like
php 连接mysql连接被重置的解决方法
2011/02/15 PHP
php中的比较运算符详解
2013/10/28 PHP
php中将一个对象保存到Session中的方法
2015/03/13 PHP
PHP如何将图片文件上传到另外一台服务器上
2019/08/26 PHP
javascript对象的property和prototype是这样一种关系
2007/03/24 Javascript
jQuery中json对象的复制方式介绍(数组及对象)
2013/06/08 Javascript
JQuery与JS里submit()的区别示例介绍
2014/02/17 Javascript
javascript自定义的addClass()方法
2014/05/28 Javascript
Jquery焦点与失去焦点示例应用
2014/06/10 Javascript
SeaJS 与 RequireJS 的差异对比
2014/12/08 Javascript
jQuery实现返回顶部功能
2016/02/23 Javascript
jquery获取easyui日期控件的值实现方法
2016/11/09 Javascript
基于jQuery实现的幻灯图片切换
2016/12/02 Javascript
JavaScript登录记住密码操作(超简单代码)
2017/03/22 Javascript
JS实现标签页切换效果
2017/05/04 Javascript
vue富文本编辑器组件vue-quill-edit使用教程
2018/09/21 Javascript
Vue组件实现触底判断
2019/06/26 Javascript
Vue.js路由实现选项卡简单实例
2019/07/24 Javascript
vue-cli设置css不生效的解决方法
2020/02/07 Javascript
vue页面跳转实现页面缓存操作
2020/07/22 Javascript
Javascript中window.name属性详解
2020/11/19 Javascript
[01:32]2016国际邀请赛中国区预选赛IG战队首日赛后采访
2016/06/27 DOTA
python脚本内运行linux命令的方法
2015/07/02 Python
Python安装模块的常见问题及解决方法
2018/02/05 Python
numpy中三维数组中加入元素后的位置详解
2019/11/28 Python
使用python 的matplotlib 画轨道实例
2020/01/19 Python
keras .h5转移动端的.tflite文件实现方式
2020/05/25 Python
Python wordcloud库安装方法总结
2020/12/31 Python
HTML5拖拽文件上传的示例代码
2021/03/04 HTML / CSS
BabyBjörn婴儿背带法国官网:BabyBjorn法国
2018/06/16 全球购物
Does C# support multiple inheritance? (C#支持多重继承吗)
2012/01/04 面试题
化学学院毕业生自荐信范文
2013/12/17 职场文书
学子宴答谢词
2014/01/25 职场文书
如何撰写一封出色的求职信
2014/04/27 职场文书
2015年世界无烟日演讲稿
2015/03/18 职场文书
Python可变与不可变数据和深拷贝与浅拷贝
2022/04/06 Python