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