def insert_one_proj_to_db(): name = 'AAA' owner = 1 start_day = datetime(2018, 12, 1) end_day = datetime(2018, 12, 2) one_day_plan_list = [Project.OneDayPlan() for _ in range(2)] spots = Spot.query[0:3] one_day_plan_list[0].add_spot(spots[0], 60) one_day_plan_list[0].add_spot(spots[1], 90) one_day_plan_list[1].add_spot(spots[2], 180) params = [name, owner, start_day, end_day, one_day_plan_list] proj = Project(*params) db.session.add(proj) db.session.commit()
def create_own_proj(): user_id = _get_user_from_cookie(request.cookies.get(COOKIE_KEY)) if not user_id: return _get_response('fail', message='user_id is missing') try: content = request.get_json() name = content['name'] start_day = strftime_to_datetime(content['start_day']) tot_days = content['tot_days'] except: return _get_response('fail', message='input is not correct') one_day_plan_list = [Project.OneDayPlan() for _ in range(tot_days)] params = [name, user_id, start_day, tot_days, one_day_plan_list] proj = Project(*params) db.session.add(proj) db.session.commit() return _get_response('success')