def register():
errors = ''
form = RegistrationForm(request.form)
if request.method == 'POST':
if form.validate():
session = Session()
Users = Tables()['Users']
new_user = Users(
username=form.username.data,
password=bcrypt.generate_password_hash(form.password.data),
email=(form.email.data).encode('utf-8'),
firstname=form.firstname.data,
lastname=form.lastname.data,
)
session.add(new_user)
session.commit()
session.close()
return redirect(url_for('login'))
else:
log.info('form errors:')
log.info(form.errors)
return render_template(
'account_management/register.html',
form=form,
errors=form.errors,
)
return render_template(
'account_management/register.html',
form=form,
errors=errors,
)
def delete_analysis():
"""Delete the selected analysis from the database."""
user = get_current_user()
session = Session()
Analysis = Tables()['Analysis']
success = False
aSelected = request.args.get('aSelected')
if len(aSelected) == 0:
return jsonify(success=success)
result = (
session.query(Analysis)
.filter(Analysis.id == aSelected)
.first()
)
if result is None:
return jsonify(success=success)
if (result.public
or (result.username == user.username)
or (user.labgroup in result.labgroup)):
success = True
session.delete(result)
session.commit()
else:
log.info("You do not have permission to delete this analysis.")
return jsonify(success=success)
session.close()
return jsonify(success=success)
def set_saved_selections():
user = get_current_user()
if not user.username:
return jsonify(
response="user not logged in, can't save selections",
null=True,
)
session = Session()
Users = Tables()['Users']
saved_selections = request.args.get('stringed_selections')
user_entry = (
session.query(Users)
.filter(Users.username == user.username)
.first()
)
user_entry.selections = saved_selections
session.commit()
session.close()
return jsonify(response='selections saved', null=False)
def edit_analysis():
"""Take input from Analysis Editor modal and save it to the database.
Button : Edit Analysis
"""
user = get_current_user()
session = Session()
Analysis = Tables()['Analysis']
modTime = datetime.datetime.now().replace(microsecond=0)
eName = request.args.get('name')
eId = request.args.get('id')
eStatus = request.args.get('status')
eTags = request.args.get('tags')
eQuestion = request.args.get('question')
eAnswer = request.args.get('answer')
eLoad = request.args.get('load')
eMod = request.args.get('mod')
eFit = request.args.get('fit')
eTree = json.dumps([eLoad, eMod, eFit])
if eId == '__none':
checkExists = False
else:
checkExists = (
session.query(Analysis)
.filter(Analysis.id == eId)
.first()
)
if checkExists:
a = checkExists
if (
a.public
or (user.labgroup in a.labgroup)
or (a.username == user.username)
):
a.name = eName
a.status = eStatus
a.question = eQuestion
a.answer = eAnswer
a.tags = eTags
try:
a.lastmod = modTime
except:
a.lastmod = str(modTime)
a.modeltree = eTree
else:
log.info("You do not have permission to modify this analysis.")
return jsonify(
success=("failed")
)
# If it doesn't exist, add new sql alchemy object with the
# appropriate attributes, which should get assigned to a new id
else:
# TODO: Currently copies user's labgroup by default.
# Is that the behavior we want?
try:
a = Analysis(
name=eName, status=eStatus, question=eQuestion,
answer=eAnswer, tags=eTags, batch='',
lastmod=modTime, modeltree=eTree, username=user.username,
labgroup=user.labgroup, public='0'
)
except:
a = Analysis(
name=eName, status=eStatus, question=eQuestion,
answer=eAnswer, tags=eTags, batch='',
lastmod=str(modTime), modeltree=eTree,
username=user.username, labgroup=user.labgroup, public='0'
)
session.add(a)
addedName = a.name
session.commit()
session.close()
# Clear out cached analysis selection, otherwise the simple caching logic
# thinks the analysis selection didn't change so it will load previous
# modeltree
global _previous_update_models
_previous_update_models = ('', '', '', [])
# After handling submissions, return user to main page so that it
# refreshes with new analysis included in list
return jsonify(success="Analysis %s saved successfully." % addedName)
def update_cells():
"""Update the list of cells in the cell selector after a batch
is selected (this will cascade from an analysis selection).
Also updates current batch in Analysis for current analysis.
"""
session = Session()
db_tables = Tables()
Batches = db_tables['Batches']
sBatch = db_tables['sBatch']
Analysis = db_tables['Analysis']
# Only get the numerals for the selected batch, not the description.
bSelected = request.args.get('bSelected')
aSelected = request.args.get('aSelected')
search = request.args.get('cellSearch')
global _previous_update_cells
previous_batch, previous_celllist = _previous_update_cells
if bSelected == previous_batch:
celllist = previous_celllist
else:
celllist = [
i[0] for i in
session.query(Batches.cellid)
.filter(Batches.batch == bSelected[:3])
.all()
]
_previous_update_cells = (bSelected, celllist)
batchname = (
session.query(sBatch)
.filter(sBatch.id == bSelected[:3])
.first()
)
if batchname:
batch = str(bSelected[:3] + ': ' + batchname.name)
else:
batch = bSelected
analysis = (
session.query(Analysis)
.filter(Analysis.name == aSelected)
.first()
)
# don't change batch association if batch is blank
if analysis and bSelected:
analysis.batch = batch
session.commit()
session.close()
# remove pairs
filtered_cellids = [c for c in celllist if '+' not in c]
# remove siteids
filtered_cellids = [c for c in filtered_cellids if '-' in c]
# filter by search string
filtered_cellids = simple_search(search, filtered_cellids)
return jsonify(celllist=filtered_cellids)