Пример #1
0
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')