Как заменить алгоритм телефонного справочника на эквивалентный?

# -*- coding: cp1251 -*-
print('Телефонный справочник')
names=open('phone.txt','r')
a=names.readlines()
surname={}
for i in a:
            person = i.split(' ')
            surname[person[0]]=[person[1], person[2]]
number={}
for i in a:
            person=i.split(' ')
            number[person[2]]=[person[0], person[1]]
while 1:
    print('1 – Поиск по фамилии')
    print('2 - Поиск по номеру телефона')
    print('3 – Добавить абонента')
    print('4 – Отобразить список всех людей в алфавитном порядке')
    print('5 – Выход из программы')
    data = raw_input('Выберете пункт меню. Введите его номер: ')
    if data == '1':    
        for o in a:
            o = raw_input('Введите фамилию: ')
            if o in surname:
                print('Имя: %s Номер: %s' % (surname[o][0], surname[o][1]))
                break
            else:
                print('Абонент с такой фамилией не найден')
                break       
    if data == '2':
        for x in a:
            x = raw_input('Введите телефон: ')
            if x in number:
                print('Фамилия: %s Имя: %s' % (number[o][0], number[o][1]))
                break
            else:
                print('Абонент с таким телефоном не найден')
                break
    if data == '3':
        f = raw_input('Фамилия: ')
        n = raw_input('Имя: ')
        p = raw_input('Телефон: ')
        with open("phone.txt", "r+") as Phone:
            Phone.write(f + " " + n + " " + p)
        print('Абонент добавлен')
    if data == '4':
        alph = surname.keys()
        alph.sort()
        for i in alph:
            print(i)
    if data == '5':
        print('Выход из программы')
        break

#!/usr/bin/python2.7
#coding:utf-8
 
import os
 
class Phone_book(object):
    def __init__(self, ):
        self.contacts = open('phone.txt').readlines()
        if self.contacts[-1][-1] != '\n':
            self.contacts[-1] = self.contacts[-1] + '\n'
 
    def add_contact(self):
        name = raw_input('Введите имя и фамилию:')
        number = raw_input('Введите номер телефона:')
        self.contacts.append(name + ' ' + number + '\n')
        f = open('phone.txt', 'w')
        for i in self.contacts:
            f.write(i)
        f.close()
        raw_input('Контакт успешно добавлен \n \
                  Нажмите Enter, чтобы вернуться в меню')
 
    def find(self):
        name = raw_input('Введите данные для поиска: ')
        temp = 0
        for i in self.contacts:
            if i.find(name) != -1:
                print(i)
                temp += 1
        if not temp:
            print 'Контакт не найден'
        raw_input('Нажмите Enter, чтобы вернуться в меню')
 
    def print_contacts(self):
        for i in sorted(self.contacts):
            print(i)
        raw_input('Нажмите Enter, чтобы вернуться в меню')
 
 
if __name__ == '__main__':
    contacts = Phone_book()
    dict_items = {1:contacts.find,
                  2:contacts.find,
                  3:contacts.add_contact,
                  4:contacts.print_contacts,
                  5:exit}
    while True:
        print '1 – поиск по фамилии \n' \
              '2 – поиск по телефону \n' \
              '3 – добавить человека \n' \
              '4 – отобразить список контактов \n' \
              '5 – выход \n'
        choice = int(raw_input('Введите номер нужного пункта: '))
        dict_items[choice]()
        os.system('clear')

>>> def hello():
...     print "Hello1"
...     
... 
>>> def hello2():
...     print "hello2"
...     
... 
>>> book = {1:hello, 2:hello2}
>>> book[1]
<function hello at 0x1c65488>
>>> book[1]()
Hello1
>>> book[raw_input()]()
1
Traceback (most recent call last):
  File "<input>", line 1, in <module>
KeyError: '1'
>>> book[int(raw_input())]()
 
Traceback (most recent call last):
  File "<input>", line 1, in <module>
ValueError: invalid literal for int() with base 10: ''
>>> book[int(raw_input())]()
1
Hello1
>>>

#!/usr/bin/python2.7
#coding:utf-8
 
import os
 
class Phone_book(object):
    def __init__(self, ):
        self.contacts = open('phone.txt').readlines()
        if self.contacts[-1][-1] != '\n':
            self.contacts[-1] = self.contacts[-1] + '\n'
 
    def add_contact(self):
        name = raw_input('Введите имя и фамилию:')
        number = raw_input('Введите номер телефона:')
        self.contacts.append(name + ' ' + number + '\n')
        f = open('phone.txt', 'w')
        for i in self.contacts:
            f.write(i)
        f.close()
        raw_input('Контакт успешно добавлен \n \
                  Нажмите Enter, чтобы вернуться в меню')
 
    def find(self):
        name = raw_input('Введите данные для поиска: ')
        temp = 0
        for i in self.contacts:
            if i.find(name) != -1:
                print(i)
                temp += 1
        if not temp:
            print 'Контакт не найден'
        raw_input('Нажмите Enter, чтобы вернуться в меню')
 
    def print_contacts(self):
        for i in sorted(self.contacts):
            print(i)
        raw_input('Нажмите Enter, чтобы вернуться в меню')
 
 
if __name__ == '__main__':
    contacts = Phone_book()
    dict_items = {1:contacts.find,
                  2:contacts.find,
                  3:contacts.add_contact,
                  4:contacts.print_contacts,
                  5:exit}
    while True:
        print '1 – поиск по фамилии \n' \
              '2 – поиск по телефону \n' \
              '3 – добавить человека \n' \
              '4 – отобразить список контактов \n' \
              '5 – выход \n'
        choice = int(raw_input('Введите номер нужного пункта: '))
        dict_items[choice]()
        os.system('clear')