def wizard(): if current_user.organisation_name: organisation = Organisations.query.filter_by(id=current_user.organisation_id).first() form = OrganisationsForm(obj=organisation) form.name.data = current_user.organisation_name else: form = OrganisationsForm() if form.validate_on_submit(): if current_user.organisation_name: form.populate_obj(organisation) else: organisation = Organisations(form.name.data) organisation.domain = form.domain.data user = User.query.get_or_404(current_user.id) organisation.users = [user] db.session.add(organisation) db.session.commit() flash(_('Organisation added')) return redirect(url_for("home.homepage")) return render_template('wizard.html', form=form)