evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Move attribution to standalone attribute [e-g] (#80513)

Browse source
This commit is contained in:
epenet 2022-10-18 10:04:54 +02:00 • committed by GitHub
parent cb530e398c
commit c717fd19de
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
9 changed files with 25 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

12
homeassistant/components/gtfs/sensor.py
View file

@ -16,7 +16,7 @@ from homeassistant.components.sensor import (
SensorDeviceClass,
SensorEntity,
)
from homeassistant.const import ATTR_ATTRIBUTION, CONF_NAME, CONF_OFFSET, STATE_UNKNOWN
from homeassistant.const import CONF_NAME, CONF_OFFSET, STATE_UNKNOWN
from homeassistant.core import HomeAssistant
import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.entity_platform import AddEntitiesCallback
@ -648,6 +648,11 @@ class GTFSDepartureSensor(SensorEntity):
# Assign attributes, icon and name
self.update_attributes()
if self._agency:
self._attr_attribution = self._agency.agency_name
else:
self._attr_attribution = None
if self._route:
self._icon = ICONS.get(self._route.route_type, ICON)
else:
@ -702,11 +707,6 @@ class GTFSDepartureSensor(SensorEntity):
elif ATTR_INFO in self._attributes:
del self._attributes[ATTR_INFO]
if self._agency:
self._attributes[ATTR_ATTRIBUTION] = self._agency.agency_name
elif ATTR_ATTRIBUTION in self._attributes:
del self._attributes[ATTR_ATTRIBUTION]
# Add extra metadata
key = "agency_id"
if self._agency and key not in self._attributes: