Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove wallet

pull/1599/head
Fabian Affolter 2016-03-24 12:37:40 +01:00
parent 5085e337e5
commit 369d234bda
1 changed files with 6 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
homeassistant/components/sensor/bitcoin.py
View File

@ -1,5 +1,5 @@
""" """
Bitcoin information service that uses blockchain.info and its online wallet. Bitcoin information service that uses blockchain.info.
For more details about this platform, please refer to the documentation at For more details about this platform, please refer to the documentation at
https://home-assistant.io/components/sensor.bitcoin/ https://home-assistant.io/components/sensor.bitcoin/
@ -13,7 +13,6 @@ from homeassistant.util import Throttle
REQUIREMENTS = ['blockchain==1.3.1'] REQUIREMENTS = ['blockchain==1.3.1']
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
OPTION_TYPES = { OPTION_TYPES = {
'wallet': ['Wallet balance', 'BTC'],
'exchangerate': ['Exchange rate (1 BTC)', None], 'exchangerate': ['Exchange rate (1 BTC)', None],
'trade_volume_btc': ['Trade volume', 'BTC'], 'trade_volume_btc': ['Trade volume', 'BTC'],
'miners_revenue_usd': ['Miners revenue', 'USD'], 'miners_revenue_usd': ['Miners revenue', 'USD'],
@ -43,33 +42,17 @@ MIN_TIME_BETWEEN_UPDATES = timedelta(seconds=120)
def setup_platform(hass, config, add_devices, discovery_info=None): def setup_platform(hass, config, add_devices, discovery_info=None):
"""Setup the Bitcoin sensor.""" """Setup the Bitcoin sensors."""
from blockchain.wallet import Wallet from blockchain import exchangerates
from blockchain import exchangerates, exceptions
wallet_id = config.get('wallet', None)
password = config.get('password', None)
currency = config.get('currency', 'USD') currency = config.get('currency', 'USD')
if currency not in exchangerates.get_ticker(): if currency not in exchangerates.get_ticker():
_LOGGER.error('Currency "%s" is not available. Using "USD".', currency) _LOGGER.error('Currency "%s" is not available. Using "USD"', currency)
currency = 'USD' currency = 'USD'
if wallet_id is not None and password is not None:
wallet = Wallet(wallet_id, password)
try:
wallet.get_balance()
except exceptions.APIException as error:
_LOGGER.error(error)
wallet = None
else:
wallet = None
data = BitcoinData() data = BitcoinData()
dev = [] dev = []
if wallet is not None and password is not None:
dev.append(BitcoinSensor(data, 'wallet', currency, wallet))
for variable in config['display_options']: for variable in config['display_options']:
if variable not in OPTION_TYPES: if variable not in OPTION_TYPES:
_LOGGER.error('Option type: "%s" does not exist', variable) _LOGGER.error('Option type: "%s" does not exist', variable)
@ -83,13 +66,12 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
class BitcoinSensor(Entity): class BitcoinSensor(Entity):
"""Representation of a Bitcoin sensor.""" """Representation of a Bitcoin sensor."""
def __init__(self, data, option_type, currency, wallet=''): def __init__(self, data, option_type, currency):
"""Initialize the sensor.""" """Initialize the sensor."""
self.data = data self.data = data
self._name = OPTION_TYPES[option_type][0] self._name = OPTION_TYPES[option_type][0]
self._unit_of_measurement = OPTION_TYPES[option_type][1] self._unit_of_measurement = OPTION_TYPES[option_type][1]
self._currency = currency self._currency = currency
self._wallet = wallet
self.type = option_type self.type = option_type
self._state = None self._state = None
self.update() self.update()
@ -122,10 +104,7 @@ class BitcoinSensor(Entity):
ticker = self.data.ticker ticker = self.data.ticker
# pylint: disable=no-member # pylint: disable=no-member
if self.type == 'wallet' and self._wallet is not None: if self.type == 'exchangerate':
self._state = '{0:.8f}'.format(self._wallet.get_balance() *
0.00000001)
elif self.type == 'exchangerate':
self._state = ticker[self._currency].p15min self._state = ticker[self._currency].p15min
self._unit_of_measurement = self._currency self._unit_of_measurement = self._currency
elif self.type == 'trade_volume_btc': elif self.type == 'trade_volume_btc':