def finish_plan_api():
pk = flask.request.form.get('pk')
user_id = flask.session['user_id']
plan = Plan.fetchone(pk=pk)
if not plan:
raise error.PlanNoteFound()
plan_log = PlanLog()
plan_log.user_id = user_id
plan_log.plan_id = plan.pk
plan_log.create_at = datetime.now()
plan_log.update_at = datetime.now()
plan_log.is_enable = 1
plan_log.save(commit=False)
for item in plan.plan_items:
note = Note()
note.kind_id = item.kind.pk
note.user_id = user_id
note.quantity = item.quantity
note.plan_log_id = plan_log.pk
note.create_at = datetime.now()
note.update_at = datetime.now()
note.save()
return {'pk': pk}
def set_plan_item(pk):
user_id = flask.session['user_id']
plan = Plan.fetchone(pk=pk, user_id=user_id, is_enable=1)
if not plan:
raise error.PlanNoteFound()
kinds = NoteKind.fetchall(
user_id=user_id, is_enable=1, order=NoteKind.create_at.desc())
return flask.render_template('plan_set_item.html', plan=plan, kinds=kinds)
def set_plan_item_quantity(plan_id, kind_id):
user_id = flask.session['user_id']
plan = Plan.fetchone(pk=plan_id, user_id=user_id, is_enable=1)
if not plan:
raise error.PlanNoteFound()
kind = NoteKind.fetchone(pk=kind_id, user_id=user_id, is_enable=1)
if not kind:
raise error.KindNotExistError()
return flask.render_template('plan_set_quantity.html', plan=plan, kind=kind)
def remove_plan_item_api():
pk = flask.request.form.get('pk')
user_id = flask.session['user_id']
plan = Plan.fetchone(pk=pk, user_id=user_id)
if not plan:
raise error.PlanNoteFound()
item = PlanItem.fetchone(pk=pk)
if not item:
raise error.PlanItemNoteFound()
item.is_enable = 0
item.save()
return {'pk': item.pk}
def create_plan_item_api():
plan_id = flask.request.form.get('plan_id')
kind_id = flask.request.form.get('kind_id')
quantity = flask.request.form.get('quantity')
user_id = flask.session['user_id']
plan = Plan.fetchone(pk=plan_id)
if not plan:
raise error.PlanNoteFound()
item = PlanItem()
item.kind_id = kind_id
item.user_id = user_id
item.plan_id = plan_id
item.quantity = quantity
item.create_at = datetime.now()
item.update_at = datetime.now()
item.is_enable = 1
item.save()
return {'pk': item.pk}
def plan_item(pk):
user_id = flask.session['user_id']
plan = Plan.fetchone(pk=pk, user_id=user_id, is_enable=1)
return flask.render_template('plan_item.html', plan=plan)
