1. 程式人生 > 程式設計 >Django --Xadmin 判斷登入者身份例項

Django --Xadmin 判斷登入者身份例項

一,為了讓xadmin 登入者只能看到自己建立的資料

1,model

class UserTB(models.Model):
  name=models.CharField('姓名',max_length=30,blank=True,null=True)
  email=models.CharField('郵箱',max_length=200,null=True)
  integral = models.IntegerField('積分',default=0)
  create_time=models.DateTimeField('建立時間',auto_now_add=True)
  user = models.OneToOneField(User,null=True) #
  def __unicode__(self):
    return self.name
  class Meta:
    verbose_name='員工管理'
    verbose_name_plural='員工管理'
class UnitPage(models.Model):
  title=models.Charfield('標題',max_length=50)
  uploader=models.ForeignKey(UserTB,verbose_name='建立者',related_name='deuser',null=True,on_delete=models.SET_NULL)
  create_time=models.DateTimeField('建立時間',auto_now_add=True)
  def __unicode__(self):
    return self.title
  class Meta:
    verbose_name='文章管理'
    verbose_name_plural='文章管理'

2,adminx

class ProdeUnitDisplay(object):
  list_display=['title','create_time'] #需要顯示欄位
  search_fields=['title',] #可搜尋的欄位
  def queryset(self):
    if not self.request.user.is_superuser: #判斷是否是超級使用者
      st = UserTB.objects.get(user=self.request.user) #找到使用者表裡對應的使用者
      print 'st:',st
      sr = self.model.objects.filter(uploader=st) #找到該model 裡該使用者建立的資料
      return sr
 
    return self.model.objects.all()
xadmin.site.register(UnitPage,ProdeUnitDisplay)

補充知識:Django2整合xadmin詳解-5-獲取登入使用者資訊並填充相應Model欄位

一 場景問題

繼續使用IDC Model,在此Model增加user欄位,用於記錄添加當前資料的使用者:

class IDC(models.Model):
  user = models.ForeignKey(User,on_delete=models.CASCADE,editable=False,null=True) # 記錄建立該資料的使用者
  name = models.CharField(max_length=64)
  contact = models.CharField(max_length=32)
  phone = models.CharField(max_length=32)
  address = models.CharField(max_length=128)
  create_time = models.DateField(auto_now=True)

  def __str__(self):
    return self.name

  class Meta:
    verbose_name = "IDC機房"
    verbose_name_plural = verbose_name

問題:如何在增加IDC資料的時候,獲取當前登入的使用者資訊,並儲存到user欄位裡?

二 解決方法

開啟IdcManager目錄的adminx.py,在IDCAdmin類裡增加save_models方法,程式碼如下:

@xadmin.sites.register(IDC)
class IDCAdmin(object):
  list_display = ("user","name","contact","phone","address","create_time")
  list_display_links = ("name",)

  def save_models(self):
    self.new_obj.user = self.request.user
    super().save_models()

三 總結

xadmin的文件和資料比較少,這個問題雖然也就三行程式碼的事情,但是我在百度是沒找到解決方案的。

最終是檢視xadmin的程式碼摸索解決的。

那麼,應該檢視哪部分程式碼?考慮到這個操作是在儲存資料的時候觸發的,那麼就找xadmin處理儲存操作(編輯介面)的程式碼:

venv\Lib\site-packages\xadmin\views\edit.py

以上這篇Django --Xadmin 判斷登入者身份例項就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。