From 43a7247ddeb9545d03cabd1ce6b2f4fc46c59663 Mon Sep 17 00:00:00 2001
From: dougiteixeira <31328123+dougiteixeira@users.noreply.github.com>
Date: Wed, 29 Mar 2023 18:04:37 -0300
Subject: [PATCH] Move ProxmoxEntity to entity.py (#90480)

* Move ProxmoxEntity to entity.py

* Update homeassistant/components/proxmoxve/entity.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* Update homeassistant/components/proxmoxve/entity.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* Update homeassistant/components/proxmoxve/entity.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* Update homeassistant/components/proxmoxve/entity.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* Update homeassistant/components/proxmoxve/binary_sensor.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

---------

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>
---
 .../components/proxmoxve/__init__.py          | 53 +------------------
 .../components/proxmoxve/binary_sensor.py     |  3 +-
 homeassistant/components/proxmoxve/entity.py  | 39 ++++++++++++++
 3 files changed, 42 insertions(+), 53 deletions(-)
 create mode 100644 homeassistant/components/proxmoxve/entity.py

diff --git a/homeassistant/components/proxmoxve/__init__.py b/homeassistant/components/proxmoxve/__init__.py
index 2764f22b080..b61f0ca4df5 100644
--- a/homeassistant/components/proxmoxve/__init__.py
+++ b/homeassistant/components/proxmoxve/__init__.py
@@ -22,10 +22,7 @@ from homeassistant.core import HomeAssistant
 import homeassistant.helpers.config_validation as cv
 from homeassistant.helpers.discovery import async_load_platform
 from homeassistant.helpers.typing import ConfigType
-from homeassistant.helpers.update_coordinator import (
-    CoordinatorEntity,
-    DataUpdateCoordinator,
-)
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
 
 from .const import (
     _LOGGER,
@@ -252,54 +249,6 @@ def call_api_container_vm(
     return status
 
 
-class ProxmoxEntity(CoordinatorEntity):
-    """Represents any entity created for the Proxmox VE platform."""
-
-    def __init__(
-        self,
-        coordinator: DataUpdateCoordinator,
-        unique_id: str,
-        name: str,
-        icon: str,
-        host_name: str,
-        node_name: str,
-        vm_id: int | None = None,
-    ) -> None:
-        """Initialize the Proxmox entity."""
-        super().__init__(coordinator)
-
-        self.coordinator = coordinator
-        self._unique_id = unique_id
-        self._name = name
-        self._host_name = host_name
-        self._icon = icon
-        self._available = True
-        self._node_name = node_name
-        self._vm_id = vm_id
-
-        self._state = None
-
-    @property
-    def unique_id(self) -> str:
-        """Return the unique ID for this sensor."""
-        return self._unique_id
-
-    @property
-    def name(self) -> str:
-        """Return the name of the entity."""
-        return self._name
-
-    @property
-    def icon(self) -> str:
-        """Return the mdi icon of the entity."""
-        return self._icon
-
-    @property
-    def available(self) -> bool:
-        """Return True if entity is available."""
-        return self.coordinator.last_update_success and self._available
-
-
 class ProxmoxClient:
     """A wrapper for the proxmoxer ProxmoxAPI client."""
 
diff --git a/homeassistant/components/proxmoxve/binary_sensor.py b/homeassistant/components/proxmoxve/binary_sensor.py
index 828c8191148..ea02e547e98 100644
--- a/homeassistant/components/proxmoxve/binary_sensor.py
+++ b/homeassistant/components/proxmoxve/binary_sensor.py
@@ -10,7 +10,8 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
 
-from . import COORDINATORS, DOMAIN, PROXMOX_CLIENTS, ProxmoxEntity
+from .const import COORDINATORS, DOMAIN, PROXMOX_CLIENTS
+from .entity import ProxmoxEntity
 
 
 async def async_setup_platform(
diff --git a/homeassistant/components/proxmoxve/entity.py b/homeassistant/components/proxmoxve/entity.py
new file mode 100644
index 00000000000..5dfd264df2d
--- /dev/null
+++ b/homeassistant/components/proxmoxve/entity.py
@@ -0,0 +1,39 @@
+"""Proxmox parent entity class."""
+
+from homeassistant.helpers.update_coordinator import (
+    CoordinatorEntity,
+    DataUpdateCoordinator,
+)
+
+
+class ProxmoxEntity(CoordinatorEntity):
+    """Represents any entity created for the Proxmox VE platform."""
+
+    def __init__(
+        self,
+        coordinator: DataUpdateCoordinator,
+        unique_id: str,
+        name: str,
+        icon: str,
+        host_name: str,
+        node_name: str,
+        vm_id: int | None = None,
+    ) -> None:
+        """Initialize the Proxmox entity."""
+        super().__init__(coordinator)
+
+        self.coordinator = coordinator
+        self._attr_unique_id = unique_id
+        self._attr_name = name
+        self._host_name = host_name
+        self._attr_icon = icon
+        self._available = True
+        self._node_name = node_name
+        self._vm_id = vm_id
+
+        self._state = None
+
+    @property
+    def available(self) -> bool:
+        """Return True if entity is available."""
+        return self.coordinator.last_update_success and self._available