diff --git a/homeassistant/components/rainmachine/__init__.py b/homeassistant/components/rainmachine/__init__.py index 5778d9db4df..7017f80a50f 100644 --- a/homeassistant/components/rainmachine/__init__.py +++ b/homeassistant/components/rainmachine/__init__.py @@ -271,6 +271,21 @@ class RainMachineEntity(Entity): self._name = None self.rainmachine = rainmachine + @property + def device_info(self): + """Return device registry information for this entity.""" + return { + 'identifiers': { + (DOMAIN, self.rainmachine.client.mac) + }, + 'name': self.rainmachine.client.name, + 'manufacturer': 'RainMachine', + 'model': 'Version {0} (API: {1})'.format( + self.rainmachine.client.hardware_version, + self.rainmachine.client.api_version), + 'sw_version': self.rainmachine.client.software_version, + } + @property def device_state_attributes(self) -> dict: """Return the state attributes."""