Browse Source

调整参数

yf_fyh 3 years ago
parent
commit
732d391b5b
2 changed files with 3 additions and 3 deletions
  1. 1 1
      utils/MyRateThrottle.py
  2. 2 2
      utils/permissions.py

+ 1 - 1
utils/MyRateThrottle.py

@@ -23,5 +23,5 @@ class DeviceDetailRateThrottle(SimpleRateThrottle):
     def get_cache_key(self, request, view):
         user = request.user
         uid = user["uid"]
-        deviceId = request.data.get("deviceId")
+        deviceId = request.data.get("device_id")
         return uid + deviceId

+ 2 - 2
utils/permissions.py

@@ -36,8 +36,8 @@ class DeviceDetailPermission(BasePermission):
             user = request.user
             uid = user.get("uid")
             user_modules = user.get("user_modules")
-            deviceId = request.data.get("deviceId")
-            device = Device.objects.filter(device_id=deviceId, owner=int(uid))
+            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: