def admin_login(): """ Login for an admin account """ if g.admin is not None: return redirect(url_for('admin_home', admin_id=g.admin['project_id'])) form = LoginForm(request.form) if form.validate_on_submit(): # On submit, grab name & password project_name = form.project_name.data password = form.password.data # Try login db = DB() resp = db.auth(project_name, password) if resp['status'] and resp['admin']: session['admin_project_id'] = resp['project_id'] admin_detail = db.get_project_detail(session['admin_project_id']) admin_id = admin_detail['project_id'] return redirect(url_for('admin_home', admin_id=admin_id)) elif not resp['admin']: flash(u'Invalid admin account!') else: flash(resp['message']) return render_template('admin_login.html', form=form)
def login(): """ Handles project account authentication """ if g.project is not None: return redirect(url_for('home', project_name=g.project['project_name'])) form = LoginForm(request.form) if form.validate_on_submit(): # On submit, grab name & password project_name = form.project_name.data password = form.password.data # Try login db = DB() resp = db.auth(project_name, password) if resp['status']: session['project_id'] = resp['project_id'] project_detail = db.get_project_detail(session['project_id']) project_name = project_detail['project_name'] return redirect(url_for('home', project_name=project_name)) else: flash(resp['message']) return render_template('login.html', form=form)