jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-25 14:17:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add ERC20 tokens to etherscan.io sensor (#11916)

Browse Source 
* upgrade python-etherscan-api

* add support for erc20 tokens
This commit is contained in:
Nolan Gilley 2018-01-26 00:08:08 -05:00 committed by Paulus Schoutsen
parent 920f9f132b
commit 9a71717047
2 changed files with 28 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
homeassistant/components/sensor/etherscan.py
View File

@ -13,18 +13,20 @@ from homeassistant.const import ATTR_ATTRIBUTION, CONF_NAME
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.entity import Entity from homeassistant.helpers.entity import Entity
REQUIREMENTS = ['python-etherscan-api==0.0.2'] REQUIREMENTS = ['python-etherscan-api==0.0.3']
CONF_ADDRESS = 'address' CONF_ADDRESS = 'address'
CONF_TOKEN = 'token'
CONF_TOKEN_ADDRESS = 'token_address'
CONF_ATTRIBUTION = "Data provided by etherscan.io" CONF_ATTRIBUTION = "Data provided by etherscan.io"
DEFAULT_NAME = 'Ethereum Balance'
SCAN_INTERVAL = timedelta(minutes=5) SCAN_INTERVAL = timedelta(minutes=5)
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Required(CONF_ADDRESS): cv.string, vol.Required(CONF_ADDRESS): cv.string,
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string, vol.Optional(CONF_NAME): cv.string,
vol.Optional(CONF_TOKEN): cv.string,
vol.Optional(CONF_TOKEN_ADDRESS): cv.string,
}) })
@ -32,19 +34,30 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
"""Set up the Etherscan.io sensors.""" """Set up the Etherscan.io sensors."""
address = config.get(CONF_ADDRESS) address = config.get(CONF_ADDRESS)
name = config.get(CONF_NAME) name = config.get(CONF_NAME)
token = config.get(CONF_TOKEN)
token_address = config.get(CONF_TOKEN_ADDRESS)
add_devices([EtherscanSensor(name, address)], True) if token:
token = token.upper()
if not name:
name = "%s Balance" % token
if not name:
name = "ETH Balance"
add_devices([EtherscanSensor(name, address, token, token_address)], True)
class EtherscanSensor(Entity): class EtherscanSensor(Entity):
"""Representation of an Etherscan.io sensor.""" """Representation of an Etherscan.io sensor."""
def __init__(self, name, address): def __init__(self, name, address, token, token_address):
"""Initialize the sensor.""" """Initialize the sensor."""
self._name = name self._name = name
self.address = address self._address = address
self._token_address = token_address
self._token = token
self._state = None self._state = None
self._unit_of_measurement = 'ETH' self._unit_of_measurement = self._token or "ETH"
@property @property
def name(self): def name(self):
@ -71,4 +84,9 @@ class EtherscanSensor(Entity):
def update(self): def update(self):
"""Get the latest state of the sensor.""" """Get the latest state of the sensor."""
from pyetherscan import get_balance from pyetherscan import get_balance
self._state = get_balance(self.address) if self._token_address:
self._state = get_balance(self._address, self._token_address)
elif self._token:
self._state = get_balance(self._address, self._token)
else:
self._state = get_balance(self._address)

2
requirements_all.txt
View File

@ -879,7 +879,7 @@ python-ecobee-api==0.0.15
# python-eq3bt==0.1.8 # python-eq3bt==0.1.8
# homeassistant.components.sensor.etherscan # homeassistant.components.sensor.etherscan
python-etherscan-api==0.0.2 python-etherscan-api==0.0.3
# homeassistant.components.sensor.darksky # homeassistant.components.sensor.darksky
# homeassistant.components.weather.darksky # homeassistant.components.weather.darksky