diff --git a/apps/counter/api/mobile/serializers.py b/apps/counter/api/mobile/serializers.py index 9f3973e..77a951c 100644 --- a/apps/counter/api/mobile/serializers.py +++ b/apps/counter/api/mobile/serializers.py @@ -71,6 +71,9 @@ class DeviceInfoMobileSerializer(serializers.ModelSerializer): energy = serializers.SerializerMethodField() coordinate = serializers.SerializerMethodField() location = serializers.SerializerMethodField() + location_group = serializers.SerializerMethodField() + image_url = serializers.SerializerMethodField() + icon_url = serializers.SerializerMethodField() led_status = serializers.SerializerMethodField() led_health = serializers.SerializerMethodField() @@ -85,10 +88,13 @@ class DeviceInfoMobileSerializer(serializers.ModelSerializer): 'count', 'signal', 'energy', + 'image_url', + 'icon_url', 'led_status', 'led_health', 'coordinate', 'location', + 'location_group', ] def __init__(self, *args, **kwargs): @@ -137,6 +143,11 @@ class DeviceInfoMobileSerializer(serializers.ModelSerializer): return self.mosq_device_info.location return None + def get_location_group(self, obj): + if self.mosq_device_info: + return self.mosq_device_info.location_group + return None + def get_count(self, obj): self.latest = get_device_latest_by_cache(obj.device_id) @@ -171,6 +182,16 @@ class DeviceInfoMobileSerializer(serializers.ModelSerializer): return get_vol(obj.device_id, self.latest['vol']) return '0%' + def get_image_url(self, obj): + if self.mosq_device_info: + return self.mosq_device_info.image_url + return None + + def get_icon_url(self, obj): + if self.mosq_device_info: + return self.mosq_device_info.icon_url + return None + def get_led_status(self, obj): if self.mosq_device_info: return self.mosq_device_info.led_status