diff --git a/apps/assets/api/mixin.py b/apps/assets/api/mixin.py index 6dadbcdaa..d3ea4761d 100644 --- a/apps/assets/api/mixin.py +++ b/apps/assets/api/mixin.py @@ -63,6 +63,8 @@ class SerializeToTreeNodeMixin: return AllTypes.get_types_values(exclude_custom=True) def get_icon(self, asset): + if asset.category == 'device': + return 'switch' if asset.type in self.support_types: return asset.type else: