代码:
import re
def add_details(details_dict, file_mode):
with open("address.txt", file_mode) as book:
book.write("Name: {}\n".format(details_dict['Name']))
book.write("Address: {}\n".format(details_dict['Address']))
book.write("Home Phone No.: {}\n".format(details_dict['Home Phone No.']))
book.write("Mobile Phone No.: {}".format(details_dict['Mobile Phone No.']))
book.write("\n---------------\n")
def delete_person(name):
with open("address.txt", "r+") as book:
records = re.split("[-]+", book.read(), re.M)
for data in records:
record = get_record(data)
if record.get('Name', None) != name:
add_details(record, "w")
def get_record(string):
return dict(re.findall("^(.*): (.*)$", string, re.M))
def print_record(record):
print "\n"
print "Name: {}".format(record['Name'])
print "Address: {}".format(record['Address'])
print "Home Phone No.: {}".format(record['Home Phone No.'])
print "Mobile Phone No.: {}".format(record['Mobile Phone No.'])
def search_for_person(name):
with open("address.txt", "r") as book:
records = re.split("[-]+", book.read(), re.M)
for data in records:
record = get_record(data)
if record.get('Name', None) == name:
print_record(record)
choice = raw_input("Add a new person (1), delete a person(2), or search for a person(3)?\n")
if choice == "1":
details = {'Name':"", 'Address':"", 'Home Phone No.':"", 'Mobile Phone No.':""}
details['Name'] = raw_input("Enter name of contact: ")
details['Address'] = raw_input("Enter address of contact: ")
details['Home Phone No.'] = raw_input("Enter Home Telephone No. of contact: ")
details['Mobile Phone No.'] = raw_input("Enter Mobile Telephone No. of contact: ")
add_details(details, "a")
elif choice == "2":
name = raw_input("Enter name to delete: ")
delete_person(name)
elif choice == "3":
name = raw_input("Enter name: ")
print search_for_person(name)
基本上,每当我尝试使用 delete_person() 方法删除一个人时,我都会得到以下回溯:
Traceback (most recent call last):
File "address.py", line 56, in <module>
delete_person(name)
File "address.py", line 19, in delete_person
add_details(record, "w")
File "address.py", line 6, in add_details
book.write("Name: {}\n".format(details_dict['Name']))
KeyError: 'Name'
但是,除该方法之外的所有内容都可以正常工作。考虑到我以完全相同的方式设置字典,我不应该得到错误,但无论如何我是。对此有什么帮助吗?如果需要,这是文件的布局:
Name: test
Address: testaddress
Home Phone No.: 2313123121233
Mobile Phone No.: 423423423432
---------------
Name: test2
Address: testaddress2
Home Phone No.: 342353454345
Mobile Phone No.: 231231391
---------------