def add_duty():
if 'user_id' not in session:
return redirect(url_for('login'))
else:
myname = session['username']
if request.method == "POST":
title = request.form['title']
print title
category_id = request.form['name']
is_show = request.form['is_show']
status = request.form['status']
if title and category_id:
data = Duty(category_id, session['user_id'], title, status,
is_show, time.time())
res = db.session.add(data)
db.session.commit()
print data.duty_id
if data.duty_id:
flash('add successfully! ')
return redirect(url_for('my_duty'))
else:
flash('register error!')
return redirect(url_for('add_duty'))
else:
flash('field can not be empty')
return redirect(url_for('add_duty'))
else:
category_list = Category.query.order_by(Category.category_id).all()
return render_template('add_duty.html',
category_list=category_list)
def ajax_go_on_duty(request):
assert request.GET, "Error: no get data"
ret = {}
#---------
duty = Duty.latest_open(user=request.display_user) or \
Duty(user=request.display_user)
duty.start = atd(request.GET['timestamp'])
duty.save()
#if not duty.latest_open_block():
# block = Block(duty=duty)
# block.save()
# ret['block_id'] = block.id
ret['duty'] = duty.as_json_dict()
return ret
