diff --git a/apps/counter/api/mobile/serializers.py b/apps/counter/api/mobile/serializers.py index cf073e1..062251c 100644 --- a/apps/counter/api/mobile/serializers.py +++ b/apps/counter/api/mobile/serializers.py @@ -8,7 +8,7 @@ class DeviceInfoMobileSerializer(serializers.ModelSerializer): device_name = serializers.SerializerMethodField() status = serializers.SerializerMethodField() coordinate = serializers.SerializerMethodField() - # location = serializers.SerializerMethodField() + location = serializers.SerializerMethodField() class Meta: model = DeviceInfo @@ -17,7 +17,7 @@ class DeviceInfoMobileSerializer(serializers.ModelSerializer): 'device_name', 'status', 'coordinate', - # 'location', + 'location', ] def __init__(self, *args, **kwargs): @@ -44,4 +44,6 @@ class DeviceInfoMobileSerializer(serializers.ModelSerializer): return None def get_location(self, obj): - return obj.location + if self.cur_device: + return self.cur_device.location + return None