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