def login():
error = None
if request.method == 'POST':
user = controls.get_user_info(request.form['username'])
if len(user) <= 0:
error = 'Invalid username'
elif request.form['password'] != user['password']:
error = 'Invalid password'
else:
session['logged_in'] = True
session['username'] = request.form['username']
session['first_name'] = user['first_name']
session['last_name'] = user['last_name']
# CHECK IF THIS HAS BEEN SET YET
if user['shard_id'] is not None and len(user['shard_id']) > 0:
session['shard_id'] = user['shard_id']
if user['auth_token'] is not None and len(user['auth_token']) > 0:
session['auth_token'] = user['auth_token']
if user['is_admin'] is not None and user['is_admin'] > 0:
session['admin'] = True
flash('Successfully Logged In')
return redirect(url_for('index'))
else:
if session.get('logged_in'):
return redirect(url_for('index'))
return render_template('login.html', error=error)
def login():
error = None
if request.method == 'POST':
user = controls.get_user_info(request.form['username'])
if len(user) <= 0:
error = 'Invalid username'
elif request.form['password'] != user['password']:
error = 'Invalid password'
else:
session['logged_in'] = True
session['username'] = request.form['username']
session['first_name'] = user['first_name']
session['last_name'] = user['last_name']
# CHECK IF THIS HAS BEEN SET YET
if user['shard_id'] is not None and len(user['shard_id']) > 0:
session['shard_id'] = user['shard_id']
if user['auth_token'] is not None and len(user['auth_token']) > 0:
session['auth_token'] = user['auth_token']
if user['is_admin'] is not None and user['is_admin'] > 0:
session['admin'] = True
flash('Successfully Logged In')
return redirect(url_for('index'))
else:
if session.get('logged_in'):
return redirect(url_for('index'))
return render_template('login.html', error=error)
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 upload():
# THIS USES THE NEW_NOTE CALL, so going to move that to en_lib
if request.method=='POST':
upload_file = request.files['file']
if upload_file and lib.allowed_file(upload_file.filename, app.config['ALLOWED_EXTENSIONS']):
filename = upload_file.filename
content = upload_file.stream.getvalue()
# CHECK IF THIS NOTE WAS EXPORTED BY THE DESKTOP CLIENT AND IF SO STRIP OUT COMPONENTS
pattern = re.compile('(.*?!\[CDATA\[)(.*?<\/en-note>)(.*?\]\]><\/content>.*)', re.DOTALL)
m = pattern.search(content)
if m is not None and len(m.group(2)) > 0:
content = m.group(2)
flash(new_note(content=content, title=upload_file.filename))
with open(os.path.join(app.config['UPLOADED_FILE_DEST'], filename), 'w') as f:
f.write(content)
return redirect(url_for('index'))
else:
return render_template('upload.html')
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 upload():
# THIS USES THE NEW_NOTE CALL, so going to move that to en_lib
if request.method == 'POST':
upload_file = request.files['file']
if upload_file and lib.allowed_file(upload_file.filename,
app.config['ALLOWED_EXTENSIONS']):
filename = upload_file.filename
content = upload_file.stream.getvalue()
# CHECK IF THIS NOTE WAS EXPORTED BY THE DESKTOP CLIENT AND IF SO STRIP OUT COMPONENTS
pattern = re.compile(
'(.*?!\[CDATA\[)(.*?<\/en-note>)(.*?\]\]><\/content>.*)',
re.DOTALL)
m = pattern.search(content)
if m is not None and len(m.group(2)) > 0:
content = m.group(2)
flash(new_note(content=content, title=upload_file.filename))
with open(os.path.join(app.config['UPLOADED_FILE_DEST'], filename),
'w') as f:
f.write(content)
return redirect(url_for('index'))
else:
return render_template('upload.html')
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 auth_finish():
"""After the user has authorized this application on Evernote's website,
they will be redirected back to this URL to finish the process."""
oauth_verifier = request.args.get('oauth_verifier', '')
token = oauth.Token(session['oauth_token'],
session['oauth_token_secret'])
token.set_verifier(oauth_verifier)
# client = get_oauth_client()
client = get_oauth_client(token)
# Retrieve the token credentials (Access Token) from Evernote
resp, content = client.request(app.config['EN_ACCESS_TOKEN_URL'], 'POST')
if resp['status'] != '200':
raise Exception('Invalid response %s.' % resp['status'])
access_token = dict(urlparse.parse_qsl(content))
authToken = access_token['oauth_token']
# MAY NEED SHARD/AUTH
userStore = get_userstore()
user = userStore.getUser(authToken)
# Save the users information to so we can make requests later
session['shard_id'] = user.shardId
session['auth_token'] = authToken
# INSERT THE SESSION AND LINK TO THE USERID
controls.update_user(session.get('username'),
auth_token=session.get('auth_token'),
shard_id=session.get('shard_id'),
authorized=1)
flash("Evernote Test Session Authorized!!")
return redirect(url_for('index'))
def auth_finish():
"""After the user has authorized this application on Evernote's website,
they will be redirected back to this URL to finish the process."""
oauth_verifier = request.args.get('oauth_verifier', '')
token = oauth.Token(session['oauth_token'], session['oauth_token_secret'])
token.set_verifier(oauth_verifier)
# client = get_oauth_client()
client = get_oauth_client(token)
# Retrieve the token credentials (Access Token) from Evernote
resp, content = client.request(app.config['EN_ACCESS_TOKEN_URL'], 'POST')
if resp['status'] != '200':
raise Exception('Invalid response %s.' % resp['status'])
access_token = dict(urlparse.parse_qsl(content))
authToken = access_token['oauth_token']
# MAY NEED SHARD/AUTH
userStore = get_userstore()
user = userStore.getUser(authToken)
# Save the users information to so we can make requests later
session['shard_id'] = user.shardId
session['auth_token'] = authToken
# INSERT THE SESSION AND LINK TO THE USERID
controls.update_user(session.get('username'),
auth_token=session.get('auth_token'),
shard_id=session.get('shard_id'),
authorized=1)
flash("Evernote Test Session Authorized!!")
return redirect(url_for('index'))
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 logout():
for key in session.keys():
session.pop(key, None)
flash('You were logged out')
return redirect(url_for('index'))
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 logout():
for key in session.keys():
session.pop(key, None)
flash('You were logged out')
return redirect(url_for('index'))
