def reward(id):
if request.method == 'GET':
rewards = Reward.objects(itemid=id)
return get_items(request, rewards)
if request.method == 'POST':
item = Reward.objects(itemid=id).first()
print 'Fetched Item'
if request_has_json():
item.description = request.json['description']
item.save()
return 'Updated'
else:
form = RewardForm(request.form)
form.populate_obj(item)
print 'Populated Item from Form'
print item.description
item.save()
return render_template('list_items.html',items=[item])
def stages(jobid):
if request.method == 'GET':
job = Job.objects(jobid=int(jobid)).first()
stages = Stage.objects(job=job)
itemLst = [dict(stage.to_mongo()) for stage in stages]
return mongodoc_jsonify(items=itemLst)
if request.method == 'POST':
if request_has_json():
try:
job = Job.objects(jobid=int(jobid)).first()
reward = Reward.objects(itemid=request.json['reward']).first()
model = Stage()
model.job = job
model.reward = reward
model.rewardDesc = reward.itemid
model.jobDesc = str(job.jobid)
model.stage = request.json['stage']
model.save()
model.reload()
return mongodoc_jsonify(item=model.to_mongo())
except ValidationError as e:
return jsonify(item=str(e))
def list_rewards(): rewards = Reward.objects() return get_items(request, rewards)
def edit_reward_form(id):
item = Reward.objects(itemid=id).first()
form = RewardForm(request.form, obj=item)
return render_template('add_reward.html',form=form, edit=True)
