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'))