4-1. 파이썬 데이터베이스 활용 단어장 만들기

1. create table voca (

eng varchar(50) primary key,

kor varchar(50) not null,

lev int default 1

);

 

import MySQLdb

class Menu:

    def __init__(self):

        self.service = WordsService()

    def run(self):

        while True:

            try:

                menu = int(input('1.등록하기 2.출력하기 3.검색하기 4.수정하기 5.삭제하기 6.종료하기'))

                if menu == 1:

                    self.service.insertWord()

                elif menu == 2:

                    self.service.printAll()

                elif menu == 3:

                    self.service.searchWord()

                elif menu == 4:

                    self.service.editWord()

                elif menu == 5:

                    self.service.delWord()

                elif menu == 6:

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

                    break

            except Exception as e:

                print(e)

                print('다시 입력하세요')

 

class Words:

    def __init__(self, eng, kor, lev=1):

        self.eng = eng

        self.kor = kor

        self.lev = lev

    def setEng(self, eng):

        self.eng = eng

    def getEng(self):

        return self.eng

    def setKor(self, kor):

        self.kor = kor

    def getKor(self):

        return self.kor

    def setLev(self, lev):

        self.lev = lev

    def getLev(self):

        return self.lev

 

class WordsService:

    def __init__(self):

        self.dao = WordsDao()

 

    def insertWord(self):

        eng = input('단어를 입력하세요')

        kor = input('뜻을 입력하세요')

        lev = input('레벨을 입력하세요')

        word = Words(eng, kor, lev)

        self.dao.insert(word)

    def printAll(self):

        datas = self.dao.selectAll()

        for data in datas:

            print(f"{data['eng']}, 뜻:{data['kor']}, 레벨:{data['lev']}")

 

    def searchWord(self):

        eng = input('검색할 단어를 입력하세요')

        datas = self.dao.search(eng)

        if datas:

            for data in datas:

                print(f"{data['eng']}, 뜻:{data['kor']}, 레벨:{data['lev']}")

        else:

            print('찾는 단어가 없습니다')

 

    def editWord(self):

        eng = input('수정할 단어를 입력하세요')

        word = self.dao.search(eng)

        if not word:

            print('수정할 단어가 없습니다')

        else:

            kor = input('새로운 뜻을 입력하세요')

            lev = input('새로운 레벨을 입력하세요')

            word = Words(eng, kor, lev)

            self.dao.update(word)

 

    def delWord(self):

        eng = input('삭제할 단어를 입력하세요')

        word = self.dao.search(eng)

        if not word:

            print('삭제할 단어가 없습니다')

        else:

            self.dao.delete(eng)

 

class WordsDao:

    def __init__(self):

        self.db = None

    def connect(self):

        self.db = MySQLdb.connect('localhost', 'root',  '1234',  'kdt')

    def disconnect(self):

        self.db.close()

 

    def insert(self, word):

        self.connect()

        cur = self.db.cursor()

        sql = 'insert into voca (eng, kor, lev) values (%s, %s, %s)'

        data = (voca.getEng(), voca.getKor(), voca.getLev())

        cur.execute(sql, data)

        self.db.commit()

        cur.close()

        self.disconnect()

 

    def selectAll(self):

        self.connect()

        cur = self.db.cursor(MySQLdb.cursors.DictCursor)

        sql = 'select eng, kor, lev from voca order by eng asc'

        cur.execute(sql)

        row = cur.fetchall()

        cur.close()

        self.disconnect()

        return row

 

    def search(self, eng):

        self.connect()

        cur = self.db.cursor(MySQLdb.cursors.DictCursor)

        sql = "select eng, kor, lev from voca where eng like concat('%%', %s, '%%')"

        data = (eng, )

        cur.execute(sql, data)

        row = cur.fetchall()

        cur.close()

        self.disconnect()

        return row

 

    def update(self, word):

        self.connect()

        cur = self.db.cursor()

        sql = 'update voca set kor=%s, lev=%s where eng=%s'

        data = (voca.getKor(), voca.getLev(), voca.getEng())

        result = cur.execute(sql, data)

        self.db.commit()

        if result > 0:

            print('수정되었습니다')

        else:

            print('에러!')

        cur.close()

        self.disconnect()

 

    def delete(self, eng):

        self.connect()

        cur = self.db.cursor()

        sql = 'delete from voca where eng=%s'

        data = (eng, )

        result = cur.execute(sql, data)

        self.db.commit()

        if result > 0:

            print('삭제되었습니다')

        else:

            print('에러!')

        cur.close()

        self.disconnect()

start = Menu()

start.run()

### 과제

MySQL을 이용하여 DataBase를 활용한 프로그램을 개발해보자.

* 시나리오는 자유롭게 작성

* ERD를 이미지로 첨부

* 시나리오와 테스트는 직접 작성하여 ipynb 파일로 슬랙에 업로드 제출