# encoding: utf-8

'''

@author: Techzero

@email: techzero@163.com

@time: 2014-4-30 下午1:31:04

'''

import os

import sys

import cPickle as p

class Person:

    def __init__(self, name, email):

        '''Initializes the person's data.'''

        self.name = name

        self.email = email

def create():

    """Create new person and input email"""

    global Persons

    try:

        name = raw_input("Please input name:")

        while Persons.has_key(name):

            name = raw_input("This name has already exist, please input new name:")

        email = raw_input("Please input Email:")

    except EOFError:

        print '\nEOF Error'

        sys.exit()

    Persons[name] = email

    print ""

def delete():

    """Search person by name and delete"""

    global Persons

    try:

        name = raw_input("Please input the person's name you want to delete:")

    except EOFError:

        print '\nEOF Error'

        sys.exit()

    if Persons.has_key(name):

        del Persons[name]

        save()

    else:

        print "No one called",name,"!\n"

def modify():

    """Search person by name and modify email"""

    global Persons

    try:

        name = raw_input("Please input the person's name you want to modify:")

        if Persons.has_key(name):

            del Persons[name]

            email = raw_input("Please input new email:")

            Persons[name] = email

            save()

        else:

            print "No one called",name,"!\n"

    except EOFError:

        print '\nEOF Error'

        sys.exit()

def save():

    """Save Persons to file"""

    global Persons

    File = 'person.dat'

    f = file(File, 'w')

    p.dump(Persons, f)

    f.close()

    print "Operation Done!\n"

def read():

    """Read person from file"""

    global Persons

    File = 'person.dat'

    if os.path.exists(File):

        f = file(File)

        Persons = p.load(f)

        f.close()

    else:

        File = 'person.dat'

        f = file(File, 'w')

        f.close()

def display():

    """Display all persons in the dictionary"""

    global Persons

    for name, email in Persons.items():

        print "        ",name,email

    print ""

def search():

    """Search person by name"""

    global Persons

    try:

        name = raw_input("Please input the person's name you want to search:")

    except EOFError:

        print '\nEOF Error'

        sys.exit()

    if Persons.has_key(name):

        print "        ",name,Persons[name],"\n"

    else:

        print "No one called",name,"!\n"

def menu():

    """Display a menu to choose operation"""

    choose = "0"

    while True:

        #i = os.system("cls")

        print'''1----Create

2----Delete

3----Modify

4----Search

5----Display

6----Exit'''

        try:

            choose = raw_input("Please choose an item(1-6):")

        except EOFError:

            print '\nEOF Error'

            sys.exit()

        if choose == "1":

            create()

        elif choose == "2":

            delete()

        elif choose == "3":

            modify()

        elif choose == "4":

            search()

        elif choose == "5":

            display()

        elif choose == "6":

            print "Thanks for using!"

            sys.exit()

        else:

            print ""

Persons = {}

read()

menu()