From 9d14ff810522c893bcf9533ff30f48a8cd3019c2 Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Mon, 8 Mar 2021 07:39:57 -1000 Subject: [PATCH] Add suggested_area support to Apple TV (#47015) --- homeassistant/components/apple_tv/__init__.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/apple_tv/__init__.py b/homeassistant/components/apple_tv/__init__.py index c735ebc57e7..75c4d6ccc8a 100644 --- a/homeassistant/components/apple_tv/__init__.py +++ b/homeassistant/components/apple_tv/__init__.py @@ -348,10 +348,16 @@ class AppleTVManager: "name": self.config_entry.data[CONF_NAME], } + area = attrs["name"] + name_trailer = f" {DEFAULT_NAME}" + if area.endswith(name_trailer): + area = area[: -len(name_trailer)] + attrs["suggested_area"] = area + if self.atv: dev_info = self.atv.device_info - attrs["model"] = "Apple TV " + dev_info.model.name.replace("Gen", "") + attrs["model"] = DEFAULT_NAME + " " + dev_info.model.name.replace("Gen", "") attrs["sw_version"] = dev_info.version if dev_info.mac: