def cae_situation_option(dbsession): from autonomie.models.user import ( CaeSituationOption, ) option = CaeSituationOption( is_integration=False, label=u"CaeSituationOption", ) dbsession.add(option) dbsession.flush() return option
def userdata(dbsession): from autonomie.models.user import (UserDatas, CaeSituationOption,) u = UserDatas( situation_situation=CaeSituationOption.query().first(), coordonnees_firstname="firstname", coordonnees_lastname="lastname", coordonnees_email1="*****@*****.**", ) dbsession.add(u) dbsession.flush() return u
def get_userdatas(): option = CaeSituationOption(label="Integre", is_integration=True) return UserDatas( situation_situation=option, coordonnees_lastname="test", coordonnees_firstname="test", coordonnees_email1="*****@*****.**", activity_companydatas=[CompanyDatas( title='test entreprise', name='test entreprise', )] )
def populate_situation_options(session): """ Populate the CAE situation options """ from autonomie.models.user import CaeSituationOption query = session.query(CaeSituationOption) if query.filter(CaeSituationOption.is_integration == True).count() == 0: session.add( CaeSituationOption( label=u"Réunion d'information", order=0, )) session.add( CaeSituationOption( label=u"Intégré", is_integration=True, order=1, )) session.add(CaeSituationOption( label=u"Sortie", order=2, )) session.flush()
def upgrade(): from alembic.context import get_bind op.add_column( "user_datas", sa.Column( "situation_situation_id", sa.Integer, sa.ForeignKey("cae_situation_option.id"), )) op.add_column("configurable_option", sa.Column("order", sa.Integer, default=0)) from autonomie.models.user import ( CaeSituationOption, ) from autonomie.models.base import DBSESSION temp_dict = {} for key, value in SITUATION_OPTIONS: if key == "integre": option = CaeSituationOption(label=value, is_integration=True) else: option = CaeSituationOption(label=value) DBSESSION().add(option) DBSESSION().flush() temp_dict[key] = option.id conn = get_bind() query = "select id, situation_situation from user_datas" result = conn.execute(query) for id, situation in result: option_id = temp_dict.get(situation) if option_id is None: continue query = "update user_datas set situation_situation_id='{0}' \ where id='{1}'".format(option_id, id) op.execute(query)