django xadmin action相容自定義model權限

如标題。最近在研究xadmin，發現文檔确實比較少，自己隻能連滾帶爬～

起因

因為想做一個審批的功能，用xadmin acrtion來實作。本來想用子產品現有“change”字段控制權限，但是發現使用“change”字段控制權限的話，會把自帶的 “delete” action 暴露出來，這不是我想要的。 是以在子產品裡添加了一個“approver”字段，用以控制權限。

代碼

首先修改子產品的 “Meta”，增權重限字段。

class TaskApprove(Task):
    class Meta:
        proxy = True
        verbose_name = u'審批'
        verbose_name_plural = verbose_name
        permissions = (
            ("approve","審批：使用者任務"),
        )
           

然後重新整理一下資料庫，會在使用者管理的界面中看到我們新加的權限。

接下來在adminx.py中建立自己的action,并在你的管理類中添加。

from xadmin.plugins.actions import BaseActionView
class MyAction(BaseActionView):
    action_name = u'accept'
    model_perm = 'approve'
    description =  '同意'
    def do_action(self, queryset):
        for obj in queryset:
            ##你的操作
            print(obj)
        return HttpResponse('您已同意')
class TaskApproveSettings(ViewOnlyMixin):
    actions = [MyAction,]
    list_display = ['name','targets','services','create_by']
xadmin.site.register(TaskApprove,TaskApproveSettings)
           

下面是重點， 因為xadmin中的權限檢測是寫死的 隻有 “view\change\delete\add”，而我們添加了一個“approve字段”， 這會導緻報錯 – “keyerro” 。 是以接下來我們覆寫一下xadmin中的方法，添加一個方法。

* 不知道為什麼前幾次以繼承覆寫的方式沒有成功~~*

在控制類TaskApproveSettings中添加。

首先添加一個方法,用來擷取是否有權限

def has_approve_permission(self):
    codename = get_permission_codename('approve', self.opts)
    return ('approve' not in self.remove_permissions) and self.user.has_perm('%s.%s' % (self.app_label, codename))
           

接下來修改get_model_perms 方法，将我們的權限添加進去

def get_model_perms(self):
        """
        Returns a dict of all perms for this model. This dict has the keys
        ``add``, ``change``, and ``delete`` mapping to the True/False for each
        of those actions.
        """
        return {
            'view': self.has_view_permission(),
            'add': self.has_add_permission(),
            'change': self.has_change_permission(),
            'delete': self.has_delete_permission(),
            'approve': self.has_approve_permission()
        }
           

最後是這樣的

from xadmin.plugins.actions import BaseActionView
from django.shortcuts import HttpResponse
from django.contrib.auth import get_permission_codename
class ActionApproveAccept(BaseActionView):
    action_name = u'accept'
    model_perm = 'approve'
    description =  '同意'
    global_actions = []
    def do_action(self, queryset):
        for obj in queryset:
            print(obj)
        return HttpResponse('您已同意')

class ActionApproveReject(BaseActionView):
    action_name = u'reject'
    model_perm = 'approve'
    description =  '拒絕'
    global_actions = []
    def do_action(self, queryset):
        for obj in queryset:
            print(obj)
        return HttpResponse('您已拒絕')
class TaskApproveSettings(ViewOnlyMixin):
    actions = [ActionApproveAccept,ActionApproveReject]

    list_display = ['name','targets','services','create_by']
    readonly_fields = TaskFields.create+TaskFields.params+TaskFields.approve
    form_layout = (
        Main(
            Fieldset('',
                     *TaskFields.create,
                     css_class='unsort no_title'
                     ),

            Fieldset(('參數'),
                     *TaskFields.params
                     ),
        ),
        Side(
            Fieldset(('審批狀态'),
                     *TaskFields.approve
                     ),
        )
    )
    def get_model_perms(self):
        return {
            'view': self.has_view_permission(),
            'add': self.has_add_permission(),
            'change': self.has_change_permission(),
            'delete': self.has_delete_permission(),
            'approve': self.has_approve_permission()
        }
    def has_approve_permission(self):
        codename = get_permission_codename('approve', self.opts)
        return ('approve' not in self.remove_permissions) and self.user.has_perm('%s.%s' % (self.app_label, codename))
           

重新開機服務就好啦！

我僅做了目前這些，暫時沒有發現問題。如果後面發現問題我會繼續補充，另外大家如果有更好的方法，歡迎留言糾正～！