firstapp.py

来自「《python编程金典》书中」· Python 代码 · 共 63 行

#!/usr/bin/python
# -*- coding: cp936 -*-
#filename:FirstApp.py

import cPickle as pf
import os
print "------------这是我的第一个Python应用程序------------"
class Person:
     def __init__(self,name,Email):
          self.name=name
          self.Email=Email
pdict=dict()
while True:
     comm=raw_input("请你输入命令:l-载入通讯记录，a-增加，d-删除,s-保存为文件,q-退出,p-显示:")
     if comm=='L' or comm=='l':
          filename=raw_input("通讯录文件名")
          f=file(filename)
          pdict=pf.load(f)
     if comm=='P'or comm=='p':
          if len(pdict)==0:
               print "没有联系人"
          else:
               for name,p in pdict.items():
                    print "姓名%s" %p.name
                    print "邮件%s" %p.Email
                    print "------------------"
                    
     if comm=='a'or comm=='A':
          name=raw_input("姓名：")
          Email=raw_input("电子邮件：")
          if pdict.has_key(name):
               print "已经有这个联系人"
          else:
               cp='p'+name
               cp=Person(name,Email)
               pdict[name]=cp
          print "有%d个联系人\n"%len(pdict)
          for name,p in pdict.items():
               print "姓名%s" %p.name
               print "邮件%s" %p.Email
               print "------------------"
     if comm=='d' or comm=='D':
          name=raw_input("姓名：")
          if pdict.has_key(name):
               del pdict[name]
          else:
               print"没有这个联系人"
          
     if comm=='q'or comm=='Q':
          break
     if comm=='S' or comm=='s':
          if len(pdict)==0:
               print "联系记录为空，不需要保存"
          else:
               f=file('lianxi.data','w')
               pf.dump(pdict,f)
               f.close()