From a6dce04477468a30e193b7463f896dc5ccbea6e5 Mon Sep 17 00:00:00 2001
From: fxxian <fxxian@syncore.space>
Date: Thu, 25 Apr 2024 10:33:20 +0800
Subject: [PATCH] =?UTF-8?q?chore:=20=E6=8E=A5=E5=8F=A3=E6=B7=BB=E5=8A=A0ic?=
 =?UTF-8?q?on/image/location=5Fgroup=20=E5=AD=97=E6=AE=B5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/counter/api/mobile/serializers.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

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