from wtforms_alchemy import ModelForm as ModelForm_, model_form_factory

BaseModelForm = model_form_factory(ModelForm_)


class ModelForm(BaseModelForm):
    @classmethod
    def get_session(self): # noqa
        return DBSession