def tj_tadd():
if not ('logged_in' in session and 'username' in session):
return redirect(url_for('index'))
tadd_form_basic = tj_tadd_basic_form(csrf_enabled=False)
tadd_form_speed = tj_tadd_speed_form(csrf_enabled=False)
return render_template('tadd.html', form_basic = tadd_form_basic, form_speed = tadd_form_speed)
def tj_t_edit(t_id):
if not ('logged_in' in session and 'username' in session):
return redirect(url_for('index'))
if isinstance(t_id, (long, int)):
model = TJTraining.query.filter_by(id=t_id).first()
if model:
t_attrs = model.props
t_attrs['title'] = model.title
t_attrs['traindate'] = model.traindate
t_attrs['tt'] = model.tt
t_attrs['desc'] = model.desc
td = type('TJTeditModelClass', (), t_attrs)
if model.t_tmpl == 1:
eform = tj_tadd_basic_form(obj = td)
etmpl = templ_1
elif model.t_tmpl == 2:
eform = tj_tadd_speed_form(obj = td)
etmpl = templ_2
print eform.data
print eform.validate_on_submit()
if eform.validate_on_submit():
t_attrs = eform.data
model.title = t_attrs['title']
del(t_attrs['title'])
model.traindate = t_attrs['traindate']
del(t_attrs['traindate'])
model.tt = t_attrs['tt']
del(t_attrs['tt'])
model.desc = t_attrs['desc']
del(t_attrs['desc'])
model.props = t_attrs
db.session.commit()
return redirect(url_for('tdiary'))
return render_template("tedit.html", form = eform, t_f_id = t_id, output_template = etmpl)
def tj_add_training(t_type=None):
if not ('logged_in' in session and 'username' in session):
return redirect(url_for('index'))
print "/tadd"
tadd_form_basic = tj_tadd_basic_form(csrf_enabled=False)
tadd_form_speed = tj_tadd_speed_form(csrf_enabled=False)
userid = 1
t_attrs = None
if tadd_form_basic.validate_on_submit() and t_type == 'basic':
t_attrs = tadd_form_basic.data
t_tmpl = 1
elif tadd_form_speed.validate_on_submit() and t_type == 'speed':
t_attrs = tadd_form_speed.data
t_tmpl = 2
if t_attrs:
title = t_attrs['title']
del(t_attrs['title'])
traindate = t_attrs['traindate']
del(t_attrs['traindate'])
tt = t_attrs['tt']
del(t_attrs['tt'])
desc = t_attrs['desc']
del(t_attrs['desc'])
db.session.add( TJTraining(title, userid, traindate, tt, desc, t_tmpl, t_attrs) )
db.session.commit()
return redirect(url_for('tj_tadd'))
