bigdataAPI/utils/permissions.py

# coding:utf-8

from rest_framework.permissions import BasePermission
from apps.Equipment.models import Device


class RegisterViewPermission(BasePermission):
    def has_permission(self, request, view):
        try:
            user = request.user
            if user['username'] == "管理员":
                return True
            else:
                return False
        except:
            return False


class ModulePermission(BasePermission):
    def has_permission(self, request, view):
        try:
            user = request.user
            user_modules = user.get("user_modules")
            path = request.path
            if path.startswith("/equipment") and user_modules == "1":
                return True
            else:
                return False
        except:
            return False


class DeviceDetailPermission(BasePermission):
    def has_permission(self, request, view):
        try:
            user = request.user
            uid = user.get("uid")
            user_modules = user.get("user_modules")
            deviceId = request.data.get("device_id")
            device = Device.objects.filter(device_id=deviceId, owner_id=int(uid))
            if user_modules == "1" and device.exists():
                return True
            else:
                return False
        except:
            return False