def TestRecipe(
title=None,
slug=None,
short_description=None,
serving_size=ServingSize.FIVE_TO_SIX,
image=None,
ingredients=None,
instructions=None,
featured=False,
is_public=True,
added_by=None,
popularity=None,
):
# generating a unique title
if not title:
title_base = lorem_ipsum(3)
suffix = 1
not_unique = True
while not_unique:
title = "%s %s" % (title_base, suffix)
not_unique = Recipe.objects.filter(title=title)
suffix += 1
return Recipe.objects.create(
title=title,
slug=slug,
short_description=short_description or lorem_ipsum(3),
serving_size=serving_size,
image=image,
ingredients=ingredients or lorem_ipsum(7),
instructions=instructions or lorem_ipsum(7),
featured=featured,
is_public=is_public,
added_by=added_by or TestUser(),
)
def TestRecipeTag(name=None, type=TagType.MISCELLANEOUS, is_public=True, added_by=None):
# generating unique name
if not name:
name_base = lorem_ipsum(1)
suffix = 1
not_unique = True
while not_unique:
name = "%s%s" % (name_base, suffix)
not_unique = RecipeTag.objects.filter(name=name)
suffix += 1
return RecipeTag.objects.create(name=name, type=type, is_public=is_public, added_by=added_by)
def generate_test_data():
def make_ingredient(category):
ingredient = random.choice(category)
if category in (MEASURED_INGREDIENTS, LIQUID_INGREDIENTS):
units = random.choice(UNITS)
elif category == WEIGHED_INGREDIENTS:
units = random.choice(WEIGHTS)
else:
units = None
quantity = random.choice(QUANTITIES)
if quantity not in ('1', '1/2', '1/4', '1/8'):
if units:
if units in ('cup', 'lb'):
units += "s"
else:
ingredient += "s"
if units:
return " ".join([quantity, units, ingredient])
return " ".join([quantity, ingredient])
def make_ingredients_list():
ingredients = []
for i in range(random.randint(2, 4)):
ingredients.append(make_ingredient(MEASURED_INGREDIENTS))
for i in range(random.randint(1, 2)):
ingredients.append(make_ingredient(SINGLE_INGREDIENTS))
for i in range(random.randint(1, 2)):
ingredients.append(make_ingredient(WEIGHED_INGREDIENTS))
for i in range(random.randint(0, 1)):
ingredients.append(make_ingredient(LIQUID_INGREDIENTS))
random.shuffle(ingredients)
return "<br>".join(ingredients)
def make_title():
words = []
for word_group in (NAMES, ADJECTIVES, DAYS_OF_THE_WEEK, COOKING_METHOD,
SINGLE_INGREDIENTS, DISH):
use = random.randint(0, 1)
if use or word_group in (SINGLE_INGREDIENTS, DISH):
title_word = random.choice(word_group)
# we have to pull this one out individually because title-
# casing words like "Amy's" does strange things with the
# capitalization
if word_group == SINGLE_INGREDIENTS:
title_word = title_word.title()
words.append(title_word)
return " ".join(words)
def make_instructions():
instructions = []
for inst in (OPENING_INSTRUCTIONS, SECOND_INSTRUCTIONS,
MIDDLE_INSTRUCTIONS, ENDING_INSTRUCTIONS):
number_to_choose = random.randint(1, 2)
for i in range(number_to_choose):
instructions.append(random.choice(inst))
return " ".join(instructions)
print "Making users..."
users = [TestUser() for i in range(30)]\
print "Creating tags..."
call_command('loaddata', 'recipe_tags')
for i in range(10):
TestRecipeTag(is_public=bool(random.randint(0, 1)),
added_by=random.choice(users))
print "Adding recipes and ratings..."
for i in range(250):
title = make_title()
while Recipe.objects.filter(title=title):
title = make_title()
featured = random.randint(1, 20) == 1
is_public = random.randint(1, 30) != 1
recipe = TestRecipe(title=title,
short_description=lorem_ipsum(random.randint(5, 8)),
serving_size=random.randint(0, 9),
ingredients=make_ingredients_list(),
instructions=make_instructions(),
featured=featured, is_public=is_public,
added_by=random.choice(users))
selected_tags = []
num_tags = random.randint(1, 4)
tags = RecipeTag.objects.all()
while len(selected_tags) < num_tags:
tag = random.choice(tags)
if tag not in selected_tags:
selected_tags.append(tag)
for tag in selected_tags:
recipe.tags.add(tag)
rated_users = []
num_ratings = random.randint(0, 20)
while len(rated_users) < num_ratings:
rated_by = random.choice(users)
if rated_by not in rated_users:
rated_users.append(rated_by)
for user in rated_users:
vote = random.randint(1, 10) != 10
TestRating(recipe=recipe, rated_by=user, vote=vote)
print "30 users and 250 recipes created. Done!"
