def profile():
# Handle this case because isAuthenticated only applies when LOGIN_REQUIRED is true
if current_user.is_anonymous:
flash("You must be a user to access %s" % request.path, "warning")
return redirect(url_for('main.index'))
myagents = current_user.agents
changePasswordForm = ChangePasswordForm(prefix="change-password")
displaySettingsForm = DisplaySettingsForm(prefix="display-settings", results_per_page=current_user.results_per_page, \
preview_length=current_user.preview_length)
displaySettingsForm.results_per_page.choices = [(25,25), (50,50), (75,75), (100,100)]
displaySettingsForm.preview_length.choices = [(25,25), (50,50), (75,75), (100,100)]
generateTokenForm = GenerateTokenForm()
agentNameForm = AgentNameForm()
if changePasswordForm.changePassword.data and changePasswordForm.validate_on_submit():
user = User.query.get(current_user.id)
user.set_password(changePasswordForm.password.data)
db.session.commit()
flash('Your password has been changed.', "success")
return redirect(url_for('user.profile'))
if displaySettingsForm.updateDisplaySettings.data and displaySettingsForm.validate_on_submit():
user = User.query.get(current_user.id)
user.results_per_page = displaySettingsForm.results_per_page.data
user.preview_length = displaySettingsForm.preview_length.data
db.session.commit()
flash("Display settings updated.", "success")
return redirect(url_for('user.profile'))
if agentNameForm.change_name.data and agentNameForm.validate_on_submit():
return redirect(url_for('user.profile'))
return render_template("user/profile.html", changePasswordForm=changePasswordForm, displaySettingsForm=displaySettingsForm, \
agents=myagents, generateTokenForm=generateTokenForm, agentNameForm=agentNameForm)
def changeAgentName(agent_id):
agentNameForm = AgentNameForm()
if agentNameForm.validate_on_submit():
myAgent = Agent.load_agent(agent_id)
oldname = myAgent.friendly_name
myAgent.friendly_name = agentNameForm.agent_name.data
db.session.commit()
flash("Agent name changed from %s to %s" % (oldname, myAgent.friendly_name), "success")
return redirect(request.referrer)
else:
flash("Couldn't change agent name", "danger")
return redirect(request.referrer)
def change_agent_name(agent_id):
agentNameForm = AgentNameForm()
if agentNameForm.validate_on_submit():
myAgent = Agent.load_agent(agent_id)
oldname = myAgent.friendly_name
myAgent.friendly_name = agentNameForm.agent_name.data
db.session.commit()
flash(f"Agent name changed from {oldname} to {myAgent.friendly_name}",
"success")
else:
flash("Couldn't change agent name", "danger")
return redirect(url_for("user.profile"))
def newAgent():
newAgentForm = AgentNameForm()
if newAgentForm.validate_on_submit():
myAgent = Agent(user_id=current_user.id, agentid=Agent.generate_agentid(), token=Agent.generate_token(), \
friendly_name=newAgentForm.agent_name.data)
db.session.add(myAgent)
db.session.commit()
flash("New Agent named %s created. Agent ID: %s Agent Token: %s" \
% (myAgent.friendly_name, myAgent.agentid, myAgent.token), "success")
return redirect(request.referrer)
else:
flash("Couldn't create new agent", "danger")
return redirect(request.referrer)
def new_agent():
newAgentForm = AgentNameForm()
if newAgentForm.validate_on_submit():
myAgent = Agent(
user_id=current_user.id,
agentid=Agent.generate_agentid(),
token=Agent.generate_token(),
friendly_name=newAgentForm.agent_name.data,
)
db.session.add(myAgent)
db.session.commit()
flash(
f"New Agent named {myAgent.friendly_name} created. Agent ID: {myAgent.agentid} Agent Token: {myAgent.token}",
"success",
)
else:
flash("Couldn't create new agent", "danger")
return redirect(url_for("user.profile"))
def profile():
myagents = current_user.agents
changePasswordForm = ChangePasswordForm(prefix="change-password")
displaySettingsForm = DisplaySettingsForm(
prefix="display-settings",
results_per_page=current_user.results_per_page,
preview_length=current_user.preview_length,
result_format=current_user.result_format,
)
displaySettingsForm.results_per_page.choices = [
(25, 25),
(50, 50),
(75, 75),
(100, 100),
]
displaySettingsForm.preview_length.choices = [
(25, 25),
(50, 50),
(75, 75),
(100, 100),
]
displaySettingsForm.result_format.choices = [(0, "Pretty"), (1, "Raw")]
generateTokenForm = GenerateTokenForm()
agentNameForm = AgentNameForm()
if (changePasswordForm.changePassword.data
and changePasswordForm.validate_on_submit()):
user = User.query.get(current_user.id)
user.set_password(changePasswordForm.password.data)
db.session.commit()
flash("Your password has been changed.", "success")
elif (displaySettingsForm.updateDisplaySettings.data
and displaySettingsForm.validate_on_submit()):
user = User.query.get(current_user.id)
user.results_per_page = displaySettingsForm.results_per_page.data
user.preview_length = displaySettingsForm.preview_length.data
user.result_format = displaySettingsForm.result_format.data
db.session.commit()
flash("Display settings updated.", "success")
return render_template(
"user/profile.html",
changePasswordForm=changePasswordForm,
displaySettingsForm=displaySettingsForm,
agents=myagents,
generateTokenForm=generateTokenForm,
agentNameForm=agentNameForm,
)