import enum


class CampaignWorkflowStatus(enum.Enum):
    DRAFT = 'Draft'
    PENDING_APPROVAL = 'Pending Approval'
    DISAPPROVED = 'Disapproved'
    BOOKED = 'Booked'
    COMPLETED = 'Completed'
    PAUSED = 'Paused'
    APPROVED = 'Approved'
    LIVE = 'Live'


def workflow_status_sorting(value):
    position = {
        CampaignWorkflowStatus.DRAFT.value: 1,
        CampaignWorkflowStatus.PENDING_APPROVAL.value: 2,
        CampaignWorkflowStatus.DISAPPROVED.value: 3,
        CampaignWorkflowStatus.APPROVED.value: 4,
        CampaignWorkflowStatus.BOOKED.value: 5,
        CampaignWorkflowStatus.LIVE.value: 6,
        CampaignWorkflowStatus.PAUSED.value: 7,
        CampaignWorkflowStatus.COMPLETED.value: 8,
    }
    return position.get(value)