Move Playstation Plus Entity to Sensor Platform

JackJPowell · Dec 1, 2024 · 5b00d9e · 5b00d9e
1 parent 1eef373
commit 5b00d9e
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 3 deletions.
diff --git a/custom_components/playstation_network/__init__.py b/custom_components/playstation_network/__init__.py
@@ -21,7 +21,7 @@
 PLATFORMS: list[Platform] = [
     Platform.MEDIA_PLAYER,
     Platform.SENSOR,
-    Platform.BINARY_SENSOR,
+    # Platform.BINARY_SENSOR,
     Platform.IMAGE,
 ]
 

diff --git a/custom_components/playstation_network/image.py b/custom_components/playstation_network/image.py
@@ -30,7 +30,7 @@ def __init__(self, coordinator) -> None:
 
         ImageEntity.__init__(self, coordinator.hass)
         self._attr_has_entity_name = True
-        self._attr_name = "Playstation Avatar"
+        self._attr_name = "Avatar"
         self._attr_icon = "mdi:face-man-profile"
         self._attr_image_url = self.get_avatar()
         self._attr_image_last_updated = dt_util.utcnow()

diff --git a/custom_components/playstation_network/manifest.json b/custom_components/playstation_network/manifest.json
@@ -6,7 +6,7 @@
   "dependencies": ["network"],
   "documentation": "https://github.com/JackJPowell/hass-psn",
   "homekit": {},
-  "integration_type": "device",
+  "integration_type": "service",
   "iot_class": "cloud_polling",
   "issue_tracker": "https://github.com/JackJPowell/hass-psn/issues",
   "requirements": ["PSNAWP-HA==2.2.2"],

diff --git a/custom_components/playstation_network/sensor.py b/custom_components/playstation_network/sensor.py
@@ -29,6 +29,12 @@ class PsnSensorEntityDescription(SensorEntityDescription):
     description: str = ""
 
 
+def get_ps_plus_status(coordinator_data: any) -> str:
+    if coordinator_data.get("profile", {}).get("isPlus"):
+        return "Active"
+    return "Inactive"
+
+
 def get_status(coordinator_data: any) -> str:
     """Returns online status"""
     match coordinator_data.get("platform").get("onlineStatus"):
@@ -168,6 +174,17 @@ def get_trophy_attr(coordinator_data: any) -> dict[str, str]:
         value_fn=get_status,
         attributes_fn=get_status_attr,
     ),
+    PsnSensorEntityDescription(
+        key="has_playstation_plus",
+        device_class=SensorDeviceClass.ENUM,
+        name="Playstation Plus",
+        icon="mdi:gamepad-outline",
+        options=["Active", "Inactive"],
+        entity_registry_enabled_default=True,
+        has_entity_name=True,
+        unique_id="has_playstation_plus",
+        value_fn=get_ps_plus_status,
+    ),
 )
 
 PSN_ADDITIONAL_SENSOR: tuple[PsnSensorEntityDescription, ...] = (