django xadmin action如何實現兼容自定義model權限

小編給大家分享一下django xadmin action如何實現兼容自定義model權限，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

起因

因為想做一個審批的功能，用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))

重啟服務就好啦！

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

補充知識：Dajngo 通過代碼添加xadmin用戶和權限（組）

在開發的時候，用戶要求在認證的時候自動添加xadmin登錄賬戶和分配組權限

from django.contrib.auth.models import Group,User
from django.http import JsonResponse
def test(req):
  name=req.POST['name']
  account=req.POST['account']
  password=req.POST['password']
  an=Group.objects.filter(id=1).first() #二級管理組 是管理員在xadmin后臺添加的權限組
  user = User(username=account)
  user.set_password(password)
  user.is_superuser = False
  user.is_active = True
  user.first_name = name
  user.is_staff = True
  user.save() #先生成用戶
  user.groups.add(an)
  return JsonResponse({'ret':0,'msg':'success'})

以上是“django xadmin action如何實現兼容自定義model權限”這篇文章的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！