Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .coveragerc
Original file line number Diff line number Diff line change
Expand Up @@ -715,6 +715,7 @@ omit =
homeassistant/components/sensor/kwb.py
homeassistant/components/sensor/lacrosse.py
homeassistant/components/sensor/lastfm.py
homeassistant/components/sensor/launch.py
homeassistant/components/sensor/linky.py
homeassistant/components/sensor/linux_battery.py
homeassistant/components/sensor/loopenergy.py
Expand Down
87 changes: 87 additions & 0 deletions homeassistant/components/sensor/launch.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,87 @@
"""
A sensor platform that give you information about the next space launch.

For more details about this platform, please refer to the documentation at
https://www.home-assistant.io/components/sensor.launch/
"""
import logging

import voluptuous as vol

import homeassistant.helpers.config_validation as cv
from homeassistant.components.sensor import PLATFORM_SCHEMA
from homeassistant.const import ATTR_ATTRIBUTION, CONF_NAME
from homeassistant.helpers.entity import Entity

REQUIREMENTS = ['pylaunches==0.0.2']

_LOGGER = logging.getLogger(__name__)

CONF_ATTRIBUTION = "Data provided by Launch Library."
Comment thread
ludeeus marked this conversation as resolved.
Outdated

DEFAULT_NAME = 'Launch'

PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
})


async def async_setup_platform(
hass, config, async_add_entities, discovery_info=None):
"""Create the launch sensor."""
from pylaunches.api import Launches

name = config[CONF_NAME]

launches = Launches(hass.loop)
sensor = [LaunchSensor(launches, name)]
async_add_entities(sensor, True)


class LaunchSensor(Entity):
"""Representation of a launch Sensor."""

def __init__(self, launches, name):
"""Initialize the sensor."""
self.launches = launches
self._attributes = {}
self._name = name
self._state = None

async def async_update(self):
"""Get the latest data."""
await self.launches.get_launches()
if self.launches.launches is None:
_LOGGER.error("No data recieved.")
Comment thread
ludeeus marked this conversation as resolved.
Outdated
return
try:
data = self.launches.launches[0]
self._state = data['name']
self._attributes['launch_time'] = data['start']
self._attributes['agency'] = data['agency']
self._attributes['agency_country_code'] = (data
['agency_country_code'])
Comment thread
ludeeus marked this conversation as resolved.
Outdated
self._attributes['stream'] = data['stream']
Comment thread
ludeeus marked this conversation as resolved.
self._attributes[ATTR_ATTRIBUTION] = CONF_ATTRIBUTION
except (KeyError, IndexError) as error:
_LOGGER.debug("Error getting data, %s", error)

@property
def name(self):
"""Return the name of the sensor."""
return self._name

@property
def state(self):
"""Return the state of the sensor."""
return self._state

@property
def icon(self):
"""Return the icon of the sensor."""
return 'mdi:rocket'

@property
def device_state_attributes(self):
"""Return attributes for the sensor."""
return self._attributes
3 changes: 3 additions & 0 deletions requirements_all.txt
Original file line number Diff line number Diff line change
Expand Up @@ -957,6 +957,9 @@ pylacrosse==0.3.1
# homeassistant.components.sensor.lastfm
pylast==2.4.0

# homeassistant.components.sensor.launch
pylaunches==0.0.2

# homeassistant.components.media_player.lg_netcast
pylgnetcast-homeassistant==0.2.0.dev0

Expand Down