[Python] 제품입고정리 프로그램 (MVC pattern)

---

init.py

__all__ = [
    'main',
    'productMenu',
    'productDao',
    'productVO',
    'productService'
]
# 파일화 하는법
"""
1. 먼저 패키지 만들어준다.
2. 위와같은 클래스파일에 그 해당 클래스파일에 맞는 클래스를 각각 넣어준다.
3. init파일이 자동으로 만들어지면, all 위와같은 선언을 해서 각 클래스파일을 위와 같이 넣어준다.
4. 각 클래스파일가서 import로 연결해준다.
"""

---

main.py

import Product.productMenu as menu

def main():
    m = menu.Menu()
    m.run()


main()

---

productDao.py

class Dao:  # 저장소 : 저장, 검색, 수정, 삭제
    def __init__(self):  # 생성자로 리스트를 사용했다.
        self.prod = []

    def insert(self, p):  # 프로덕트
        print('제품추가. p는 Product 객체')
        self.prod.append(p)  # 파라메터로 받아온 프로덕트 객체를 추가해준다.

    def select(self, num):  # 제품번호로 검색하여 그 위치를 반환해준다.
        for p in self.prod:  # 반복문으로 제품의 인덱스와 내용을 가져올수 있게 선언해준다.
            if num == p.num:  # 파라메터값으로 받아온 숫자와 제품의 번호와 같으면
                return p  # 해당 인덱스를 리턴으로 반환해준다.

    def update(self, p):  # 제품 가격만 수정
        print('제품 p를 번호로 찾아서 새 정보로 수정')  # 제품이 있는가 확인하고, 있다면 수정이 가능하게 구현,
        res = self.select(p.num)
        if res != None:
            res.price = p.price
        else:
            print("없는 제품")

    def delete(self, num):
        print('제품 번호로 찾아서 삭제')
        res = self.select(num)
        if res != None:
            self.prod.remove(res)
        else:
            print("없는 제품")

    def selectAll(self):
        return self.prod

 

---

productMenu.py

import Product.productService as service


class Menu:
    def __init__(self):
        self.service = service.Service()

    def run(self):
        prompt = """1. 추가\n2. 검색\n3. 수정\n4. 삭제\n5. 전체목록\n6. 종료"""
        while True:
            print(prompt)
            nummber = int(input("숫자 입력 :"))
            if nummber == 1:
                self.service.addProduct()
            elif nummber == 2:
                self.service.getProduct()
            elif nummber == 3:
                self.service.editProduct()
            elif nummber == 4:
                self.service.delProduct()
            elif nummber == 5:
                self.service.printAll()
            elif nummber == 6:
                break

---

productService.py

import Product.productDao as dao
import Product.productVO as prod
# 4번. init 클래스파일 참조.

class Service:
    def __init__(self):
        self.dao = dao.Dao()

    def addProduct(self):
        print('제품명, 가격, 수량 입력받아 Product객체 생성 후 dao.insert()로 추가')
        name = input("제품명 :")
        price = int(input("가격 :"))
        a = int(input("수랑 :"))

        self.dao.insert(prod.Product(name, price, a))

    def getProduct(self):
        print('검색할 번호 입력받아 dao.select()로 검색')
        num = int(input("num :"))  # 제품 입력받아서
        res = self.dao.select(num)  # dao에서 찾을 수 있도록 .
        # 제품이 있는지 확인.
        if res == None:
            print('없는 제품')
        else:
            res.printProduct()

    def editProduct(self):
        print('수정할 제품 번호와 새 가격 입력받아 dao.update()로 수정')
        num = int(input("num :"))  # 제품 입력받아서
        price = int(input("new price :"))
        p = prod.Product('', price, 0)
        p.num = num
        self.dao.update(p)

    def delProduct(self):
        print('삭제할 제품 번호 입력받아 dao.delete()로 삭제')
        num = int(input("num :"))  # 제품 입력받아서
        self.dao.delete(num)

    def printAll(self):
        print('dao.selectAll()로 전체 검색한 결과 출력')
        datas = self.dao.selectAll()
        for p in datas:
            p.printProduct()

---

productVO.py

class Product:
    cnt = 0  # 클래스 변수. 제품의 개수 카운팅

    def __init__(self, name, price, amount):  # 객체가 생성 될때마다 호출된다.
        Product.cnt += 1  # 중복되지 않게 카운트해준다.
        self.num = Product.cnt
        self.name = name
        self.price = price
        self.amount = amount

    def printProduct(self):
        print('num:', self.num)
        print('name:', self.name)
        print('price:', self.price)
        print('amount:', self.amount)