def init_security(user_datastore):
# Create any database tables that don't exist yet.
db.create_all()
# Create the Roles "admin" and "end-user" -- unless they already exist
user_datastore.find_or_create_role(name='superuser',
description='Administrator')
user_datastore.find_or_create_role(name='user', description='End user')
# Create two Users for testing purposes -- unless they already exists.
# In each case, use Flask-Security utility function to encrypt the password.
hashed_password = hash_password('password')
if not user_datastore.get_user('*****@*****.**'):
user_datastore.create_user(email='*****@*****.**',
password=hashed_password)
if not user_datastore.get_user('*****@*****.**'):
user_datastore.create_user(email='*****@*****.**',
password=hashed_password)
# Commit any database changes; the User and Roles must exist before we can add a Role to the User
db.session.commit()
# Give one User has the "end-user" role, while the other has the "admin" role. (This will have no effect if the
# Users already have these Roles.) Again, commit any database changes.
user_datastore.add_role_to_user('*****@*****.**', 'user')
user_datastore.add_role_to_user('*****@*****.**', 'superuser')
db.session.commit()
def forged():
db.drop_all()
db.create_all()
faker = Factory.create()
def generate_post(func_user, func_categorys, func_tags):
return Post(title=faker.sentence(),
body=faker.paragraph(),
user=func_user(),
category=func_categorys(),
tags=func_tags())
def generate_user():
return User(email=faker.email(),
username=faker.word(),
nickname=faker.name(),
password='******')
def generate_category():
return Category(name=faker.last_name(),
description=faker.sentence())
def generate_tag():
return Tag(name=faker.first_name())
def generate_comment(func_post):
return Comment(author=faker.first_name(),
email=faker.email(),
site='http://www.%s.com' % (faker.first_name()),
content=faker.sentence(),
post=func_post())
users = [generate_user() for i in xrange(10)]
db.session.add_all(users)
categorys = [generate_category() for i in xrange(10)]
db.session.add_all(categorys)
tags = [generate_tag() for i in xrange(30)]
db.session.add_all(tags)
def random_user(): return choice(users)
def random_category(): return choice(categorys)
def random_tags(): return sample(tags, randint(1, 5))
posts = [generate_post(random_user,
random_category, random_tags) for i in xrange(100)]
db.session.add_all(posts)
def random_post(): return choice(posts)
comments = [generate_comment(random_post) for i in xrange(1000)]
db.session.add_all(comments)
db.session.commit()
def create_db():
"""Create database for """
db.create_all()
def forged():
from app.core.models import db, User, Category, Post, Tag, Comment
from random import choice, sample, randint
db.drop_all()
db.create_all()
faker = Factory.create()
def generate_post(func_user, func_categorys, func_tags):
return Post(title=faker.sentence(),
body=faker.paragraph(),
user=func_user(),
category=func_categorys(),
tags=func_tags())
def generate_user():
return User(email=faker.email(),
username=faker.word(),
nickname=faker.name(),
password='******')
def generate_category():
return Category(name=faker.last_name(),
description=faker.sentence())
def generate_tag():
return Tag(name=faker.first_name())
def generate_comment(func_post):
return Comment(author=faker.first_name(),
email=faker.email(),
site='http://www.%s.com' % (faker.first_name()),
content=faker.sentence(),
post=func_post())
users = [generate_user() for i in xrange(10)]
db.session.add_all(users)
categorys = [generate_category() for i in xrange(10)]
db.session.add_all(categorys)
tags = [generate_tag() for i in xrange(30)]
db.session.add_all(tags)
def random_user(): return choice(users)
def random_category(): return choice(categorys)
def random_tags(): return sample(tags, randint(1, 5))
posts = [generate_post(random_user,
random_category, random_tags) for i in xrange(100)]
db.session.add_all(posts)
def random_post(): return choice(posts)
comments = [generate_comment(random_post) for i in xrange(1000)]
db.session.add_all(comments)
db.session.commit()
def InitializeDb(self, skeleton=False):
"""Create db schema and populate it with default inventory data"""
with self.app.app_context():
db.create_all()
if not skeleton:
main = Item(name="Main", root=True)
db.session.add(main)
main_type_group = IngredientGroup(
name="Main Type",
max_item=1,
min_item=1,
max_option=1,
min_option=1)
main.ingredientgroups.append(main_type_group)
db.session.add(main_type_group)
base_burger = Item(
name="Standard Burger(non-customizable)",
root=False,
price=12.99,
stock_unit=1)
db.session.add(base_burger)
main_type_group.options.append(base_burger)
base_burger_stock = Stock(name="Standard Burger", amount=1000)
db.session.add(base_burger_stock)
base_burger_stock.items.append(base_burger)
burger = Item(name="Customizable Burger", root=False, price=9.99)
db.session.add(burger)
main_type_group.options.append(burger)
bun_group = IngredientGroup(
name="Bun", max_item=3, min_item=2, max_option=1, min_option=1)
db.session.add(bun_group)
burger.ingredientgroups.append(bun_group)
muffin_bun = Item(
name="Muffin Bun",
root=False,
identical=True,
price=0.99,
stock_unit=1)
sesame_bun = Item(
name="Sesame Bun",
root=False,
identical=True,
price=0.99,
stock_unit=1)
standard_bun = Item(
name="Standard Bun", root=False, identical=True, stock_unit=1)
db.session.add(muffin_bun)
db.session.add(sesame_bun)
db.session.add(standard_bun)
bun_group.options.append(muffin_bun)
bun_group.options.append(sesame_bun)
bun_group.options.append(standard_bun)
muffin_bun_stock = Stock(name="Muffin Bun", amount=1000)
db.session.add(muffin_bun_stock)
muffin_bun_stock.items.append(muffin_bun)
sesame_bun_stock = Stock(name="Sesame Bun", amount=1000)
db.session.add(sesame_bun_stock)
sesame_bun_stock.items.append(sesame_bun)
standard_bun_stock = Stock(name="Standard Bun", amount=1000)
db.session.add(standard_bun_stock)
standard_bun_stock.items.append(standard_bun)
wrap = Item(
name="Wrap", root=False, identical=False, price=8.99, stock_unit=1)
db.session.add(wrap)
main_type_group.options.append(wrap)
wrap_stock = Stock(name="Wrap", amount=1000)
db.session.add(wrap_stock)
wrap_stock.items.append(wrap)
patty_group = IngredientGroup(
name="Patties", max_item=3, min_item=1, max_option=3, min_option=1)
db.session.add(patty_group)
burger.ingredientgroups.append(patty_group)
wrap.ingredientgroups.append(patty_group)
chicken_patty = Item(
name="Chicken Patty",
root=False,
identical=True,
price=0.99,
stock_unit=1)
beef_patty = Item(
name="Beef Patty",
root=False,
identical=True,
price=1.99,
stock_unit=1)
vegetarian_patty = Item(
name="Vegetarian Patty",
root=False,
identical=True,
price=0.99,
stock_unit=1)
db.session.add(chicken_patty)
db.session.add(beef_patty)
db.session.add(vegetarian_patty)
patty_group.options.append(chicken_patty)
patty_group.options.append(beef_patty)
patty_group.options.append(vegetarian_patty)
chicken_patty_stock = Stock(name="Chicken Patty", amount=1000)
db.session.add(chicken_patty_stock)
chicken_patty_stock.items.append(chicken_patty)
beef_patty_stock = Stock(name="Beef Patty", amount=1000)
db.session.add(beef_patty_stock)
beef_patty_stock.items.append(beef_patty)
vegetarian_patty_stock = Stock(name="Vegetarian Patty", amount=1000)
db.session.add(vegetarian_patty_stock)
vegetarian_patty_stock.items.append(vegetarian_patty)
ingredients = IngredientGroup(
name="Other Ingredients", max_item=5, max_option=5)
db.session.add(ingredients)
burger.ingredientgroups.append(ingredients)
wrap.ingredientgroups.append(ingredients)
tomato = Item(
name="Tomato", root=False, identical=True, price=0.99, stock_unit=1)
tomato_sauce = Item(
name="Tomato Sauce",
root=False,
identical=True,
price=0.99,
stock_unit=1)
bbq_sauce = Item(
name="BBQ Sauce",
root=False,
identical=True,
price=0.99,
stock_unit=1)
cheddar_cheese = Item(
name="Cheddar Cheese",
root=False,
identical=True,
price=0.99,
stock_unit=1)
db.session.add(tomato)
db.session.add(tomato_sauce)
db.session.add(bbq_sauce)
db.session.add(cheddar_cheese)
ingredients.options.append(tomato)
ingredients.options.append(tomato_sauce)
ingredients.options.append(bbq_sauce)
ingredients.options.append(cheddar_cheese)
tomato_stock = Stock(name="Tomato", amount=1000)
db.session.add(tomato_stock)
tomato_stock.items.append(tomato)
tomato_sauce_stock = Stock(name="Tomato Sauce", amount=1000)
db.session.add(tomato_sauce_stock)
tomato_sauce_stock.items.append(tomato_sauce)
bbq_sauce_stock = Stock(name="BBQ Sauce", amount=1000)
db.session.add(bbq_sauce_stock)
bbq_sauce_stock.items.append(bbq_sauce)
cheddar_cheese_stock = Stock(name="Cheddar Cheese", amount=1000)
db.session.add(cheddar_cheese_stock)
cheddar_cheese_stock.items.append(cheddar_cheese)
nuggets = Item(name="Nuggets", root=True, price=1.99)
db.session.add(nuggets)
nuggets_stock = Stock(name="Nuggets", amount=1000)
db.session.add(nuggets_stock)
nuggets_amount = IngredientGroup(
name="Nuggets Amount",
max_item=1,
min_item=1,
max_option=1,
min_option=1)
db.session.add(nuggets)
nuggets.ingredientgroups.append(nuggets_amount)
nuggets_3_pack = Item(
name="3-pack Nuggets",
root=False,
identical=True,
price=0,
stock_unit=3)
nuggets_6_pack = Item(
name="6-pack Nuggets",
root=False,
identical=True,
price=1,
stock_unit=6)
nuggets_12_pack = Item(
name="12-pack Nuggets",
root=False,
identical=True,
price=2,
stock_unit=12)
db.session.add(nuggets_3_pack)
db.session.add(nuggets_6_pack)
db.session.add(nuggets_12_pack)
nuggets_stock.items.append(nuggets_3_pack)
nuggets_stock.items.append(nuggets_6_pack)
nuggets_stock.items.append(nuggets_12_pack)
nuggets_amount.options.append(nuggets_3_pack)
nuggets_amount.options.append(nuggets_6_pack)
nuggets_amount.options.append(nuggets_12_pack)
fries = Item(name="Fries", root=True, price=1.99)
db.session.add(fries)
fries_stock = Stock(name="Fries (g)", amount=1000)
db.session.add(fries_stock)
fries_size = IngredientGroup(
name="fries Size",
max_item=1,
min_item=1,
max_option=1,
min_option=1)
db.session.add(fries_size)
fries.ingredientgroups.append(fries_size)
small_size = Item(
name="Small Fries",
root=False,
identical=True,
price=0,
stock_unit=150)
medium_size = Item(
name="Medium Fries",
root=False,
identical=True,
price=1,
stock_unit=200)
large_size = Item(
name="Large Fries",
root=False,
identical=True,
price=2,
stock_unit=250)
db.session.add(small_size)
db.session.add(medium_size)
db.session.add(large_size)
fries_stock.items.append(small_size)
fries_stock.items.append(medium_size)
fries_stock.items.append(large_size)
fries_size.options.append(small_size)
fries_size.options.append(medium_size)
fries_size.options.append(large_size)
sauce = IngredientGroup(name="Sauce", max_item=3, max_option=3)
db.session.add(sauce)
nuggets.ingredientgroups.append(sauce)
fries.ingredientgroups.append(sauce)
chilli_sauce_stock = Stock(name="Chilli Sauce", amount=1000)
db.session.add(chilli_sauce_stock)
tomato_sauce = Item(
name="Tomato Sauce",
root=False,
identical=True,
price=0,
max_item=1,
stock_unit=1)
bbq_sauce = Item(
name="BBQ Sauce", root=False, identical=True, price=1, stock_unit=1)
chilli_sauce = Item(
name="Chilli Sauce",
root=False,
identical=True,
price=1,
stock_unit=1)
db.session.add(tomato_sauce)
db.session.add(bbq_sauce)
db.session.add(chilli_sauce)
tomato_sauce_stock.items.append(tomato_sauce)
bbq_sauce_stock.items.append(bbq_sauce)
chilli_sauce_stock.items.append(chilli_sauce)
sauce.options.append(tomato_sauce)
sauce.options.append(bbq_sauce)
sauce.options.append(chilli_sauce)
coke = Item(name="Coke", root=True, price=1.99)
db.session.add(coke)
coke_stock = Stock(name="Coke (ml)", amount=1000)
db.session.add(coke_stock)
coke_size = IngredientGroup(
name="Coke Size",
max_item=1,
min_item=1,
max_option=1,
min_option=1)
db.session.add(coke_size)
coke.ingredientgroups.append(coke_size)
small_coke = Item(
name="Small Coke",
root=False,
identical=True,
price=0,
stock_unit=150)
medium_coke = Item(
name="Medium Coke",
root=False,
identical=True,
price=1,
stock_unit=250)
large_coke = Item(
name="Large Coke",
root=False,
identical=True,
price=2,
stock_unit=350)
db.session.add(small_coke)
db.session.add(medium_coke)
db.session.add(large_coke)
coke_stock.items.append(small_coke)
coke_stock.items.append(medium_coke)
coke_stock.items.append(large_coke)
coke_size.options.append(small_coke)
coke_size.options.append(medium_coke)
coke_size.options.append(large_coke)
sundaes = Item(name="Sundaes", root=True, price=2.99)
db.session.add(sundaes)
sundaes_stock = Stock(name="Sundaes (ml)", amount=1000)
db.session.add(sundaes_stock)
sundaes_flavor = IngredientGroup(
name="Sundaes Flavors",
max_item=1,
min_item=1,
max_option=1,
min_option=1)
db.session.add(sundaes)
sundaes.ingredientgroups.append(sundaes_flavor)
chocolate_flavor = Item(
name="Chocolate Sundaes", root=False, identical=True, price=0)
strawberry_flavor = Item(
name="Strawberry Sundaes", root=False, identical=True, price=0)
db.session.add(chocolate_flavor)
db.session.add(strawberry_flavor)
sundaes_flavor.options.append(chocolate_flavor)
sundaes_flavor.options.append(strawberry_flavor)
sundaes_size = IngredientGroup(
name="Sundaes Size",
max_item=1,
min_item=1,
max_option=1,
min_option=1)
db.session.add(sundaes_size)
sundaes.ingredientgroups.append(sundaes_size)
small_sundaes = Item(
name="Small Sundaes",
root=False,
identical=True,
price=0,
stock_unit=100)
medium_sundaes = Item(
name="Medium Sundaes",
root=False,
identical=True,
price=1,
stock_unit=150)
large_sundaes = Item(
name="Large Sundaes",
root=False,
identical=True,
price=2,
stock_unit=200)
db.session.add(small_sundaes)
db.session.add(medium_sundaes)
db.session.add(large_sundaes)
sundaes_stock.items.append(small_sundaes)
sundaes_stock.items.append(medium_sundaes)
sundaes_stock.items.append(large_sundaes)
sundaes_size.options.append(small_sundaes)
sundaes_size.options.append(medium_sundaes)
sundaes_size.options.append(large_sundaes)
db.session.commit()
def create_db():
"""Create database for """
db.create_all()
