def generate_stock_product(self):
"""
Adding all product in stock
"""
stocks = []
restaurant_repo = repos.RestaurantRepository(config.db)
ingredient_repo = repos.IngredientRepository(config.db)
for restaurant in restaurant_repo.get_all():
for ingredient in ingredient_repo.get_all():
stocks.append(
models.ProductStock(restaurant=restaurant,
ingredient=ingredient,
quantity=randint(1, 50)))
return stocks
def pizza_composition(self):
"""
Generate the pizza ingredient
"""
compositions = []
product_repo = repos.ProductRepository(config.db)
ingredient_repo = repos.IngredientRepository(config.db)
for product_name, ingredients in config.RECIPE.items():
product = product_repo.get_by_name(product_name)
for iid in ingredients:
ingredient = ingredient_repo.get(iid)
# for ingredient in ingredient_repo.get_all():
compositions.append(
models.Composition(product=product[0],
ingredient=ingredient,
weight=Decimal(randint(1, 50))))
return compositions
def insert_ingredient(self, generate):
ingredient = repos.IngredientRepository(db)
ingredient.save_all(generate.gen_ingredient())
def insert_product(self, generate):
product = repos.ProductRepository(db)
ingredient = repos.IngredientRepository(db)
products, ingredients = generate.gen_product()
product.save_all(products)
ingredient.save_all(ingredients)
def pizza_ingredient(self, generate):
invoice = repos.IngredientRepository(db)
invoice.save_all(generate.gen_pizza_ingredient())