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 import引入不同路径下的模块
Jul 11 Python
Python Json模块中dumps、loads、dump、load函数介绍
May 15 Python
python3中函数参数的四种简单用法
Jul 09 Python
python特性语法之遍历、公共方法、引用
Aug 08 Python
详解Python with/as使用说明
Dec 13 Python
浅谈Python基础—判断和循环
Mar 22 Python
Python学习笔记之pandas索引列、过滤、分组、求和功能示例
Jun 03 Python
Python Numpy计算各类距离的方法
Jul 05 Python
在linux系统下安装python librtmp包的实现方法
Jul 22 Python
pytorch 更改预训练模型网络结构的方法
Aug 19 Python
Python如何实现小程序 无限求和平均
Feb 18 Python
使用Python封装excel操作指南
Jan 29 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
Ajax+PHP 边学边练 之二 实例
2009/11/24 PHP
php中3种方法统计字符串中每种字符的个数并排序
2012/08/27 PHP
关于PHP文件的自动运行方法分析
2016/05/13 PHP
ThinkPHP实现生成和校验验证码功能
2017/04/28 PHP
PHP实现基于回溯法求解迷宫问题的方法详解
2017/08/17 PHP
ThinkPHP5.0框架实现切换数据库的方法分析
2019/10/30 PHP
javascript 原型模式实现OOP的再研究
2009/04/09 Javascript
js 动态加载事件的几种方法总结
2013/12/25 Javascript
js计算时间差代码【包括计算,天,时,分,秒】
2016/04/26 Javascript
js仿淘宝和百度文库的评分功能
2016/05/15 Javascript
微信小程序 教程之wxapp 视图容器 view
2016/10/19 Javascript
解决bootstrap中使用modal加载kindeditor时弹出层文本框不能输入的问题
2017/06/05 Javascript
JS中把函数作为另一函数的参数传递方法(总结)
2017/06/28 Javascript
解决VUEX刷新的时候出现数据消失
2017/07/03 Javascript
关于Promise 异步编程的实例讲解
2017/09/01 Javascript
AngularJS中table表格基本操作示例
2017/10/10 Javascript
封装运动框架实战左右与上下滑动的焦点轮播图(实例)
2017/10/17 Javascript
详解webpack+ES6+Sass搭建多页面应用
2018/11/05 Javascript
详解js加减乘除精确计算
2019/03/19 Javascript
前端使用crypto.js进行加密的函数代码
2020/08/16 Javascript
Python基于贪心算法解决背包问题示例
2017/11/27 Python
python write无法写入文件的解决方法
2019/01/23 Python
详解Ubuntu16.04安装Python3.7及其pip3并切换为默认版本
2019/02/25 Python
python 实现将多条曲线画在一幅图上的方法
2019/07/07 Python
Python进程的通信Queue、Pipe实例分析
2020/03/30 Python
python实现在内存中读写str和二进制数据代码
2020/04/24 Python
python中如何写类
2020/06/29 Python
解决python便携版无法直接运行py文件的问题
2020/09/01 Python
Expedia英国:全球最大的在线旅游公司
2017/09/07 全球购物
6PM官网:折扣鞋、服装及配饰
2018/08/03 全球购物
团委竞选演讲稿
2014/04/24 职场文书
副处级干部考察材料
2014/05/17 职场文书
小学生九一八纪念日83周年演讲稿500字
2014/09/17 职场文书
幼儿园教师师德表现自我评价
2015/03/05 职场文书
丧事答谢词大全
2015/09/30 职场文书
职场新人知识:如何制定一份合理的工作计划?
2019/09/11 职场文书