1

我正在制作一个请假流程，申请人将申请请假，请假将排队，以便批准者批准或拒绝请假申请。我遇到了django-fsm和drf-fsm-transitions，我想到了实现这种工作流程。我正在使用django 1.11++python3来django_rest_framework构建我的 API。

在我已导入的 django 应用程序from django_fsm import transition, FSMIntegerField 中models.py。

离开类

class Leave(models.Model):
    LEAVE_STATUS_CREATED = 0
    LEAVE_STATUS_APPLIED = 1
    LEAVE_STATUS_APPROVED = 2
    LEAVE_STATUS_REJECTED = 3
    LEAVE_STATUS_PENDING = 4
    LEAVE_STATUS_COMPLETE = 5

    LEAVE_STATUS_CHOICES = (
        (LEAVE_STATUS_CREATED, 'created'),
        (LEAVE_STATUS_APPLIED, 'applied'),
        (LEAVE_STATUS_APPROVED, 'approved'),
        (LEAVE_STATUS_REJECTED, 'rejected'),
        (LEAVE_STATUS_PENDING, 'pending'),
        (LEAVE_STATUS_COMPLETE, 'complete'),


    )

    leave_id = models.AutoField(primary_key=True)
    applicant = models.ForeignKey(
        Employee, related_name='applicant', on_delete=models.CASCADE, null=True)
    approver = models.ForeignKey(
        Employee, related_name='approver', on_delete=models.CASCADE, null=True)
    applied_on = models.DateTimeField(auto_now_add=True)
    responded_on = models.DateTimeField(auto_now=True, null=True)
    leave_type = models.ForeignKey(LeaveType, null=True)
    approved = models.BooleanField(default=False)
    rejected = models.BooleanField(default=False)
    start_date = models.DateField()
    return_date = models.DateField()
    leave_status = FSMIntegerField(
        choices=LEAVE_STATUS_CHOICES, default=LEAVE_STATUS_CREATED, protected=True)
    comment = models.TextField(max_length=200)
    leave_subject = models.CharField(max_length=40)
    leave_reason = models.TextField(max_length=200)
    total_days = models.IntegerField(null=True)

过渡

然后我在休假类中实现了以下转换：

  @transition(field=leave_status, source=LEAVE_STATUS_CREATED, target=LEAVE_STATUS_APPLIED)
    def apply(self, comment):
        self.applicant = applicant
        self.leave_type = leave_type
        self.start_date = start_date
        self.return_date = return_date
        self.leave_subject = leave_subject
        self.leave_reason = leave_reason
        print("Apply  %s Leave").format(self.comment)


    @transition(field=leave_status, source=LEAVE_STATUS_APPLIED, target=LEAVE_STATUS_APPROVED)
    def approve(self, approver, responded_on, comment):
        self.approver = approver
        self.responded_on = responded_on
        self.comment = comment
        print("approve the leave")

    @transition(field=leave_status, source=[LEAVE_STATUS_CREATED, LEAVE_STATUS_APPLIED], target=LEAVE_STATUS_REJECTED)
    def reject(self, approver, responded_on, comment):
        self.approver = approver
        self.responded_on = responded_on
        self.comment = comment
        print("rejected leave ")

然后在我导入的视图中from drf_fsm_transitions.viewset_mixins import get_viewset_transition_action_mixin

我在运行服务器时收到以下日志：

transitions = instance.get_all_status_transitions()
AttributeError: 'Leave' object has no attribute 'get_all_status_transitions'

这是我的视图类的样子：

class LeaveViewSet(
        get_viewset_transition_action_mixin(Leave),
        viewsets.ModelViewSet):
    queryset = Leave.objects.all()

在 django 中使用 django-fsm 时如何实现基于类的视图

4

0 回答 0