def init_db(config):
tables = [
Entry, Users, Tag, Image, Hop, Grain, Extract, HoppedExtract, Yeast,
Water, Misc, Mineral, Fining, Flavor, Spice, Herb, BJCPStyle,
BJCPCategory, MashTun, BoilKettle, EquipmentSet, MashProfile, MashStep,
MashStepOrder, Recipe, RecipeIngredient, Inventory, Comment, Role
] #, Permission]
for table in tables:
try:
table.createTable()
except OperationalError:
pass
else: # NEED TO FIX THIS
if table.__name__ == 'Users':
adef = config['ADMIN_USERNAME']
admin = Users(email=adef,
first_name=adef,
last_name=adef,
alias=adef)
admin.set_pass(config['PASSWORD_SALT'],
config['ADMIN_PASSWORD'])
admin.admin = True
# uncomment when you're sorted out your little permissions thingy
# for role in config['SYSTEM_ROLES']:
# r = Role(name=role)
# admin.addRole(config['SYSTEM_ROLES'].index(config['ADMIN']))
if table.__name__ == 'BJCPCategory':
process_bjcp_styles()
if table.__name__ == 'Inventory':
process_bt_database()
def create_user():
user_form = CreateUserForm()
if user_form.validate_on_submit():
user = Users(first_name = user_form.first_name.data,
last_name = user_form.last_name.data,
email = user_form.email.data,
password = generate_password(user_form.password.data),
alias = user_form.alias.data)
if user_form.avatar.data:
i = Image(url=user_form.alias.data)
user.avatar = i
flash("%s %s has been created" % (user.first_name, user.last_name))
return redirect(url_for('list_users'))
else:
user = {'first_name': '',
'last_name': '',
'email': '',
'password': '',
'avatar': ''}
return render_template('edit_user.html', data={'form': user_form,
'user': user})
