def init_activity(): for each_record in activity_init: print(Activity.find_by_name(each_record['name'])) if Activity.find_by_name(each_record['name']) == None: category_id_for_new_record = (Category.find_by_category( each_record['category_name']).id) new_record = Activity(name=each_record['name'], category_id=category_id_for_new_record, username_id=current_user.id) Activity.save_to_db(new_record) return render_template('habit.html')
def add(): form_category = AddCategory() form_activity = AddActivity() all_cat = Category.json_all() if form_category.validate_on_submit( ) and form_category.submit_new_category.data: a_category = Category(category=form_category.category_name.data, username_id=current_user.id) Category.save_to_db(a_category) # print(f'well done, we changed {form_category.name} into {form_category.category_name.data}') return render_template('add.html', form_activity=form_activity, form_category=form_category) if form_activity.validate_on_submit( ) and form_activity.submit_new_activity.data: try: a_activity = Activity(name=form_activity.activity_name.data, category_id=form_activity.category_id.data, username_id=current_user.id) # if form_activity.box.data: # a_activity.category_id = form_activity.activity_category_id.data Activity.save_to_db(a_activity) flash(f'well done, we added {form_activity.activity_name.data}') return render_template('add.html', form_activity=form_activity, form_category=form_category, added_activity=True) except: return render_template('add.html', form_activity=form_activity, form_category=form_category, added_activity=True) return render_template('add.html', form_activity=form_activity, form_category=form_category)