python_example/phonebook_sqlite3.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
import sqlite3

table_name = "phonebook"
conn = sqlite3.connect("phonebook.db")
cur = conn.cursor()

def create_table():
    global conn
    try:

        cur.execute("select * from sqlite_master where type='table' and tbl_name='phonebook'")
        rows = cur.fetchall()

        if rows:
            print(f"DB(테이블) {table_name}이 이미 존재합니다. ")
        else:
            conn.execute('create table phonebook(no integer primary key autoincrement, name text, hpno text)')
            conn.commit()
            print("DB(테이블) 생성 성공")
    except:
        print("DB(테이블) 생성 실패")

    print_menu()
    return


def print_menu():
    print()
    print("="*60)
    print("1. 조회, 2. 입력, 3. 수정, 4. 삭제, 5. 종료, 6. DB생성")
    print("="*60)


def phonebook_search():
    global conn, cur
    print("1. 조회")
    a = input("조회할 이름 또는 전화번호를 입력하세요: ").strip()

    cur.execute("select * from phonebook where (name like '%"+a+"%' or hpno like '%"+a+"%')")
    rows = cur.fetchall()

    idx = 0
    for row in rows:
        print("[{}] {} {}".format(row[0], row[1], row[2]))
        idx += 1

    if idx == 0:
        print("조회된 자료가 없습니다.")

    print_menu()
    return


def phonebook_insert():
    global conn, cur
    print("2. 입력")
    a = input("입력할 이름을 입력하세요: ").strip()
    b = input("입력할 전화번호를 입력하세요: ").strip()
    if a == "" or b == "":
        print("입력값이 부족합니다. 다시 입력해주세요.")
        phonebook_insert()
        return

    try:
        cur.execute('insert into phonebook (name, hpno) values (?, ?)', (a, b))
        conn.commit()

        print("{} {} 입력 성공".format(a, b))
    except:
        print("{} {} 입력 실패".format(a, b))

    print_menu()
    return

def phonebook_update():
    global conn, cur
    print("3. 수정")
    a = input("이름을 입력하세요: ").strip()
    b = input("변경할 전화번호를 입력하세요: ").strip()
    if a == "" or b == "":
        print("입력값이 부족합니다. 다시 입력해주세요.")
        phonebook_update()
        return

    try:
        cur.execute('update phonebook set hpno=? where name=?', (b, a))
        conn.commit()
        print("{} {} 수정 성공".format(a, b))
    except:
        print("{} {} 수정 실패".format(a, b))

    print_menu()
    return

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

    try:
        cur.execute('delete from phonebook where hpno=?', (a,))
        conn.commit()
        print("{} 삭제 성공".format(a))
    except:
        print("{} 삭제 실패".format(a))

    print_menu()
    return


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

while True:
    m = input("기능을 선택하세요.(1~6 입력) ").strip()
    if m == "": continue

    if m == "1":
        phonebook_search()

    elif m == "2":
        phonebook_insert()

    elif m == "3":
        phonebook_update()

    elif m == "4":
        phonebook_delete()

    elif m == "5":
        break

    elif m == "6":
        create_table()

cur.close()
conn.close()