def schedule_delete(schedule_id):
schedule = AdSchedule.get(schedule_id)
if not schedule:
abort(404)
item = schedule.item
schedule.delete()
flash(u'删除成功!', 'success')
return redirect(url_for("schedule.item_detail", item_id=item.id))
def schedule_update(schedule_id):
schedule = AdSchedule.get(schedule_id)
if not schedule:
abort(404)
schedule.date = datetime.strptime(request.form.get('date'), DATE_FORMAT).date()
schedule.start = datetime.strptime(request.form.get('start'), TIME_FORMAT).time()
schedule.end = datetime.strptime(request.form.get('end'), TIME_FORMAT).time()
schedule.num = request.form.get('num')
schedule.save()
flash(u'投放安排保存成功!', 'success')
return redirect(url_for("schedule.item_detail", item_id=schedule.item.id))
def test_schedule(session):
schedule = add_schedule()
schedule2 = AdSchedule.get(schedule.id)
assert schedule2 is not None