[Python Project 02-6] 프로그램 종료 기능 구현 및 전체 코드 (PyMySQL)

# 6-1. 프로그램 종료하기 (quit)

elif cmd == 'q':

connect.commit()

print("프로그램을 종료합니다.")

프로그램 종료 시에는 특별한 것 없이 'commit'만 한번 더 해주었습니다.

***사용 가능 명령어 리스트***

a  : 모든 데이터 조회(all)

f  : 조건에 맞는 데이터만 조회(find)

i  : 수강생 입력(insert)

d  : 수강생 삭제(delete)

r  : 정보 변경(revise)

q  : 저장하고 나가기(quit)

command(명령어 입력): q

프로그램을 종료합니다.

# 6-2. 프로그램 전체 코드

def print_result(result):

print("\n------------------------------------------------------------------------")

cnames = ['이름', '생년월일', '소속', '전화번호', '거주지']

for c in cnames:

print(c.center(11, ' '), end='\t')

print("\n------------------------------------------------------------------------")

ccodes = ['name', 'birth', 'team', 'phone', 'address']

for r in result:

for c in ccodes:

print(str(r[c]).center(11, ' '), end='\t')

print("")

print("------------------------------------------------------------------------")

u = input("계정명: ")

p = input("패스워드: ")

import pymysql

connect = pymysql.connect(host='localhost', user=u, password=p, db='test', charset='utf8')

cursor = connect.cursor(pymysql.cursors.DictCursor)

cmd = 'na'

while cmd != 'q':

print("\n***사용 가능 명령어 리스트***")

print("a  : 모든 데이터 조회(all)")

print("f  : 조건에 맞는 데이터만 조회(find)")

print("i  : 수강생 입력(insert)")

print("d  : 수강생 삭제(delete)")

print("r  : 정보 변경(revise)")

print("q  : 저장하고 나가기(quit)\n")

cmd = input("command(명령어 입력): ")

print("")

if cmd == 'a':

sql = "SELECT * FROM student"

cursor.execute(sql)

result = cursor.fetchall()

print_result(result)

elif cmd == 'f':

print("무엇을 기준으로 검색하시겠습니까? (이름/나이/소속/전화번호/거주지)")

con = input("> ")

if con == '이름':

print("\n검색하실 수강생의 이름을 입력해 주세요.")

con2 = input("> ")

sql = "SELECT * FROM student WHERE name = '{}'".format(con2)

cursor.execute(sql)

result = cursor.fetchall()

print_result(result)

elif con == '나이':

print("\n※ 숫자만 입력 해 주세요.\n")

input_age1 = int(input("몇 살 이상을 검색하시겠습니까? "))

input_age2 = int(input("몇 살 이하을 검색하시겠습니까? "))

import datetime, time

d = datetime.date.today()

year1 = d.year - input_age2 + 1

year2 = d.year - input_age1 + 1

sql = "SELECT * FROM student WHERE LEFT(birth, 4) BETWEEN {} AND {}".format(year1, year2)

cursor.execute(sql)

result = cursor.fetchall()

print_result(result)

elif con == '소속':

print("\n검색하실 수강 그룹을 입력해 주세요. (A/B/C)")

con2 = input("> ")

sql = "SELECT * FROM student WHERE team = '{}'".format(con2)

cursor.execute(sql)

result = cursor.fetchall()

print_result(result)

elif con == '전화번호':

print("\n검색하실 전화번호을 입력해 주세요. (숫자만 입력 해 주세요.)")

con2 = input("> ")

sql = "SELECT * FROM student WHERE phone = '{}'".format(con2)

cursor.execute(sql)

result = cursor.fetchall()

print_result(result)

elif con == '거주지':

print("\n검색하실 거주지를 입력해 주세요.")

con2 = input("> ")

sql = "SELECT * FROM student WHERE address = '{}'".format(con2)

cursor.execute(sql)

result = cursor.fetchall()

print_result(result)

elif cmd == 'i':

name = input("수강생 이름: ")

birth = input("생년월일(숫자 8자리): ")

team = input("소속 그룹(A/B/C): ")

phone = input("전화번호(숫자 11자리): ")

address = input("거주지: ")

sql = "INSERT INTO student VALUES('{}',{},'{}','{}','{}')".format(name, birth, team, phone, address)

cursor.execute(sql)

connect.commit()

print("해당 데이터의 입력이 완료되었습니다.")

elif cmd == 'd':

print("※ 수강생 삭제는 전화번호를 통해서만 가능합니다.")

phone = input("목록에서 삭제할 수강생의 전화번호(숫자 11자리): ")

sql = "DELETE FROM student WHERE phone = '{}')".format(phone)

cursor.execute(sql)

connect.commit()

print("해당 데이터의 삭제가 완료되었습니다.")

elif cmd == 'r':

print("※ 정보를 변경할 수강생의 선택은 전화번호를 통해서만 가능합니다.")

phone = input("전화번호(숫자 11자리): ")

sql = "SELECT * FROM student WHERE phone = '{}')".format(phone)

cursor.execute(sql)

result = cursor.fetchall()

print_result(result)

con = input("\n어떤 정보를 변경하시겠습니까(이름/생년월일/소속/전화번호/거주지)? ")    

if con == '이름':

print("\n해당 수강생의 이름을 무엇으로 수정하시겠습니까?")

name = input("> ")

sql = "UPDATE student SET name = '{}' where phone = '{}'".format(name, phone)

cursor.execute(sql)

result = cursor.fetchall()

connect.commit()

print("\n정보 변경이 완료되었습니다.")

elif con == '생년월일':

print("해당 수강생의 생년월일을 무엇으로 수정하시겠습니까? (숫자 8자리)")

birth = input("> ")

sql = "UPDATE student SET birth = '{}' where phone = '{}'".format(birth, phone)

cursor.execute(sql)

result = cursor.fetchall()

connect.commit()

print("\n정보 변경이 완료되었습니다.")

elif con == '소속':

print("\n해당 수강생의 소속을 무엇으로 수정하시겠습니까? (A/B/C)")

team = input("> ")

sql = "UPDATE student SET team = '{}' where phone = '{}'".format(team, phone)

cursor.execute(sql)

result = cursor.fetchall()

connect.commit()

print("\n정보 변경이 완료되었습니다.")

elif con == '전화번호':

print("\n해당 수강생의 전화번호를 무엇으로 수정하시겠습니까? (숫자 11자리)")

new_phone = input("> ")

sql = "UPDATE student SET phone = '{}' where phone = '{}'".format(new_phone, phone)

cursor.execute(sql)

result = cursor.fetchall()

connect.commit()

print("\n정보 변경이 완료되었습니다.")

elif con == '거주지':

print("\n해당 수강생의 거주지를 어디로 수정하시겠습니까? ")

address = input("> ")

sql = "UPDATE student SET address = '{}' where phone = '{}'".format(address, phone)

cursor.execute(sql)

result = cursor.fetchall()

connect.commit()

print("\n정보 변경이 완료되었습니다.")

elif cmd == 'q':

connect.commit()

print("프로그램을 종료합니다.")

else:

print("잘못된 명령어가 입력되었습니다. 다시 입력해 주세요.")