def add_update_user():
if not LoginController.athenticate_user():
return redirect(url_for("google.login"))
resp = google.get("/oauth2/v2/userinfo")
email = resp.json()["email"]
message = "User created/updated successfully."
form = UserForm()
if form.validate_on_submit():
username = form.username.data
dept_id = form.dept_id.data
login_ip = form.login_ip.data
user = UserModel.find_by_username(username)
if user:
user.dept_id = dept_id
user.login_ip = login_ip
else:
user = UserModel(username, dept_id, login_ip)
try:
user.save()
except:
message = "Unable to save/update user"
flash(message)
return render_template('add_user.html', form=form, email=email)
def list_all():
if not LoginController.athenticate_user():
return redirect(url_for('logout'))
resp = google.get("/oauth2/v2/userinfo")
email = resp.json()["email"]
# all_agents = UserModel.find_all()
user = UserController()
all_agents = user.get_all()
return render_template('list_all.html', all_agents=all_agents, email=email)
