Iota wallet

.coveragerc

@@ -368,6 +368,7 @@ omit =
     homeassistant/components/image_processing/dlib_face_detect.py
     homeassistant/components/image_processing/dlib_face_identify.py
     homeassistant/components/image_processing/seven_segments.py
+    homeassistant/components/iota.py
     homeassistant/components/keyboard.py
     homeassistant/components/keyboard_remote.py
     homeassistant/components/light/avion.py
@@ -549,6 +550,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,82 @@
+"""
+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): cv.string,
+        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."""
+
+    # Load platforms
+    iota_config = config[DOMAIN]
+    for platform in IOTA_PLATFORMS:
+        load_platform(hass, platform, DOMAIN, iota_config, 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 device_state_attributes(self):
+        """Return the state attributes of the device."""
+        attr = {CONF_WALLET_NAME: self._name}
+        return attr
+
+    @property
+    def api(self):
+        """Construct 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,103 @@
+"""
+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
+
+from homeassistant.components.iota import IotaDevice
+
+_LOGGER = logging.getLogger(__name__)
+
+DEPENDENCIES = ['iota']
+
+SCAN_INTERVAL = timedelta(minutes=10)
+
+
+def setup_platform(hass, config, add_devices, discovery_info=None):
+    """Set up the IOTA sensor."""
+    # Add sensors for wallet balance
+    iota_config = discovery_info
+    sensors = [IotaBalanceSensor(wallet, iota_config)
+               for wallet in iota_config['wallets']]
+
+    # Add sensor for node information
+    sensors.append(IotaNodeSensor(iota_config=iota_config))
+
+    add_devices(sensors)
+
+
+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['testnet'])
+        self._state = None
+
+    @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['testnet'])
+        self._state = None
+        self._attr = {}
+
+    @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')
+
+        # convert values to raw string formats
+        self._attr = {k: str(v) for k, v in node_info.items()}
+
+        # add values of iri config
+        self._attr['url'] = self.iri
+        self._attr['testnet'] = self.is_testnet

requirements_all.txt

@@ -811,6 +811,9 @@ pynut2==2.1.2
 # homeassistant.components.binary_sensor.nx584
 pynx584==0.4
 
+# homeassistant.components.iota
+pyota==2.0.3
+
 # homeassistant.components.sensor.otp
 pyotp==2.2.6