(29)Python_카페 주문 관리 프로그램

• create database test 를 mysql에서 만들어주고 해야한다.
• DB 접속을 test로 하기 때문
• 관리자 모드는 manager table에 등록된 정보로 로그인을 해야 관리자 모드를 접속할 수 있다.
• 프로그램 종료는 관리자 모드의 영업 종료를 누르면 종료가 된다.

class Coffee:
    def __init__(self, coffee, price=0, stock=0):
        self.coffee = coffee
        self.price = price
        self.stock = stock
    
    def setCoffee(self, coffee):
        self.coffee = coffee
    
    def getCoffee(self):
        return self.coffee

    def setPrice(self, price):
        self.price = price
    
    def getPrice(self):
        return self.price

    def setStock(self, stock):
        self.stock = stock
    
    def getStock(self):
        return self.stock

 

import MySQLdb

class CoffeesDao:
    def __init__(self):
        self.db = None
    
    def connect(self):
        self.db = MySQLdb.connect('localhost', 'root', '1234', 'test')
        
    def disconnect(self):
        self.db.close()
    
    def insert(self, coffee):
        self.connect()
        cur = self.db.cursor()
        sql = "insert into bana (b_coffee, b_price, b_stock) values (%s, %s, %s)"
        data = (coffee.getCoffee(), coffee.getPrice(), coffee.getStock())
        cur.execute(sql, data)
        self.db.commit()
        self.disconnect()     
        
    def selectAll(self):
        self.connect()
        cur = self.db.cursor()
        sql = 'select b_coffee, b_price, b_stock from bana'
        cur.execute(sql)
        row = cur.fetchall()
        self.disconnect()
        return row

    def search(self, coffee):
        self.connect()
        cur = self.db.cursor(MySQLdb.cursors.DictCursor)
        sql = 'select b_coffee, b_price, b_stock from bana where b_coffee=%s'
        data = (coffee,)
        cur.execute(sql, data)
        row = cur.fetchone()
        self.disconnect()
        return row
    
    def update(self, word):
        self.connect()
        cur = self.db.cursor()
        sql = 'update bana set b_price=%s, b_stock=%s where b_coffee=%s'
        data = (word.getPrice(), word.getStock(), word.getCoffee())
        result = cur.execute(sql, data)
        self.db.commit()
        if result > 0:
            print('수정되었습니다')
        else:
            print('해당 커피가 없습니다')
        self.disconnect()
    
    def delete(self, coffee):
        self.connect()
        cur = self.db.cursor()
        sql = 'delete from bana where b_coffee=%s'
        data = (coffee,)
        result = cur.execute(sql, data)
        self.db.commit()
        if result > 0:
            print('삭제되었습니다')
        else:
            print('해당 단어가 없습니다')
        self.disconnect()
        
    def sales(self):
        self.connect()
        cur = self.db.cursor()
        sql = 'select b_coffee, b_price*b_sales from bana'
        cur.execute(sql)
        row = cur.fetchall()
        self.disconnect()
        return row
    
    def orderCount(self,coffee):
        self.connect()
        cur = self.db.cursor()
        sql = 'select b_coffee, b_stock from bana where b_coffee=%s'
        data = (coffee,)
        cur.execute(sql, data)
        row = cur.fetchone()
        if row[1] == 0:
            print('커피 재고가 없습니다')
        else:
            sql = 'update bana set b_sales=b_sales+1, b_stock=b_stock-1 where b_coffee=%s'
            data = (coffee,)
            cur.execute(sql, data)
            print('커피주문이 완료되었습니다.','커피를 받아가세요')
        self.db.commit()
        self.disconnect()
        
    def loginManager(self,managerId, mamagerpw):
        self.connect()
        cur = self.db.cursor()
        sql = 'select m_userid, m_userpw from manager'
        cur.execute(sql)
        row = cur.fetchall()
        self.disconnect()
        return row
    
    def insertManager(self, userid, userpw):
        self.connect()
        cur = self.db.cursor()
        sql = "insert into manager (m_userid, m_userpw) values (%s, %s)"
        data = (userid, userpw)
        cur.execute(sql, data)
        self.db.commit()
        self.disconnect()
        
    def delManager(self, userid):
        self.connect()
        cur = self.db.cursor()
        sql = 'delete from manager where m_userid=%s'
        data = (userid,)
        result = cur.execute(sql, data)
        self.db.commit()
        if result > 0:
            print('삭제되었습니다')
        else:
            print('해당 매니저가 없습니다')
        self.disconnect()

 

