(23.02.08)Python 프로그래밍 : Pickle 모듈을 사용한 직렬화(Serialization), 예외처리

pickle 모듈을 사용한 직렬화(Serialization)

• 직렬화 : pickle.dump(data,fout)
• 역직렬화 : pickle.load(fin)

 

 

 

 

직렬화를 이용한 CRUD

• Emp 클래스
• 리스트에 Emp객체를 저장하고 리스트를 직렬화

# Emp (id, name, dept, phone)
# 추가(a), 목록(s), 검색(f), 수정(u), 삭제(d), 종료(x)

 

 

import pickle
class Emp:
    def __init__(self, id=None, name=None, dept=None, phone=None):
        self.id = id
        self.name = name
        self.dept = dept
        self.phone = phone
        
    def __str__(self):
        return '%s\t%s\t%s\t%s'%(self.id,self.name,self.dept,self.phone)
    
    def __eq__(self, other):
        return self.id==other.id

 

 

emp_list = []
 

글로벌 위치에 선언한다. 메소드 밖에 선언

 

 

def menu():
    print()
    msg = '추가(a), 목록(s), 검색(f), 수정(u), 삭제(d), 종료(x):'
    m_in = input(msg).strip()
    return m_in

 

 

def input_emp():
    sinput = input('id name dept phone:').strip()
    in_list = sinput.split()
    if len(in_list)<4:
        print('사원정보입력 오류')
        return None
    id,name,dept,phone = in_list
    return Emp(id=id, name=name, dept=dept, phone=phone)

 

 

import pickle
def overwrite(emp_list):
    with open('C:/Users/parkj/1Python_Projects/text/pickle_emp_crud.txt', 'wb') as fout:
        pickle.dump(emp_list, fout)
        print('직렬화 성공')

 

 

def add():
    emp = input_emp()
    emp_list.append(emp)
    overwrite(emp_list)
    print('사원정보 추가 성공')

 

 

import pickle
def load_list():
    global emp_list
    with open('C:/Users/parkj/1Python_Projects/text/pickle_emp_crud.txt', 'rb') as fin:
        emp_list = pickle.load(fin)
    emp_list = sorted(emp_list, key=lambda e:e.id)

 

 

import pickle
 
def show_list():
    load_list()
    print()
    for e in emp_list:
        print(e)

 

 

def find(label):
    key_id = input(label+':').strip()
    key_emp = Emp(id=key_id)
    load_list()
    found = None
    if key_emp in emp_list:
        idx = emp_list.index(key_emp)
        found = emp_list[idx]
    if found is None:
        print('검색결과가 없습니다')
    return found

 

 

def update():
    up_emp = find('수정할 사원번호')
    if up_emp is None:
        return
    new_phone = input('새 전화번호:').strip()
    up_emp.phone = new_phone
    overwrite(emp_list)
    print('사원정보 수정 성공')

 

 

def delete():
    del_emp = find('삭제할 사원번호')
    load_list()
    if del_emp in emp_list:
        emp_list.remove(del_emp)
        overwrite(emp_list)
        print('사원정보 삭제 성공')
    else:
        print('사원정보 삭제 실패')

 

 

while True:
    m_in = menu()
    if m_in == 'a':
        add()
    elif m_in=='s':
        show_list()
    elif m_in=='f':
        print(find('검색할 사원번호'))
    elif m_in=='u':
        update()
    elif m_in=='d':
        delete()
    elif m_in=='x':
        break
    else:
        print('메뉴입력 오류')
        
print('---------------------프로그램 종료---------------------')

 

실행결과:

 

 

---

 

예외처리