示例#1
0
    def test_save_to_database(self):

        user = User("*****@*****.**", "abc123", "usertwo")
        user.save_to_database()

        job = Job("Server")
        job.save_to_database()

        user_job = UserJob(user, job)

        user_job.save_to_database()
示例#2
0
def add_user():
    form = AddUserForm(request.form)
    users = User.query.all()
    if form.validate_on_submit():
        username = request.form['username']
        employee_number = request.form['employee_number']
        email = request.form['email']
        password = request.form['password']
        admin_password = request.form['admin_password']
        admin_status = check_admin_password(admin_password)
        user = User(employee_number=employee_number,
                    username=username,
                    email=email,
                    password=password,
                    admin=admin_status)
        user.save_to_database()
        return redirect(url_for('.user_list', users=users))

    return render_template('admin/add_user.html', users=users, form=form)