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)
