def settings():
if request.method == 'GET':
# GET USER INFO FOR SETTINGS
user = controls.get_user_info(session.get('username'))
if not user:
return redirect(url_for('logout'))
authorized_users = controls.get_authorized_users()
lead = controls.get_lead(session.get('username'))
if len(lead) == 0:
lead={'first_name':'', 'last_name':'', 'username':''}
else:
lead=lead[0]
return render_template('settings.html', authorized_users = authorized_users, lead=lead['username'], user=user)
elif request.method== 'POST':
# UPDATE USER INFO WITH: ALL VALUES FROM THE FORM
lead_id = request.form['lead'] if request.form['lead'] != 'None' else None
payload ={'first_name':request.form['first_name'],
'last_name':request.form['last_name'],
'lead_id':lead_id}
if request.form['password'] != '':
payload['password']=request.form['password']
if 'auth' in request.form:
payload['authorized']=0
payload['auth_token']=None
session.pop('shard_id')
session.pop('auth_token')
session['authorized'] = 0
payload['use_email'] = request.form['use_email']
#payload['use_email'] = 1 if 'use_email' in request.form else 0
controls.update_user(session.get('username'), **payload)
flash('Settings Updated Successfully')
session['first_name'] = request.form['first_name']
session['last_name'] = request.form['last_name']
return redirect(url_for('settings', lead=lead_id))
def settings():
if request.method == 'GET':
# GET USER INFO FOR SETTINGS
user = controls.get_user_info(session.get('username'))
if not user:
return redirect(url_for('logout'))
authorized_users = controls.get_authorized_users()
lead = controls.get_lead(session.get('username'))
if len(lead) == 0:
lead = {'first_name': '', 'last_name': '', 'username': ''}
else:
lead = lead[0]
return render_template('settings.html',
authorized_users=authorized_users,
lead=lead['username'],
user=user)
elif request.method == 'POST':
# UPDATE USER INFO WITH: ALL VALUES FROM THE FORM
lead_id = request.form[
'lead'] if request.form['lead'] != 'None' else None
payload = {
'first_name': request.form['first_name'],
'last_name': request.form['last_name'],
'lead_id': lead_id
}
if request.form['password'] != '':
payload['password'] = request.form['password']
if 'auth' in request.form:
payload['authorized'] = 0
payload['auth_token'] = None
session.pop('shard_id')
session.pop('auth_token')
session['authorized'] = 0
payload['use_email'] = request.form['use_email']
#payload['use_email'] = 1 if 'use_email' in request.form else 0
controls.update_user(session.get('username'), **payload)
flash('Settings Updated Successfully')
session['first_name'] = request.form['first_name']
session['last_name'] = request.form['last_name']
return redirect(url_for('settings', lead=lead_id))
def register_user():
"""Create a user in the database"""
error = None
# REST CALLS
authorized_users = controls.get_authorized_users()
if request.method == 'POST':
if len(request.form['username'].strip()) <= 0:
error = 'Username cannot be blank'
else:
if controls.user_exists(request.form['username'].strip()):
error = 'Username already exists'
elif len(request.form['password'].strip()) <= 0:
error = 'Password cannot be blank'
elif not lib.valid_email(request.form['email_address']):
error='Invalid Email Address'
else:
if request.form['lead'] == 'None':
lead = None
else:
lead = request.form['lead']
if request.form['use_email'] == 'True':
use_email = 1
else:
use_email = 0
controls.create_user(request.form['username'],
request.form['password'],
request.form['first_name'],
request.form['last_name'],
request.form['email_address'],
lead,
use_email)
flash("New User was successfully registered")
session['logged_in'] = True
session['first_name'] = request.form['first_name']
session['last_name'] = request.form['last_name']
session['username'] = request.form['username']
return redirect(url_for('login'))
return render_template('register.html',
error=error,
authorized_users=authorized_users)
def register_user():
"""Create a user in the database"""
error = None
# REST CALLS
authorized_users = controls.get_authorized_users()
if request.method == 'POST':
if len(request.form['username'].strip()) <= 0:
error = 'Username cannot be blank'
else:
if controls.user_exists(request.form['username'].strip()):
error = 'Username already exists'
elif len(request.form['password'].strip()) <= 0:
error = 'Password cannot be blank'
elif not lib.valid_email(request.form['email_address']):
error = 'Invalid Email Address'
else:
if request.form['lead'] == 'None':
lead = None
else:
lead = request.form['lead']
if request.form['use_email'] == 'True':
use_email = 1
else:
use_email = 0
controls.create_user(request.form['username'],
request.form['password'],
request.form['first_name'],
request.form['last_name'],
request.form['email_address'], lead,
use_email)
flash("New User was successfully registered")
session['logged_in'] = True
session['first_name'] = request.form['first_name']
session['last_name'] = request.form['last_name']
session['username'] = request.form['username']
return redirect(url_for('login'))
return render_template('register.html',
error=error,
authorized_users=authorized_users)
def index():
error_list = None
# CHECK IF THE USER IS LOGGED IN, OTHERWISE REDIRECT
if not session.get('logged_in'):
return redirect(url_for('login'))
# GET THE USER INFO, REDIRECT IF NOTHING IS RETURNED
user = controls.get_user_info(session.get('username'))
if not user:
redirect(url_for('logout'))
authorized_users = controls.get_authorized_users()
# RETURNS A JSON ObJECT OF USERNAME, FIRST AND LAST
lead = controls.get_lead(session.get('username'))
if len(lead) == 0:
lead = {'username': '', 'first_name': '', 'last_name': ''}
else:
lead = lead[0]
notebooks = []
if session.get('auth_token'):
# CHECK AND SEE IF USER HAS SELECTED NOTEBOOKS
if user['notebook_ids'] is not None and len(user['notebook_ids']) > 0:
guid_list = user['notebook_ids'].split(',')
flash("Getting Notebooks and Notes")
notebooks = get_notebooks(session.get('auth_token'),
session.get('shard_id'), guid_list)
else:
return redirect(url_for('configure', guid_list=None))
# POST REQUESTS
if request.method == 'POST':
error_list = []
note_list = request.form.getlist('note')
# GET THE NOTE CONTENT FOR EACH NOTES GUID
for value in note_list:
try:
title, guid = value.split(':')
note = get_note(session.get('auth_token'),
session.get('shard_id'), guid)
except Exception as e:
error = "Error retrieving note using guid %s" % e
return render_template('main.html', error=error)
# ATTEMPT TO PARSE THE NOTE INTO A ENXML OBJECT
# GET DOCTYPE FROM NOTE CONTENT
rg = re.compile('(<!DOCTYPE .*?>)')
m = rg.search(note)
try:
# CREATE EN OBJECT INITILIZING WITH NOTE AND DOCTYPE
en = EN(note, doctype=m.group(1))
# BREAK OUT TABLES
en.break_out_tables()
try:
en.set_activity_type('O')
en.set_activity_type('I')
except:
error_list.append(
"Using an old Template, update your template in the future"
)
error_list.append(
"Unable to set Issues and Activities within the note")
flash(
update_note(session.get('auth_token'),
session.get('shard_id'),
en.tostring(),
guid=guid,
title=title))
# IF EMAIL IS SET
if user['use_email'] == 1:
lib.send_email(en, app.config['EMAIL_SERVER'],
user['email_address'])
flash("Note sent as email to %s" % user['email_address'])
# IF LEAD IS SET
if request.form['lead'] != 'None':
# GET LEAD AUTH_TOKEN
lead = controls.get_user_info(request.form['lead'])
flash(
new_note(shard_id=lead['shard_id'],
content=en.tostring(),
title=title,
auth_token=lead['auth_token']))
flash("Sent note to lead: %s" % request.form['lead'])
controls.set_lead(session.get('username'),
request.form['lead'])
# CHANGE THE USERS LEAD ID TO THAT USED
else:
controls.delete_lead(session.get('username'))
lead = {'username': '', 'first_name': '', 'last_name': ''}
flash("Bug Time: %s" % en.times['B'])
flash("Setup Time: %s" % en.times['S'])
flash("Test Time: %s" % en.times['T'])
flash("Session Length: %s" % en.session_length)
en.export_xml(app.config['UPLOADED_FILE_DEST'],
"%s.xml" % title)
except Exception as e:
error_list.append(
"ERROR parsing out data from note %s. Exception: [%s]" %
(title, e))
return render_template('main.html',
notebooks=notebooks,
error=error_list,
authorized_users=authorized_users,
lead=lead)
def index():
error_list = None
# CHECK IF THE USER IS LOGGED IN, OTHERWISE REDIRECT
if not session.get('logged_in'):
return redirect(url_for('login'))
# GET THE USER INFO, REDIRECT IF NOTHING IS RETURNED
user = controls.get_user_info(session.get('username'))
if not user:
redirect(url_for('logout'))
authorized_users = controls.get_authorized_users()
# RETURNS A JSON ObJECT OF USERNAME, FIRST AND LAST
lead = controls.get_lead(session.get('username'))
if len(lead) == 0:
lead = {'username':'', 'first_name':'', 'last_name':''}
else:
lead = lead[0]
notebooks = []
if session.get('auth_token'):
# CHECK AND SEE IF USER HAS SELECTED NOTEBOOKS
if user['notebook_ids'] is not None and len(user['notebook_ids']) > 0:
guid_list = user['notebook_ids'].split(',')
flash("Getting Notebooks and Notes")
notebooks = get_notebooks(session.get('auth_token'),
session.get('shard_id'),
guid_list)
else:
return redirect(url_for('configure', guid_list=None))
# POST REQUESTS
if request.method=='POST':
error_list = []
note_list = request.form.getlist('note')
# GET THE NOTE CONTENT FOR EACH NOTES GUID
for value in note_list:
try:
title, guid = value.split(':')
note = get_note(session.get('auth_token'),
session.get('shard_id'),
guid)
except Exception as e:
error = "Error retrieving note using guid %s" %e
return render_template('main.html',
error=error)
# ATTEMPT TO PARSE THE NOTE INTO A ENXML OBJECT
# GET DOCTYPE FROM NOTE CONTENT
rg = re.compile('(<!DOCTYPE .*?>)')
m = rg.search(note)
try:
# CREATE EN OBJECT INITILIZING WITH NOTE AND DOCTYPE
en = EN(note, doctype=m.group(1))
# BREAK OUT TABLES
en.break_out_tables()
try:
en.set_activity_type('O')
en.set_activity_type('I')
except:
error_list.append("Using an old Template, update your template in the future")
error_list.append("Unable to set Issues and Activities within the note")
flash(update_note(session.get('auth_token'),
session.get('shard_id'),
en.tostring(),
guid=guid,
title=title))
# IF EMAIL IS SET
if user['use_email'] == 1:
lib.send_email(en, app.config['EMAIL_SERVER'], user['email_address'])
flash("Note sent as email to %s" %user['email_address'])
# IF LEAD IS SET
if request.form['lead'] != 'None':
# GET LEAD AUTH_TOKEN
lead = controls.get_user_info(request.form['lead'])
flash(new_note(shard_id=lead['shard_id'],
content=en.tostring(),
title=title,
auth_token=lead['auth_token']))
flash("Sent note to lead: %s" %request.form['lead'])
controls.set_lead(session.get('username'), request.form['lead'])
# CHANGE THE USERS LEAD ID TO THAT USED
else:
controls.delete_lead(session.get('username'))
lead = {'username':'', 'first_name':'', 'last_name':''}
flash("Bug Time: %s" %en.times['B'])
flash("Setup Time: %s" %en.times['S'])
flash("Test Time: %s" %en.times['T'])
flash("Session Length: %s" %en.session_length)
en.export_xml(app.config['UPLOADED_FILE_DEST'], "%s.xml" %title)
except Exception as e:
error_list.append("ERROR parsing out data from note %s. Exception: [%s]" %(title, e))
return render_template('main.html',
notebooks=notebooks,
error=error_list,
authorized_users=authorized_users,
lead=lead)