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)