def profile():
company = Company.query.get_or_404(current_user.company.id)
form = CompanyForm(obj=company)
if form.validate_on_submit():
form.update_company(company)
return redirect(url_for('company.profile'))
return render_template('company/profile.html', form=form)
def edit_company(company_id):
company = Company.query.get_or_404(company_id)
form = CompanyForm(obj=company)
if form.validate_on_submit():
form.update_company(company)
flash('公司更新成功', 'success')
return redirect(url_for('admin.company'))
return render_template('admin/edit_company.html',
form=form,
company=company)