def populate_states():
workflow = get_workflow()
WFSteps = workflow.WFSteps
WFStates = workflow.WFStates
states = (
(WFStates.INITIAL_STATE, WFSteps.NO_STEP),
(WFStates.DRAFT_STATE, WFSteps.NO_STEP),
(WFStates.FI_NORMALIZE_STATE, WFSteps.SUBMITTED_STEP),
(WFStates.AUTHOR_NORMALIZE_STATE, WFSteps.SUBMITTED_STEP),
(WFStates.DI_APPRAISAL_STATE, WFSteps.STUDY_STEP),
(WFStates.RETURNED_BY_DI_STATE, WFSteps.STUDY_STEP),
(WFStates.DI_APPROVED_STATE, WFSteps.SUGGESTED_STEP),
(WFStates.SELECTED_STATE, WFSteps.SELECTED_STEP),
(WFStates.PROJECT_STATE, WFSteps.PROJECT_STEP),
(WFStates.PROTOTYPE_STATE, WFSteps.PROTOTYPE_STEP),
(WFStates.EXTENDED_STATE, WFSteps.EXTENDED_STEP),
(WFStates.FI_REFUSED_STATE, WFSteps.SUBMITTED_STEP),
(WFStates.PROTOTYPE_REFUSED_STATE, WFSteps.PROTOTYPE_STEP),
(WFStates.DI_REFUSED_STATE, WFSteps.STUDY_STEP),
(WFStates.PROJECT_REFUSED_STATE, WFSteps.PROJECT_STEP),
(WFStates.SELECT_REFUSED_STATE, WFSteps.SELECTED_STEP),
(WFStates.APPROVAL_REFUSED_STATE, WFSteps.SUGGESTED_STEP),
(WFStates.DSIG_BASKET_STATE, WFSteps.NO_STEP),
)
for label, step in states:
step = StepData.get_by(label=step)
StateData(label=label, step=step)
session.flush()
