def admin_administration(selected_table):
if 'username' not in session or get_user_rol(
session['username']) != 'Admin':
return redirect('/')
else:
logout = Markup(
'<p class="nav-link text-warning" style="font-size: 160%">' +
str(escape(session['username'])) +
'</p> <a class="nav-link text-warning" href="/logout" style="font-size: 160%">\
<span class="glyphicon glyphicon-user"></span>\
Log-out</a>') # Logout HTML link
if selected_table == 'user':
navigation = '<ul class="nav nav-tabs"><li class="nav-item"><a class="nav-link active" ' \
'href="/administration/user">Users</a></li><li class="nav-item"><a class="nav-link" ' \
'href="/administration/model">Models</a></li></ul>'
elif selected_table == 'model':
navigation = '<ul class="nav nav-tabs"><li class="nav-item"><a class="nav-link" ' \
'href="/administration/user">Users</a></li><li class="nav-item"><a class="nav-link active" ' \
'href="/administration/model">Models</a></li></ul>'
else:
return make_response(
render_template(
'ERROR.html',
error="The selected_table or URL does not exist"))
table = generate_table_from_db(selected_table)
return make_response(
render_template(
'administration.html',
navigation=Markup(navigation),
selected_table=Markup(table),
signin=logout)) # Redirect to admin, show logout link
def admin_insert(selected_table):
if get_user_rol(session['username']) == 'Admin':
if selected_table == 'user':
new_user = request.form['username']
new_password = request.form['password']
new_email = request.form['email']
rol = request.form['rol']
insert_new_user(new_user, new_email, new_password, rol)
return redirect('/')
elif selected_table == 'model':
disease = request.form['disease']
model_type = request.form['model_type']
dataset_description = request.files['dataset_description']
model_path = request.files['model_path']
test_data_path = request.files['test_data_path']
filename = time.strftime("%Y-%m-%d_%H%M%S") + secure_filename(
test_data_path.filename)
test_data_path.save(
os.path.join(app.config['MODEL_DATA_TEST_DIR'], filename))
new_model(
disease, model_type, dataset_description, model_path,
os.path.join(app.config['MODEL_DATA_TEST_DIR'], filename))
return redirect('/administration/model')
return make_response(
render_template('ERROR.html', error="Forbidden access"))
def generate_records_table(username):
rol = get_user_rol(username)
if rol == "Doctor":
cols = ('PATIENT ID', 'DATE', 'DATA', 'MODEL DISEASE', 'MODEL TYPE', 'OUTPUT')
body = '<table class="table" id="table">\
<thead>'
body += new_head(cols)
body += '</thead> \
<tbody>'
conn = Connection()
prediction = conn.do_query_mult_col(
'SELECT PRE.patient_id, PRE.datetime, PRE.expression_file_path, M.disease, M.model_type, PRE.result FROM prediction PRE, user U, model M WHERE U.email=\"' + username + '\" and U.id=PRE.user_id and PRE.model_id=M.id;')
if prediction is not None: # There are data to show
for row in prediction:
body += new_row(row)
body += ' </tbody>\
</table>'
return body
elif rol == "Admin":
cols = ('PREDICTION ID', 'USER ID', 'DATE', 'MODEL')
body = '<table class="table" id="table">\
<thead>'
body += new_head(cols)
body += '</thead> \
<tbody>'
conn = Connection()
prediction = conn.do_query_mult_col(
'SELECT PRE.id, PRE.user_id, PRE.datetime, PRE.model_id FROM prediction PRE;')
if prediction is not None: # There are data to show
for row in prediction:
body += new_row(row)
body += ' </tbody>\
</table>'
return body
def register_page():
if request.method == 'POST':
firstname = request.form['firstname']
lastname = request.form['lastname']
email = request.form['email']
if user_registration(firstname, lastname, email,
request.form['password'],
request.form['repeated_password'],
"Doctor"): # If user registered sucessfully
session['username'] = email # Set user session
if get_user_rol(session['username']) == "Doctor":
return redirect(url_for('main_page'))
elif get_user_rol(session['username']) == "Admin":
return redirect('/administration/user')
else:
return make_response(
render_template('register.html', error='Registration error'))
return redirect(url_for('main_page'))
def update_user():
if get_user_rol(session['username']) == 'Admin':
uid = request.form['uid']
rol = request.form['rol']
print('RECIEVED UID AND ROL:')
print(uid)
print(rol)
update_user_rol(uid, rol)
return redirect('/')
return make_response(
render_template('ERROR.html', error="Forbidden access"))
def delete_user(selected_table, uid):
if get_user_rol(session['username']) == 'Admin':
if delete_by_id(selected_table, uid):
# It was possible to delete
return redirect("/administration/" + selected_table)
else:
return make_response(
render_template('ERROR.html',
error="Invalid action: Cannot delete"))
else:
return make_response(
render_template('ERROR.html', error="Forbidden access"))
def main_page():
cancer_options, model_options = get_models_html_selector()
if 'username' in session: # If user already logged in
if get_user_rol(session['username']) == 'Admin':
return redirect('/administration')
logout = Markup(
'<p class="nav-link text-warning" style="font-size: 160%">' +
str(escape(session['username'])) +
'</p> <a class="nav-link text-warning" href="/logout" style="font-size: 160%">\
<span class="glyphicon glyphicon-user"></span>\
Log-out</a>') # Logout HTML link
nav = Markup(
'<li class="nav-item active"> <a class ="nav-link text-warning active" style=" font-weight: bold; '
'font-size: 160%" href="" > '
'Predictor </a></li><li class="nav-item"> <a class ="nav-link text-warning" style="font-size: '
'160%" href="/records" > Records </a></li>')
return make_response(
render_template(
'index.html',
navbar=nav,
signin=logout,
cancer_options=Markup(cancer_options),
model_options=Markup(model_options),
results=Markup(
predict_data))) # Redirect to home, show logout link
else:
signin = Markup(
' <a class="nav-link text-warning" href="/login" style="font-size: 160%">\
<span class="glyphicon glyphicon-user"></span>\
Sign-in/Log-in</a>'
) # If user not logged in, show login link
anonymous_nav = Markup(
'<li class="nav-item active"><a class ="nav-link text-warning active" style="font-weight: bold; '
'font-size: 160%" href="" > '
'Predictor </a></li>')
# requirements = generate_table_data_format(6)
cancer_options, model_options = get_models_html_selector()
return render_template(
'index.html',
navbar=anonymous_nav,
signin=signin, # requirements=requirements,
cancer_options=Markup(cancer_options),
model_options=Markup(model_options),
results=Markup(predict_data)
) # Redirect to home, show signin link if not logged in.
def records_page():
rol = get_user_rol(session['username'])
if 'username' not in session or rol != 'Doctor':
return make_response(
render_template('ERROR.html', error="Forbidden access")
) # Redirect to home, show logout link
else:
logout = Markup(
'<p class="nav-link text-warning" style="font-size: 160%">' +
str(escape(session['username'])) +
'</p> <a class="nav-link text-warning" href="/logout" style="font-size: 160%">\
<span class="glyphicon glyphicon-user"></span>\
Log-out</a>') # Logout HTML link
table = generate_records_table(session['username'])
return make_response(
render_template(
'records.html', signin=logout,
table=Markup(table))) # Redirect to records, show logout link
def login():
global predict_data
predict_data = ""
if request.method == 'POST':
user = request.form['user']
password = request.form['password']
if user_validation(user, password): # If user in system
session['username'] = user # Set user session
rol = get_user_rol(session['username'])
if rol == "Doctor":
return redirect(url_for('main_page'))
elif rol == "Admin":
return redirect('/administration/user')
else:
return make_response(
render_template('login.html', message='Login error')
) # If user not in db, show login error
else:
return redirect(url_for('main_page'))
def admin_statistics(selected_content):
if 'username' not in session or get_user_rol(
session['username']) != 'Admin':
return make_response(
render_template('ERROR.html'),
error="Forbidden Access") # Redirect to home, show logout link
else:
logout = Markup(
'<p class="nav-link text-warning" style="font-size: 160%">' +
str(escape(session['username'])) +
'</p> <a class="nav-link text-warning" href="/logout" style="font-size: 160%">\
<span class="glyphicon glyphicon-user"></span>\
Log-out</a>') # Logout HTML link
if selected_content == 'tables':
navigation = '<ul class="nav nav-tabs"><li class="nav-item"><a class="nav-link active" ' \
'href="/statistics/tables">Tables</a></li><li class="nav-item"><a class="nav-link" ' \
'href="/statistics/graphs">Graphs</a></li></ul>'
content = generate_records_table(session['username'])
elif selected_content == 'graphs':
navigation = '<ul class="nav nav-tabs"><li class="nav-item"><a class="nav-link" ' \
'href="/statistics/tables">Tables</a></li><li class="nav-item"><a class="nav-link active" ' \
'href="/statistics/graphs">Graphs</a></li></ul>'
content = hist_from_db()
else:
return make_response(
render_template(
'ERROR.html',
error="The selected_table or URL does not exist"))
return make_response(
render_template(
'statistics.html',
content=Markup(content),
navigation=Markup(navigation),
signin=logout)) # Redirect to admin, show logout link
def admin_administration_home():
if get_user_rol(session['username']) == 'Admin':
return redirect('/administration/user')
else:
return redirect('/')
def admin_statistics_home():
if get_user_rol(session['username']) == 'Admin':
return redirect('/statistics/tables')
else:
return redirect('/')