from staff.lib.startrek.issues import add_comment

from staff.budget_position.workflow_service.entities.abstract_workflow import AbstractWorkflow


class MaternityWorkflow(AbstractWorkflow):
    code = '9'
    description = 'Move to maternity'

    def mark_pushed_to_oebs(self) -> None:
        super().mark_pushed_to_oebs()
        budget_position_code = self.changes[0].new_budget_position.code
        ticket_key = self.changes[0].ticket
        # По хорошему надо выделить отдельный headcounts service и положить эту строчку в его реализацию
        # в районе maternity view в потолчнице, и отсюда его дергать
        # но ради одной строки я не буду пилить еще один интерфейс и разруливать сборку зависимостей еще выше
        # такой вот компромисс
        add_comment(ticket_key, f'**Создана новая БП**: {budget_position_code}')
