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 create_note_api(): kind_id = flask.request.form.get('kind_id') quantity = flask.request.form.get('quantity') content = flask.request.form.get('content') user_id = int(flask.session['user_id']) # Check kind owner note_kind = NoteKind.fetchone(pk=kind_id, user_id=user_id) if not note_kind: raise error.KindNotExistError() note = Note() note.kind_id = kind_id note.user_id = user_id note.content = content note.quantity = quantity note.save() return {'pk': note.pk}