mirror of https://github.com/esphome/esphome.git
134 lines
4.1 KiB
Python
134 lines
4.1 KiB
Python
import datetime
|
|
import logging
|
|
import math
|
|
|
|
import voluptuous as vol
|
|
|
|
import esphomeyaml.config_validation as cv
|
|
from esphomeyaml.const import CONF_TIMEZONE
|
|
from esphomeyaml.helpers import add, add_job, esphomelib_ns
|
|
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
PLATFORM_SCHEMA = cv.PLATFORM_SCHEMA.extend({
|
|
|
|
})
|
|
|
|
time_ns = esphomelib_ns.namespace('time')
|
|
|
|
|
|
def _tz_timedelta(td):
|
|
offset_hour = int(td.total_seconds() / (60 * 60))
|
|
offset_minute = int(abs(td.total_seconds() / 60)) % 60
|
|
offset_second = int(abs(td.total_seconds())) % 60
|
|
if offset_hour == 0 and offset_minute == 0 and offset_second == 0:
|
|
return '0'
|
|
elif offset_minute == 0 and offset_second == 0:
|
|
return '{}'.format(offset_hour)
|
|
elif offset_second == 0:
|
|
return '{}:{}'.format(offset_hour, offset_minute)
|
|
return '{}:{}:{}'.format(offset_hour, offset_minute, offset_second)
|
|
|
|
|
|
# https://stackoverflow.com/a/16804556/8924614
|
|
def _week_of_month(dt):
|
|
first_day = dt.replace(day=1)
|
|
dom = dt.day
|
|
adjusted_dom = dom + first_day.weekday()
|
|
return int(math.ceil(adjusted_dom / 7.0))
|
|
|
|
|
|
def _tz_dst_str(dt):
|
|
td = datetime.timedelta(hours=dt.hour, minutes=dt.minute, seconds=dt.second)
|
|
return 'M{}.{}.{}/{}'.format(dt.month, _week_of_month(dt), dt.isoweekday() % 7,
|
|
_tz_timedelta(td))
|
|
|
|
|
|
def detect_tz():
|
|
try:
|
|
import tzlocal
|
|
import pytz
|
|
except ImportError:
|
|
raise vol.Invalid("No timezone specified and 'tzlocal' not installed. To automatically "
|
|
"detect the timezone please install tzlocal (pip2 install tzlocal)")
|
|
try:
|
|
tz = tzlocal.get_localzone()
|
|
except pytz.exceptions.UnknownTimeZoneError:
|
|
_LOGGER.warning("Could not auto-detect timezone. Using UTC...")
|
|
return 'UTC'
|
|
|
|
def _dst(dt, is_dst):
|
|
try:
|
|
return tz.dst(dt, is_dst=is_dst)
|
|
except TypeError: # stupid pytz...
|
|
return tz.dst(dt)
|
|
|
|
def _tzname(dt, is_dst):
|
|
try:
|
|
return tz.tzname(dt, is_dst=is_dst)
|
|
except TypeError: # stupid pytz...
|
|
return tz.tzname(dt)
|
|
|
|
def _utcoffset(dt, is_dst):
|
|
try:
|
|
return tz.utcoffset(dt, is_dst=is_dst)
|
|
except TypeError: # stupid pytz...
|
|
return tz.utcoffset(dt)
|
|
|
|
dst_begins = None
|
|
dst_tzname = None
|
|
dst_utcoffset = None
|
|
dst_ends = None
|
|
norm_tzname = None
|
|
norm_utcoffset = None
|
|
|
|
hour = datetime.timedelta(hours=1)
|
|
this_year = datetime.datetime.now().year
|
|
dt = datetime.datetime(year=this_year, month=1, day=1)
|
|
last_dst = None
|
|
while dt.year == this_year:
|
|
current_dst = _dst(dt, not last_dst)
|
|
is_dst = bool(current_dst)
|
|
if is_dst != last_dst:
|
|
if is_dst:
|
|
dst_begins = dt
|
|
dst_tzname = _tzname(dt, True)
|
|
dst_utcoffset = _utcoffset(dt, True)
|
|
else:
|
|
dst_ends = dt + hour
|
|
norm_tzname = _tzname(dt, False)
|
|
norm_utcoffset = _utcoffset(dt, False)
|
|
last_dst = is_dst
|
|
dt += hour
|
|
|
|
tzbase = '{}{}'.format(norm_tzname, _tz_timedelta(-1 * norm_utcoffset))
|
|
if dst_begins is None:
|
|
# No DST in this timezone
|
|
_LOGGER.info("Auto-detected timezone '%s' with UTC offset %s",
|
|
norm_tzname, _tz_timedelta(norm_utcoffset))
|
|
return tzbase
|
|
tzext = '{}{},{},{}'.format(dst_tzname, _tz_timedelta(-1 * dst_utcoffset),
|
|
_tz_dst_str(dst_begins), _tz_dst_str(dst_ends))
|
|
_LOGGER.info("Auto-detected timezone '%s' with UTC offset %s and daylight savings time from "
|
|
"%s to %s",
|
|
norm_tzname, _tz_timedelta(norm_utcoffset), dst_begins.strftime("%x %X"),
|
|
dst_ends.strftime("%x %X"))
|
|
return tzbase + tzext
|
|
|
|
|
|
TIME_PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
|
|
vol.Optional(CONF_TIMEZONE, default=detect_tz): cv.string,
|
|
})
|
|
|
|
|
|
def setup_time_core_(time_var, config):
|
|
add(time_var.set_timezone(config[CONF_TIMEZONE]))
|
|
|
|
|
|
def setup_time(time_var, config):
|
|
add_job(setup_time_core_, time_var, config)
|
|
|
|
|
|
BUILD_FLAGS = '-DUSE_TIME'
|