Example #1
0
def register():
    form = RegisterForm(request.form, csrf_enabled=False)
    if form.validate_on_submit():
        default_role = user_datastore.find_role('Contractor')
        new_user = user_datastore.create_user(username=form.username.data,
                                              email=form.email.data,
                                              password=form.password.data)
        user_datastore.add_role_to_user(new_user, default_role)
        db.session.commit()
        flash("Thank you for registering. You can now log in.", 'success')
        return redirect(url_for('public.home'))
    else:
        flash_errors(form)
    return render_template('public/register.html', form=form)
Example #2
0
                password='******',
                active=1, email='*****@*****.**')
    User.create(username='******',
                password='******',
                active=1, email='*****@*****.**')
    User.create(username='******',
                password='******',
                active=1, email='*****@*****.**')
    User.create(username='******',
                password='******',
                active=1, email='*****@*****.**')

users = User.query.all()

for user in users:
    if user.username == 'contractor':
        role = user_datastore.find_role('Contractor')
        user_datastore.add_role_to_user(user, role)
    elif user.username == 'employee':
        role = user_datastore.find_role('Employee')
        user_datastore.add_role_to_user(user, role)
    elif user.username == 'arbiter':
        role = user_datastore.find_role('Arbiter')
        user_datastore.add_role_to_user(user, role)
    elif user.username == 'staff':
        role = user_datastore.find_role('Staff')
        user_datastore.add_role_to_user(user, role)
    elif user.username == 'admin':
        role = user_datastore.find_role('Admin')
        user_datastore.add_role_to_user(user, role)
    db.session.commit()