def gen_product_stock(self):
"""
Adding the product in stock
"""
product_stock = []
restaurant_repo = repos.RestaurantRepository(config.db)
product_repo = repos.ProductRepository(config.db)
for restaurant in restaurant_repo.get_all():
for product in product_repo.get_all():
product_stock.append(
models.ProductStock(restaurant=restaurant,
ingredient=product,
quantity=randint(10, 100)))
return product_stock
def generate_shopping(self):
"""
Generate the shopping market
"""
shoppings = []
order_repository = repos.OrderRepository(config.db)
product_repository = repos.ProductRepository(config.db)
for i in range(30):
shoppings.append(
models.ShoppingCart(order=choice(order_repository.get_all()),
product=choice(
product_repository.get_all()),
quantity=1,
price=self.fake.fake_price()))
return shoppings
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_product_barcode(self, generate):
product = repos.ProductRepository(db)
product.save_all(generate.gen_product_barcode())
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_product(self, generate):
invoice = repos.ProductRepository(db)
invoice.save_all(generate.gen_pizza())