148 lines
4.9 KiB
Python
148 lines
4.9 KiB
Python
"""Support for the NOAA Tides and Currents API."""
|
|
from datetime import datetime, timedelta
|
|
import logging
|
|
|
|
import noaa_coops as coops # pylint: disable=import-error
|
|
import requests
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.sensor import PLATFORM_SCHEMA
|
|
from homeassistant.const import (
|
|
ATTR_ATTRIBUTION,
|
|
CONF_NAME,
|
|
CONF_TIME_ZONE,
|
|
CONF_UNIT_SYSTEM,
|
|
)
|
|
from homeassistant.exceptions import PlatformNotReady
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.helpers.entity import Entity
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
CONF_STATION_ID = "station_id"
|
|
|
|
DEFAULT_ATTRIBUTION = "Data provided by NOAA"
|
|
DEFAULT_NAME = "NOAA Tides"
|
|
DEFAULT_TIMEZONE = "lst_ldt"
|
|
|
|
SCAN_INTERVAL = timedelta(minutes=60)
|
|
|
|
TIMEZONES = ["gmt", "lst", "lst_ldt"]
|
|
UNIT_SYSTEMS = ["english", "metric"]
|
|
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
|
|
{
|
|
vol.Required(CONF_STATION_ID): cv.string,
|
|
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
|
|
vol.Optional(CONF_TIME_ZONE, default=DEFAULT_TIMEZONE): vol.In(TIMEZONES),
|
|
vol.Optional(CONF_UNIT_SYSTEM): vol.In(UNIT_SYSTEMS),
|
|
}
|
|
)
|
|
|
|
|
|
def setup_platform(hass, config, add_entities, discovery_info=None):
|
|
"""Set up the NOAA Tides and Currents sensor."""
|
|
station_id = config[CONF_STATION_ID]
|
|
name = config.get(CONF_NAME)
|
|
timezone = config.get(CONF_TIME_ZONE)
|
|
|
|
if CONF_UNIT_SYSTEM in config:
|
|
unit_system = config[CONF_UNIT_SYSTEM]
|
|
elif hass.config.units.is_metric:
|
|
unit_system = UNIT_SYSTEMS[1]
|
|
else:
|
|
unit_system = UNIT_SYSTEMS[0]
|
|
|
|
try:
|
|
station = coops.Station(station_id, unit_system)
|
|
except KeyError:
|
|
_LOGGER.error("NOAA Tides Sensor station_id %s does not exist", station_id)
|
|
return
|
|
except requests.exceptions.ConnectionError as exception:
|
|
_LOGGER.error(
|
|
"Connection error during setup in NOAA Tides Sensor for station_id: %s",
|
|
station_id,
|
|
)
|
|
raise PlatformNotReady from exception
|
|
|
|
noaa_sensor = NOAATidesAndCurrentsSensor(
|
|
name, station_id, timezone, unit_system, station
|
|
)
|
|
|
|
add_entities([noaa_sensor], True)
|
|
|
|
|
|
class NOAATidesAndCurrentsSensor(Entity):
|
|
"""Representation of a NOAA Tides and Currents sensor."""
|
|
|
|
def __init__(self, name, station_id, timezone, unit_system, station):
|
|
"""Initialize the sensor."""
|
|
self._name = name
|
|
self._station_id = station_id
|
|
self._timezone = timezone
|
|
self._unit_system = unit_system
|
|
self._station = station
|
|
self.data = None
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the sensor."""
|
|
return self._name
|
|
|
|
@property
|
|
def device_state_attributes(self):
|
|
"""Return the state attributes of this device."""
|
|
attr = {ATTR_ATTRIBUTION: DEFAULT_ATTRIBUTION}
|
|
if self.data is None:
|
|
return attr
|
|
if self.data["hi_lo"][1] == "H":
|
|
attr["high_tide_time"] = self.data.index[1].strftime("%Y-%m-%dT%H:%M")
|
|
attr["high_tide_height"] = self.data["predicted_wl"][1]
|
|
attr["low_tide_time"] = self.data.index[2].strftime("%Y-%m-%dT%H:%M")
|
|
attr["low_tide_height"] = self.data["predicted_wl"][2]
|
|
elif self.data["hi_lo"][1] == "L":
|
|
attr["low_tide_time"] = self.data.index[1].strftime("%Y-%m-%dT%H:%M")
|
|
attr["low_tide_height"] = self.data["predicted_wl"][1]
|
|
attr["high_tide_time"] = self.data.index[2].strftime("%Y-%m-%dT%H:%M")
|
|
attr["high_tide_height"] = self.data["predicted_wl"][2]
|
|
return attr
|
|
|
|
@property
|
|
def state(self):
|
|
"""Return the state of the device."""
|
|
if self.data is None:
|
|
return None
|
|
api_time = self.data.index[0]
|
|
if self.data["hi_lo"][0] == "H":
|
|
tidetime = api_time.strftime("%-I:%M %p")
|
|
return f"High tide at {tidetime}"
|
|
if self.data["hi_lo"][0] == "L":
|
|
tidetime = api_time.strftime("%-I:%M %p")
|
|
return f"Low tide at {tidetime}"
|
|
return None
|
|
|
|
def update(self):
|
|
"""Get the latest data from NOAA Tides and Currents API."""
|
|
begin = datetime.now()
|
|
delta = timedelta(days=2)
|
|
end = begin + delta
|
|
try:
|
|
df_predictions = self._station.get_data(
|
|
begin_date=begin.strftime("%Y%m%d %H:%M"),
|
|
end_date=end.strftime("%Y%m%d %H:%M"),
|
|
product="predictions",
|
|
datum="MLLW",
|
|
interval="hilo",
|
|
units=self._unit_system,
|
|
time_zone=self._timezone,
|
|
)
|
|
self.data = df_predictions.head()
|
|
_LOGGER.debug("Data = %s", self.data)
|
|
_LOGGER.debug(
|
|
"Recent Tide data queried with start time set to %s",
|
|
begin.strftime("%m-%d-%Y %H:%M"),
|
|
)
|
|
except ValueError as err:
|
|
_LOGGER.error("Check NOAA Tides and Currents: %s", err.args)
|
|
self.data = None
|