def init():
# create the database
db.configure_mappers()
db.create_all()
# create the super user
u = User(name=settings.SUPER_USER,
email=settings.SUPER_USER_EMAIL,
password=settings.SUPER_USER_PASSWORD,
admin=True,
super_user=True)
# optionally add super user apikey
if getattr(settings, 'SUPER_USER_APIKEY', None):
u.apikey = settings.SUPER_USER_APIKEY
db_session.add(u)
# load sql extensions + functions
for sql in load_sql():
db_session.execute(sql)
# load built-in sous-chefs
for sc in load_sous_chefs():
sc = sous_chef_schema.validate(sc)
s = SousChef(**sc)
db_session.add(s)
# commit
db_session.commit()
def sous_chefs(org):
"""
(Re)load all sous chefs.
"""
# load all sous-chefs
for sc, fp in init.load_sous_chefs():
sc = sous_chef_schema.validate(sc, fp)
sc['org_id'] = org.id
sc_obj = db.session.query(SousChef).filter_by(slug=sc['slug']).first()
if not sc_obj:
log.info('Creating sous chef: "{slug}"'.format(**sc))
sc_obj = SousChef(**sc)
else:
log.warning('Updating sous chef: "{slug}"'.format(**sc))
sc = sous_chef_schema.update(sc_obj.to_dict(), sc)
for name, value in sc.items():
setattr(sc_obj, name, value)
db.session.add(sc_obj)
db.session.commit()
return org
def sous_chefs(org):
"""
(Re)load all sous chefs.
"""
# load all sous-chefs
for sc, fp in init.load_sous_chefs():
sc = sous_chef_schema.validate(sc, fp)
sc['org_id'] = org.id
sc_obj = db.session.query(SousChef).filter_by(
slug=sc['slug']).first()
if not sc_obj:
log.info('Creating sous chef: "{slug}"'.format(**sc))
sc_obj = SousChef(**sc)
else:
log.warning('Updating sous chef: "{slug}"'.format(**sc))
sc = sous_chef_schema.update(sc_obj.to_dict(), sc)
for name, value in sc.items():
setattr(sc_obj, name, value)
db.session.add(sc_obj)
db.session.commit()
return org
def test_schema(self):
for sc, fp in load_sous_chefs(SOUS_CHEF_DIR, incl_internal=False):
sous_chef_schema.validate(sc, fp)
assert True
def test_schema(self):
for sc, fp in load_sous_chefs(SOUS_CHEF_DIR, incl_internal=False):
sous_chef_schema.validate(sc, fp)
assert True
def run(opts, **kw):
for sc, fp in load_sous_chefs(sous_chef_dir=opts.module_path, incl_internal=False):
print sc_docs.create(sc, fp, opts.format)
def run(opts, **kwargs):
# create the database
try:
with app.app_context():
echo('Creating database "{}"'.format(settings.SQLALCHEMY_DATABASE_URI),
no_color=opts.no_color)
db.configure_mappers()
db.create_all()
# create the super user
u = User.query.filter_by(email=settings.SUPER_USER_EMAIL).first()
if not u:
echo('Creating super user "{}"'.format(settings.SUPER_USER_EMAIL),
no_color=opts.no_color)
u = User(name=settings.SUPER_USER,
email=settings.SUPER_USER_EMAIL,
password=settings.SUPER_USER_PASSWORD,
admin=True,
super_user=True)
# optionally add super user apikey
if getattr(settings, 'SUPER_USER_APIKEY', None):
u.apikey = settings.SUPER_USER_APIKEY
else:
echo('Updating super user "{}"'.format(settings.SUPER_USER_EMAIL),
no_color=opts.no_color)
u.name=settings.SUPER_USER,
u.email=settings.SUPER_USER_EMAIL,
u.password=settings.SUPER_USER_PASSWORD,
u.admin=True
super_user=True
db.session.add(u)
echo('(Re)Loading SQL Extensions', no_color=opts.no_color)
# load sql extensions + functions
for sql in load_sql():
db.session.execute(sql)
# load built-in sous-chefs
for sc in load_sous_chefs():
sc = sous_chef_schema.validate(sc)
sc_obj = db.session.query(SousChef).filter_by(slug=sc['slug']).first()
if not sc_obj:
echo('Importing Sous Chef "{}"'.format(sc['slug']),
no_color=opts.no_color)
sc_obj = SousChef(**sc)
else:
echo('Updating Sous Chef "{}"'.format(sc['slug']),
no_color=opts.no_color)
sc = sous_chef_schema.update(sc_obj.to_dict(), sc)
# udpate
for name, value in sc.items():
setattr(sc_obj, name, value)
db.session.add(sc_obj)
# commit
db.session.commit()
db.session.close()
except Exception as e:
db.session.rollback()
db.session.close()
raise e