feat: mobile device 接口新增 led 状态与健康信息

apps/counter/api/mobile/serializers.py

@@ -1,9 +1,25 @@
+import pytz
+from enum import Enum
 from typing import Optional
+from django.utils.timezone import datetime
 from rest_framework import serializers
 from counter.models import DeviceCount, DeviceInfo
 from mosquito.models import DeviceInfo as MosquitoDeviceInfo
 
 
+class LedHealthStatus(Enum):
+    Good = '良好'
+    General = '一般'
+    Bad = '较差'
+    Unknown = '未知'
+
+    def __repr__(self):
+        return self.value
+
+    def __str__(self):
+        return self.value
+
+
 class DeviceInfoMobileSerializer(serializers.ModelSerializer):
     device_name = serializers.SerializerMethodField()
     status = serializers.SerializerMethodField()
@@ -12,6 +28,8 @@ class DeviceInfoMobileSerializer(serializers.ModelSerializer):
     vol = serializers.SerializerMethodField()
     coordinate = serializers.SerializerMethodField()
     location = serializers.SerializerMethodField()
+    led_status = serializers.SerializerMethodField()
+    led_health = serializers.SerializerMethodField()
 
     class Meta:
         model = DeviceInfo
@@ -22,6 +40,8 @@ class DeviceInfoMobileSerializer(serializers.ModelSerializer):
             'count',
             'csq',
             'vol',
+            'led_status',
+            'led_health',
             'coordinate',
             'location',
         ]
@@ -29,14 +49,14 @@ class DeviceInfoMobileSerializer(serializers.ModelSerializer):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.latest = None
-        self.cur_device = None
+        self.mosq_device = None
         self.counter_device = None
 
     def get_device_name(self, obj) -> Optional[str]:
         qs = MosquitoDeviceInfo.objects.filter(device_id=obj.device_id)
         if qs.count() > 0:
             device = qs.first()
-            self.cur_device = device
+            self.mosq_device = device
             return device.device_name
         return None
 
@@ -44,20 +64,20 @@ class DeviceInfoMobileSerializer(serializers.ModelSerializer):
         return obj.online
 
     def get_coordinate(self, obj):
-        if self.cur_device:
-            lon, lat = [self.cur_device.longitude, self.cur_device.latitude]
+        if self.mosq_device:
+            lon, lat = [self.mosq_device.longitude, self.mosq_device.latitude]
             if lon and lat:
                 return [lon, lat]
         return None
 
     def get_location(self, obj):
-        if self.cur_device:
-            return self.cur_device.location
+        if self.mosq_device:
+            return self.mosq_device.location
         return None
 
     def get_count(self, obj):
         device = DeviceCount.objects.filter(
-            device_id=self.cur_device.device_id).order_by('-data_time').first()
+            device_id=self.mosq_device.device_id).order_by('-data_time').first()
         if device:
             self.counter_device = device
             return device.count
@@ -72,3 +92,20 @@ class DeviceInfoMobileSerializer(serializers.ModelSerializer):
         if self.counter_device:
             return self.counter_device.vol
         return None
+
+    def get_led_status(self, obj):
+        if self.mosq_device:
+            return self.mosq_device.led_status
+        return None
+
+    def get_led_health(self, obj):
+        if self.mosq_device:
+            diff = datetime.now(tz=pytz.timezone('UTC')) - self.mosq_device.launch_time
+            hours = diff.days * 24 + int(diff.seconds / 3600)
+            rate = hours / self.mosq_device.led_lifetime
+            if rate <= 0.3:
+                return LedHealthStatus.Good.value
+            if 0.3 < rate <= 0.6:
+                return LedHealthStatus.General.value
+            return LedHealthStatus.Bad.value
+        return LedHealthStatus.Unknown.value

apps/mosquito/migrations/0025_auto_20240306_1531.py

@@ -0,0 +1,18 @@
+# Generated by Django 2.2.4 on 2024-03-06 15:31
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('mosquito', '0024_auto_20240303_1255'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='deviceinfo',
+            name='led_lifetime',
+            field=models.PositiveIntegerField(default=30000, verbose_name='LED 寿命'),
+        ),
+    ]

apps/mosquito/models.py

@@ -101,7 +101,7 @@ class DeviceInfo(models.Model):
     latitude = models.FloatField(blank=True, null=True, verbose_name='维度')
     location = models.CharField(max_length=100, blank=True, null=True, verbose_name='地理位置')
     led_status = models.CharField(max_length=10, choices=LED_CHOICE, default='off', verbose_name='LED 状态')
-    led_lifetime = models.PositiveIntegerField(default=10000, verbose_name='LED 寿命')
+    led_lifetime = models.PositiveIntegerField(default=30000, verbose_name='LED 寿命')
     launch_time = models.DateTimeField(blank=True, null=True, verbose_name='第一次启动时间')
 
     # 太阳能