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 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_order(self):
"""
Generate the order
"""
results = []
status_repo = repos.StatusRepository(config.db)
actor_repo = repos.ActorRepository(config.db)
restaurant_repo = repos.RestaurantRepository(config.db)
for i in range(15):
actor = choice(actor_repo.get_all())
results.append(
models.Order(order_date=self.fake.fake_date(),
status=choice(status_repo.get_all()),
actor=actor,
restaurant=choice(restaurant_repo.get_all()),
address=actor.address))
return results
def gen_employee_list(self):
"""
Generate the employee
"""
employee = []
restaurant_repo = repos.RestaurantRepository(config.db)
status_repo = repos.StatusRepository(config.db)
for i in range(50):
employee.append(
models.Employee(
social_security_numb=self.fake.fake_number_random(
1, 99999999999999),
quality=self.fake.fake_quality()[0]['quality'],
date_entry=self.fake.fake_date(),
status=choice(status_repo.get_all()),
restaurant=choice(restaurant_repo.get_all()),
actor=models.Actor(
first_name=self.fake.fake_first_name(),
last_name=self.fake.fake_last_name(),
authentication_password=self.fake.fake_password(),
address=models.Address(*self.fake.fake_address()),
phone=models.Phone(phone=self.fake.fake_phone()),
mail=models.Email(mail=self.fake.fake_mail()))))
return employee
def insert_restaurant_list(self, generate):
restaurants = repos.RestaurantRepository(db)
restaurants.save_all(generate.gen_restaurants_list())