Iota wallet

.coveragerc

@@ -354,6 +354,7 @@ omit =
     homeassistant/components/feedreader.py
     homeassistant/components/foursquare.py
     homeassistant/components/ifttt.py
+    homeassistant/components/iota.py
     homeassistant/components/image_processing/dlib_face_detect.py
     homeassistant/components/image_processing/dlib_face_identify.py
     homeassistant/components/image_processing/seven_segments.py
@@ -538,6 +539,7 @@ omit =
     homeassistant/components/sensor/imap.py
     homeassistant/components/sensor/imap_email_content.py
     homeassistant/components/sensor/influxdb.py
+    homeassistant/components/sensor/iota.py
     homeassistant/components/sensor/irish_rail_transport.py
     homeassistant/components/sensor/kwb.py
     homeassistant/components/sensor/lacrosse.py

homeassistant/components/iota.py

@@ -0,0 +1,98 @@
+"""
+Support for IOTA wallets.
+
+For more details about this component, please refer to the documentation at
+https://home-assistant.io/components/iota/
+"""
+import logging
+from datetime import timedelta
+
+import voluptuous as vol
+
+import homeassistant.helpers.config_validation as cv
+from homeassistant.helpers.discovery import load_platform
+from homeassistant.helpers.entity import Entity
+
+DOMAIN = 'iota'
+
+REQUIREMENTS = ['pyota==2.0.3']
+
+IOTA_PLATFORMS = ['sensor']
+
+SCAN_INTERVAL = timedelta(minutes=10)
+
+CONF_IRI = 'iri'
+CONF_TESTNET = 'testnet'
+CONF_WALLETS = 'wallets'
+CONF_WALLET_NAME = 'name'
+CONF_WALLET_SEED = 'seed'
+
+WALLET_CONFIG = vol.Schema({
+    vol.Required(CONF_WALLET_NAME): cv.string,
+    vol.Required(CONF_WALLET_SEED): cv.string,
+})
+
+CONFIG_SCHEMA = vol.Schema({
+    DOMAIN: vol.Schema({
+        vol.Required(CONF_IRI): vol.All(str),
+        vol.Optional(CONF_TESTNET, default=False): cv.boolean,
+        vol.Required(CONF_WALLETS): vol.All(cv.ensure_list, [WALLET_CONFIG])
+    })
+}, extra=vol.ALLOW_EXTRA)
+
+_LOGGER = logging.getLogger(__name__)
+
+
+def setup(hass, config):
+    """Setup IOTA component"""
+
+    # Set domain sepecific data
+    iota_config = config[DOMAIN]
+    hass.data[DOMAIN] = {
+        'iri': iota_config[CONF_IRI],
+        'is_testnet': iota_config[CONF_TESTNET],
+        'wallets': iota_config[CONF_WALLETS]
+    }
+
+    # Set states of IRI config
+    hass.states.set('iota.iri', hass.data[DOMAIN]['iri'])
+    hass.states.set('iota.is_testnet', hass.data[DOMAIN]['is_testnet'])
+
+    # Load platforms
+    for platform in IOTA_PLATFORMS:
+        load_platform(hass, platform, DOMAIN, {}, iota_config)
+
+    return True
+
+
+class IotaDevice(Entity):
+    """Representation of a IOTA device."""
+
+    def __init__(self, name, seed, iri, is_testnet=False):
+        """Initialisation of the IOTA device."""
+        self._name = name
+        self._seed = seed
+        self.iri = iri
+        self.is_testnet = is_testnet
+
+    @property
+    def name(self):
+        """Return the default name of the device."""
+        return self._name
+
+    @property
+    def should_poll(self):
+        """Get polling requirement from IOTA device."""
+        return True
+
+    @property
+    def device_state_attributes(self):
+        """Return the state attributes of the device."""
+        attr = {CONF_WALLET_NAME: self._name}
+        return attr
+
+    @property
+    def api(self):
+        """ Constructs API object for interaction with the IRI node"""
+        from iota import Iota
+        return Iota(adapter=self.iri, seed=self._seed)

homeassistant/components/sensor/iota.py

@@ -0,0 +1,87 @@
+import logging
+
+from homeassistant.components.iota import DOMAIN as IOTA_DOMAIN, IotaDevice
+
+_LOGGER = logging.getLogger(__name__)
+
+DEPENDENCIES = ['iota']
+
+
+def setup_platform(hass, config, add_devices, discovery_info=None):
+    """Set up the IOTA sensor."""
+
+    # Add sensors for wallet balance
+    iota_config = hass.data[IOTA_DOMAIN]
+    balance_sensors = [IotaBalanceSensor(wallet, iota_config) for wallet in iota_config['wallets']]
+    add_devices(balance_sensors)
+
+    # Add sensor for node information
+    add_devices([IotaNodeSensor(iota_config=iota_config)])
+
+
+class IotaBalanceSensor(IotaDevice):
+    """Implement an IOTA sensor for displaying wallets balance."""
+
+    def __init__(self, wallet_config, iota_config):
+        """Initialize the sensor."""
+
+        super().__init__(name=wallet_config['name'], seed=wallet_config['seed'],
+                         iri=iota_config['iri'], is_testnet=iota_config['is_testnet'])
+        self._state = 0
+
+    @property
+    def name(self):
+        """Return the name of the sensor."""
+        return '{} Balance'.format(self._name)
+
+    @property
+    def state(self):
+        """Return the state of the sensor."""
+        return self._state
+
+    @property
+    def unit_of_measurement(self):
+        """Return the unit of measurement."""
+        return 'IOTA'
+
+    def update(self):
+        """Fetch new balance from IRI."""
+        self._state = self.api.get_inputs()['totalBalance']
+
+
+class IotaNodeSensor(IotaDevice):
+    """Implement an IOTA sensor for displaying attributes of node."""
+
+    def __init__(self, iota_config):
+        """Initialize the sensor."""
+
+        super().__init__(name='Node Info', seed=None,
+                         iri=iota_config['iri'], is_testnet=iota_config['is_testnet'])
+        self._state = ""
+        self._attr = dict()
+
+    @property
+    def name(self):
+        """Return the name of the sensor."""
+        return 'IOTA Node'
+
+    @property
+    def state(self):
+        """Return the state of the sensor."""
+        return self._state
+
+    @property
+    def device_state_attributes(self):
+        """Return the state attributes of the device."""
+        return self._attr
+
+    @property
+    def unit_of_measurement(self):
+        """Return the unit of measurement."""
+        return ''
+
+    def update(self):
+        """Fetch new attribures IRI node."""
+        node_info = self.api.get_node_info()
+        self._state = node_info.get('appVersion')
+        self._attr = {k: str(v) for k, v in node_info.items()}  # convert values to raw string formats

requirements_all.txt

@@ -1209,3 +1209,6 @@ zeroconf==0.19.1
 
 # homeassistant.components.media_player.ziggo_mediabox_xl
 ziggo-mediabox-xl==1.0.0
+
+# homeassistant.components.iota
+pyota==2.0.3