mathieu/HA-EasyComputerManager
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 10 Wiki Activity

some refactor and change device identifier to mac
Some checks failed
Validate with hassfest / validate (push) Failing after 1s
Details
HACS Action / HACS Action (push) Failing after 3s
Details

Browse Source
This commit is contained in:
Mathieu Broillet 2024-01-01 16:27:22 +01:00
parent 95bd521d06
commit 197e453b43
Signed by: mathieu
GPG Key ID: C0E9E0E95AF03319
1 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
custom_components/easy_computer_manager/switch.py
View File

@ -44,10 +44,12 @@ async def async_setup_entry(
config: ConfigEntry, config: ConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddEntitiesCallback,
) -> None: ) -> None:
# Setup the computer switch # Retrieve the data from the config flow
mac_address: str = config.data.get(CONF_MAC) mac_address: str = config.data.get(CONF_MAC)
broadcast_address: str | None = config.data.get(CONF_BROADCAST_ADDRESS) broadcast_address: str | None = config.data.get(CONF_BROADCAST_ADDRESS)
broadcast_port: int | None = config.data.get(CONF_BROADCAST_PORT) broadcast_port: int | None = config.data.get(CONF_BROADCAST_PORT)
# broadcast_address: str | None = config.data.get(CONF_BROADCAST_ADDRESS)
# broadcast_port: int | None = config.data.get(CONF_BROADCAST_PORT)
host: str = config.data.get(CONF_HOST) host: str = config.data.get(CONF_HOST)
name: str = config.data.get(CONF_NAME) name: str = config.data.get(CONF_NAME)
dualboot: bool = config.data.get("dualboot") dualboot: bool = config.data.get("dualboot")
@ -95,7 +97,7 @@ async def async_setup_entry(
(SERVICE_DEBUG_INFO, {}, SupportsResponse.ONLY), (SERVICE_DEBUG_INFO, {}, SupportsResponse.ONLY),
] ]
# Register the services # Register the services that depends on the switch
for service in services: for service in services:
platform.async_register_entity_service( platform.async_register_entity_service(
service[0], service[0],
@ -114,8 +116,8 @@ class ComputerSwitch(SwitchEntity):
name: str, name: str,
host: str | None, host: str | None,
mac_address: str, mac_address: str,
broadcast_address: str | None, # broadcast_address: str | None,
broadcast_port: int | None, # broadcast_port: int | None,
dualboot: bool | False, dualboot: bool | False,
username: str, username: str,
password: str, password: str,
@ -127,8 +129,8 @@ class ComputerSwitch(SwitchEntity):
self._attr_name = name self._attr_name = name
self._host = host self._host = host
self._mac_address = mac_address self._mac_address = mac_address
self._broadcast_address = broadcast_address # self._broadcast_address = broadcast_address
self._broadcast_port = broadcast_port # self._broadcast_port = broadcast_port
self._dualboot = dualboot self._dualboot = dualboot
self._username = username self._username = username
self._password = password self._password = password
@ -147,7 +149,7 @@ class ComputerSwitch(SwitchEntity):
return None return None
return DeviceInfo( return DeviceInfo(
identifiers={(DOMAIN, self._host)}, identifiers={(DOMAIN, self._mac_address)},
name=self._attr_name, name=self._attr_name,
manufacturer="Generic", manufacturer="Generic",
model="Computer", model="Computer",
@ -169,16 +171,18 @@ class ComputerSwitch(SwitchEntity):
def turn_on(self, **kwargs: Any) -> None: def turn_on(self, **kwargs: Any) -> None:
"""Turn the computer on using wake on lan.""" """Turn the computer on using wake on lan."""
service_kwargs: dict[str, Any] = {} service_kwargs: dict[str, Any] = {}
if self._broadcast_address is not None: # if self._broadcast_address is not None:
service_kwargs["ip_address"] = self._broadcast_address # service_kwargs["ip_address"] = self._broadcast_address
if self._broadcast_port is not None: # if self._broadcast_port is not None:
service_kwargs["port"] = self._broadcast_port # service_kwargs["port"] = self._broadcast_port
_LOGGER.debug( _LOGGER.debug(
"Send magic packet to mac %s (broadcast: %s, port: %s)", "Send magic packet to mac %s (broadcast: %s, port: %s)",
self._mac_address, self._mac_address,
self._broadcast_address, self._broadcast_address,
self._broadcast_port, self._broadcast_port,
# self._broadcast_address,
# self._broadcast_port,
) )
wakeonlan.send_magic_packet(self._mac_address, **service_kwargs) wakeonlan.send_magic_packet(self._mac_address, **service_kwargs)