def get_context_data(self, **kwargs):
if not adminPermission(self.request.user):
qe = Equipment.objects.filter(user_id=self.request.user.id)
else:
qe = Equipment.objects.all()
data = []
for j in xrange(3):
data.append(qe.filter(work_flag=j).count())
context = super(Index, self).get_context_data(**kwargs)
context['data'] = data
return context
def get_initial_queryset(self):
if not self.model:
raise NotImplementedError("Need to provide a model or implement get_initial_queryset!")
# 管理員权限
if adminPermission(self.request.user):
queryset = self.model.objects.all()
else:
if self.name != 'equip':
queryset = self.model.objects.filter(equip__user=self.request.user.id)
else:
queryset = self.model.objects.filter(user_id=self.request.user.id)
if self.model_id != None:
queryset = queryset.filter(equip_id=self.model_id)
return queryset
def filter_queryset(self, queryset):
if not adminPermission(self.request.user):
queryset = queryset.filter(equip__user_id=self.request.user.id)
return super(EquipStateLastViewSet, self).filter_queryset(queryset)
def filter_queryset(self, queryset):
if not adminPermission(self.request.user):
queryset = queryset.filter(id=self.request.user.id)
return super(AppUserViewSet, self).filter_queryset(queryset)
