示例#1
0
文件: web.py 项目: rdcarter24/MyYogi
def display_routine():
    routine_id = request.args.get("routine_id")
    asana_img = []
    asana_time = []
    sub_routine_list =[]

    ##### if user wants saved routine
    ##### get_routine returns a list of objects of the Routine_Asana class
    ##### BUG!!!!  redirect home after running saved routine
    if routine_id:
        routine = MyYogi.get_routine(routine_id)

        for obj in routine:
            asana_img.append(obj.asana.image)
            asana_json = json.dumps(asana_img)
            asana_time.append(obj.asana.breaths)
            sub_routine_list.append(obj.sub_routine)
            sub_routine_json=json.dumps(sub_routine_list)

        return render_template("display_routine.html", asana_list=asana_json, asana_time=asana_time, sub_routine_list=sub_routine_json, saved=True)
    ##### if user wants new routine
    ##### generate_routine returns a list of objects of the Asana class
    else:
        routine = MyYogi.get_yoga_routine(training_data, session["user_id"])

        for i in range(len(routine)):
            for obj in routine[i][0]:
                asana_img.append(obj[0].image)
                asana_json = json.dumps(asana_img)
                asana_time.append(obj[1])
                sub_routine_list.append(routine[i][1])
                sub_routine_json=json.dumps(sub_routine_list)

        return render_template("display_routine.html", asana_list=asana_json, asana_time=asana_time, sub_routine_list=sub_routine_json, saved=False)