def resume(self, request, pk=None):
pk = ObjectId(pk)
res = Device.update_one({"_id": pk}, {"$set": {"status": "active"}})
if res.matched_count:
return JsonResponse(status=status.HTTP_201_CREATED)
else:
return NotFound()
return JsonResponse()
def retrieve(self, request, *args, **kwargs):
device = Device.find_one({"_id": ObjectId(kwargs["pk"])})
if device:
serialized = DeviceSerializer(device)
return JsonResponse(serialized.data)
else:
return NotFound()
def partial_update(self, request, *args, **kwargs):
pk = ObjectId(kwargs["pk"])
#TODO: valite data
res = Device.update_one({"_id": pk}, request.data)
if res.matched_count:
return JsonResponse(status=status.HTTP_201_CREATED)
else:
return NotFound()
def list(self, request):
filter_params = [
"product_name", "device_name", "status", "device_status"
]
condition = gen_condition(request.query_params, filter_params)
devices = Device.find(condition)
if devices:
serialized = DeviceSerializer(devices, many=True)
return JsonResponse(serialized.data)
else:
return NotFound()
def delete(self, request):
#TODO:
#踢出在线设备, 删除信息， 删除连接，删除acl
filter_params = ["product_name", "device_name"]
condition = gen_condition(request.query_params,
filter_params,
not_empty=True)
device = Device.find_one(condition)
if device:
d = Device(**device)
d.disconnect()
d.remove()
serialized = DeviceSerializer(device)
return JsonResponse(serialized.data)
else:
raise NotFound()
def create(self, request):
product_name = request.data["product_name"]
device_name = shortuuid.uuid()
password = shortuuid.uuid()
username = "******" % (product_name, device_name)
new_device = Device(product_name, device_name, username, password)
device_doc = new_device.to_doc()
Device.insert_one(device_doc)
acl = new_device.get_acl()
acl["username"] = new_device.username
device_acl = DeviceAcl(**acl)
DeviceAcl.insert_one(device_acl.to_doc())
serialized = DeviceSerializer(device_doc)
return JsonResponse(serialized.data, status=status.HTTP_201_CREATED)