class Contact:
def __init__(self, name, phone_number, email, addr):
self.name = name
self.phone_number = phone_number
self.email = email
self.addr = addr
def print_info(self):
print("Name: ", self.name)
print("Phone Number: ", self.phone_number)
print("E-mail: ", self.email)
print("Address: ", self.addr)
def set_contact():
name = input("Name: ")
phone_number = input("Phone Number: ")
e_mail = input("E-mail: ")
addr = input("Address: ")
contact=Contact(name,phone_number,e_mail,addr)
return contact
def print_menu():
print("1. 연락처 입력")
print("2. 연락처 출력")
print("3. 연락처 삭제")
print("4. 종료")
menu=input("메뉴를 입력 : ")
return int(menu)
def print_contact(contact_list):
for contact_iter in contact_list:
contact_iter.print_info()
print('---------------------')
def delete_contact(contact_list,name):
for i, contact in enumerate(contact_list):
if contact.name == name:
del contact_list[i]
print("삭제 완료")
return
print("삭제 실패")
def store_contact(contact_list):
f=open("contact_db.txt","wt")
for contact in contact_list:
f.write(contact.name+'/')
f.write(contact.phone_number+'/')
f.write(contact.email+'/')
f.write(contact.addr+'/')
f.write('\n')
f.close()
def load_contact(contact_list):
f=open("contact_db.txt","rt")
lines=f.readline()
data_num=int(len(lines))
while (lines=f.readline())==0 :
#name, phone, email, addr = lines[i].split('/')
#contact=Contact(name,phone,email,addr)
#contact_list.append(contact)
for i in range(data_num):
#name= lines[i].rstrip('/')
#phone = lines[i].rstrip('/')
#email = lines[i].rstrip('/')
#addr = lines[i].rstrip('/')
name, phone, email, addr = lines[i].split('/')
contact=Contact(name,phone,email,addr)
contact_list.append(contact)
f.close()
def run():
print("Contact Program!!")
contact_list = []
load_contact(contact_list)
while 1:
menu=print_menu()
if menu==1: #연락처 입력
contact = set_contact()
contact_list.append(contact)
elif menu==2: #연락처 출력
print_contact(contact_list)
elif menu==3:
del_name=input("삭제할 이름을 입력하세요:")
delete_contact(contact_list,del_name)
elif menu==4:
store_contact(contact_list)
break
if __name__ == "__main__":
run()