python_example/phonebook.py at main · nicecoding1/python_example · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
importpickle
importos.path

defprint_menu():
print()
print("="*50)
print("1. 조회, 2. 입력, 3. 수정, 4. 삭제, 5. 종료")
print("="*50)

defphonebook_search():
globalphonebook
print("1. 조회")
a=input("조회할 이름 또는 전화번호를 입력하세요: ").strip()
idx=0
foriinphonebook:
ifa=="":
print("[{}] {} {}".format(idx+1, i[0], i[1]))
idx+=1
elifaini[0] oraini[1]:
print("[{}] {} {}".format(idx+1, i[0], i[1]))
idx+=1
ifidx==0:
print("조회된 자료가 없습니다.")

print_menu()
return

defphonebook_insert():
globalphonebook
print("2. 입력")
a=input("입력할 이름을 입력하세요: ").strip()
b=input("입력할 전화번호를 입력하세요: ").strip()
ifa==""orb=="":
print("입력값이 부족합니다. 다시 입력해주세요.")
phonebook_insert()
return

phonebook.append([a, b])
print("{} {} 입력 성공".format(a, b))
print_menu()
return

defphonebook_update():
globalphonebook
print("3. 수정")
a=input("이름을 입력하세요: ").strip()
b=input("변경할 전화번호를 입력하세요: ").strip()
ifa==""orb=="":
print("입력값이 부족합니다. 다시 입력해주세요.")
phonebook_update()
return

idx=0
cnt=0
foriinphonebook:
ifi[0] ==a:
phonebook[idx][1] =b
cnt+=1
idx+=1

ifcnt>0:
print("{} {} 수정 성공".format(a, b))
else:
print("{} 이름이 존재하지 않습니다.".format(a))

print_menu()
return

defphonebook_delete():
globalphonebook
print("4. 삭제")
a=input("삭제할 전화번호를 입력하세요: ").strip()
ifa=="":
print("입력값이 부족합니다. 다시 입력해주세요.")
phonebook_delete()
return

tmp_list= []

find=0
foriinphonebook:
ifnoti[1] ==a:
tmp_list.append([i[0], i[1]])
else:
find+=1

phonebook.clear()
phonebook=tmp_list

iffind>0:
print("{} 삭제 성공".format(a))
else:
print("{} 번호가 존재하지 않습니다.".format(a))

print_menu()
return

defsave_phonebook():
globalphonebook
withopen("phonebook.pickle", "wb") asf:
pickle.dump(phonebook, f, pickle.HIGHEST_PROTOCOL)

"""
[
['홍길동','010-1234-5678'],
['김철수','010-1234-5555'],
['김대한','010-1111-2222']
]
"""

phonebook= []

ifos.path.exists("phonebook.pickle"):
withopen("phonebook.pickle", "rb") asf:
phonebook=pickle.load(f)

print("파이썬 전화번호부 Ver 1.0")
print_menu()

whileTrue:
m=input("기능을 선택하세요.(1~5 입력) ").strip()
ifm=="": continue

ifm=="1":
phonebook_search()

elifm=="2":
phonebook_insert()
save_phonebook()

elifm=="3":
phonebook_update()
save_phonebook()

elifm=="4":
phonebook_delete()
save_phonebook()

elifm=="5":
save_phonebook()
break