Mosqkiller-API/apps/counter/mixins/role.py

17 lines
359 B
Python

from counter.models import DeviceInfo
class RoleMixin(object):
def get_role(self):
user = self.request.user
user_roles = user.role.split(',')
return user_roles
class DeviceListMixin(object):
def get_device_list(self):
device_list = DeviceInfo.objects.get_queryset().order_by('-online')
return device_list