class banaService:
    def __init__(self):
        self.dao = CoffeesDao()
    
    def insertMenu(self):
        coffee = input('커피명을 입력하세요')
        price = input('커피 가격을 입력하세요')
        stock = input('재고량 입력하세요')
        coffee = Coffee(coffee, price, stock)
        self.dao.insert(coffee)
    
    def printAll(self):
        datas = self.dao.selectAll()
        for i in datas:
            print(f'커피: {i[0]}, 가격{i[1]}, 재고{i[2]}')
    
    def selectCoffee(self):
        coffee = input('검색할 커피를 입력하세요')
        word = self.dao.search(coffee)
        if word:
            print(f"커피:{word['b_coffee']}, 가격:{word['b_price']}, 재고:{word['b_stock']}")
        else:
            print('찾는 커피가 없습니다')
    
    def editCoffee(self):
        coffee = input('수정할 커피를 입력하세요')
        word = self.dao.search(coffee)
        if word == None:
            print('수정할 커피를 찾지 못했습니다')
        else:
            price = input('새로운 가격을 입력하세요')
            stock = input('새로운 재고량을 입력하세요')
            word = Coffee(coffee, price, stock)
            self.dao.update(word)
    
    def delCoffee(self):
        coffee = input('삭제할 단어를 입력하세요')
        self.dao.delete(coffee)
    
    def salesCoffee(self):
        datas = self.dao.sales()
        coffeeSum = 0
        if datas == None:
            print('매출이 없습니다.')
        else:
            for i in datas:
                print(f'커피: {i[0]}, 매출액{i[1]}')
                coffeeSum = coffeeSum + i[1]
            print(f'총 매출액{coffeeSum}')
            
    def customerPrint(self):
        datas = self.dao.selectAll()
        for i in datas:
            print(f'커피: {i[0]}, 가격{i[1]}')
            
    def order(self):
        coffee = input('주문하실 커피를 입력해주세요')
        word = self.dao.search(coffee)
        if word:
            print(f"커피:{word['b_coffee']}, 가격:{word['b_price']} 주문하시겠습니까?")
            select = int(input('1. 주문완료 2.취소'))
            if select == 1:
                self.dao.orderCount(coffee)
            elif select == 2:
                return
        else:
            print('찾는 커피가 없습니다')
        
    def managerMode(self):
        mangerId = input('아이디를 입력해주세요: ')
        mangerpw = input('비밀번호를 입력해주세요: ')
        datas = self.dao.loginManager(mangerId,mangerpw)
        for i in datas:
            if i[0] == mangerId and i[1] == mangerpw:
                return True
    
    def managerSign(self):
        userId = input('이름을 입력하세요')
        userpw = input('비밀번호를 입력하세요')
        self.dao.insertManager(userId, userpw)
        print('등록완료')
        
    def managerDel(self):
        manager = input('삭제할 매니저를 입력하세요')
        self.dao.delManager(manager)

 

import MySQLdb
import time

class Menu:
    def __init__(self):
        self.service = banaService()
        try:
            db = MySQLdb.connect('localhost', 'root', '1234', 'test')
            cur = db.cursor()
            sql = "create table bana(b_idx int not null auto_increment primary key,b_coffee varchar(20) not null,b_price int,b_stock int,b_sales int default 0)"
            cur.execute(sql)
            sql = "create table manager(m_idx int not null auto_increment primary key,m_userid varchar(20) not null,m_userpw varchar(20) not null)"
            cur.execute(sql) 
            sql = "insert into manager (m_userid, m_userpw) values ('master','1111')"
            cur.execute(sql)
            db.commit()
            db.close()
            
        except:
            pass
        
    
    def run(self):
        while True:
            try:
                mode = int(input('1.관리자 2.고객'))
                if mode == 1:
                    login = self.service.managerMode()
                    if login == True:
                        while True:
                            menu = int(input('1. 메뉴관리 2.매출확인 3.매니저관리 4.메인으로 5.영업종료'))
                            if menu == 1:
                                menuoption = int(input('1.메뉴등록 2.메뉴출력 3.메뉴검색 4.메뉴수정 5.메뉴삭제'))
                                if menuoption == 1:
                                    self.service.insertMenu()
                                elif menuoption == 2:
                                    self.service.printAll()
                                elif menuoption == 3:
                                    self.service.selectCoffee()
                                elif menuoption == 4:
                                    self.service.editCoffee()
                                elif menuoption == 5:
                                    self.service.delCoffee()
                            elif menu == 2:
                                self.service.salesCoffee()
                            elif menu == 3:
                                menuoption = int(input('1.매니저등록 2.매니저삭제'))
                                if menuoption == 1:
                                    self.service.managerSign()
                                elif menuoption == 2:
                                    self.service.managerDel()
                            elif menu == 4: 
                                break
                            elif menu == 5:
                                return
                    else:
                        print('아이디 비밀번호를 확인해주세요')
                        
                elif mode == 2:
                    menu = int(input('1.주문하기 2.메뉴보기 3.종료하기'))
                    if menu == 1:
                        self.service.order()
                    elif menu == 2:
                        self.service.customerPrint()
                    elif menu == 3:
                        continue

                        
                elif mode == 3:
                    break
                    
                else:
                    print('다시 입력하세요')
            except Exception as e:
                print('다시 입력하세요')

 

start = Menu()
start.run()