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

Use state class enum for Coinbase (#60764)

Browse source
This commit is contained in:
Tom Brien 2021-12-01 16:26:53 +00:00 • committed by GitHub
parent c985bee1dd
commit 1268cefc38
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
homeassistant/components/coinbase/sensor.py
View file

@ -1,11 +1,7 @@
"""Support for Coinbase sensors.""" """Support for Coinbase sensors."""
import logging import logging
from homeassistant.components.sensor import ( from homeassistant.components.sensor import SensorEntity, SensorStateClass
STATE_CLASS_MEASUREMENT,
STATE_CLASS_TOTAL,
SensorEntity,
)
from homeassistant.const import ATTR_ATTRIBUTION from homeassistant.const import ATTR_ATTRIBUTION
from homeassistant.helpers.device_registry import DeviceEntryType from homeassistant.helpers.device_registry import DeviceEntryType
from homeassistant.helpers.entity import DeviceInfo from homeassistant.helpers.entity import DeviceInfo
@ -110,7 +106,7 @@ class AccountSensor(SensorEntity):
API_ACCOUNT_CURRENCY API_ACCOUNT_CURRENCY
] ]
break break
self._attr_state_class = STATE_CLASS_TOTAL self._attr_state_class = SensorStateClass.TOTAL
self._attr_device_info = DeviceInfo( self._attr_device_info = DeviceInfo(
configuration_url="https://www.coinbase.com/settings/api", configuration_url="https://www.coinbase.com/settings/api",
entry_type=DeviceEntryType.SERVICE, entry_type=DeviceEntryType.SERVICE,
@ -183,7 +179,7 @@ class ExchangeRateSensor(SensorEntity):
1 / float(self._coinbase_data.exchange_rates[API_RATES][self.currency]), 2 1 / float(self._coinbase_data.exchange_rates[API_RATES][self.currency]), 2
) )
self._unit_of_measurement = exchange_base self._unit_of_measurement = exchange_base
self._attr_state_class = STATE_CLASS_MEASUREMENT self._attr_state_class = SensorStateClass.MEASUREMENT
self._attr_device_info = DeviceInfo( self._attr_device_info = DeviceInfo(
configuration_url="https://www.coinbase.com/settings/api", configuration_url="https://www.coinbase.com/settings/api",
entry_type=DeviceEntryType.SERVICE, entry_type=DeviceEntryType.SERVICE,