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)