104 lines
3.4 KiB
Python
104 lines
3.4 KiB
Python
'''
|
|
Simple location provider, serving as a fallback when more detailed data isn't available
|
|
'''
|
|
|
|
from dataclasses import dataclass
|
|
from datetime import datetime, time, timezone
|
|
from functools import lru_cache
|
|
from typing import Sequence, Tuple, Union, cast, List, Iterator
|
|
|
|
from my.config import location as user_config
|
|
|
|
from my.location.common import LatLon, DateIsh
|
|
from my.location.fallback.common import FallbackLocation, DateExact
|
|
|
|
@dataclass
|
|
class Config(user_config):
|
|
home: Union[
|
|
LatLon, # either single, 'current' location
|
|
Sequence[Tuple[ # or, a sequence of location history
|
|
DateIsh, # date when you moved to
|
|
LatLon, # the location
|
|
]]
|
|
]
|
|
|
|
# default ~30km accuracy
|
|
# this is called 'home_accuracy' since it lives on the base location.config object,
|
|
# to differentiate it from accuracy for other providers
|
|
home_accuracy: float = 30_000.0
|
|
|
|
# TODO could make current Optional and somehow determine from system settings?
|
|
@property
|
|
def _history(self) -> Sequence[Tuple[datetime, LatLon]]:
|
|
home1 = self.home
|
|
# todo ugh, can't test for isnstance LatLon, it's a tuple itself
|
|
home2: Sequence[Tuple[DateIsh, LatLon]]
|
|
if isinstance(home1[0], tuple):
|
|
# already a sequence
|
|
home2 = cast(Sequence[Tuple[DateIsh, LatLon]], home1)
|
|
else:
|
|
# must be a pair of coordinates. also doesn't really matter which date to pick?
|
|
loc = cast(LatLon, home1)
|
|
home2 = [(datetime.min, loc)]
|
|
|
|
# todo cache?
|
|
res = []
|
|
for x, loc in home2:
|
|
dt: datetime
|
|
if isinstance(x, str):
|
|
dt = datetime.fromisoformat(x)
|
|
elif isinstance(x, datetime):
|
|
dt = x
|
|
else:
|
|
dt = datetime.combine(x, time.min)
|
|
# todo not sure about doing it here, but makes it easier to compare..
|
|
if dt.tzinfo is None:
|
|
dt = dt.replace(tzinfo=timezone.utc)
|
|
res.append((dt, loc))
|
|
res = sorted(res, key=lambda p: p[0])
|
|
return res
|
|
|
|
|
|
from ...core.cfg import make_config
|
|
config = make_config(Config)
|
|
|
|
|
|
@lru_cache(maxsize=None)
|
|
def get_location(dt: datetime) -> LatLon:
|
|
'''
|
|
Interpolates the location at dt
|
|
'''
|
|
loc = list(estimate_location(dt))
|
|
assert len(loc) == 1
|
|
return loc[0].lat, loc[0].lon
|
|
|
|
|
|
# TODO: in python3.8, use functools.cached_property instead?
|
|
@lru_cache(maxsize=None)
|
|
def homes_cached() -> List[Tuple[datetime, LatLon]]:
|
|
return list(config._history)
|
|
|
|
|
|
def estimate_location(dt: DateExact) -> Iterator[FallbackLocation]:
|
|
from my.location.fallback.common import _datetime_timestamp
|
|
d: float = _datetime_timestamp(dt)
|
|
hist = list(reversed(homes_cached()))
|
|
for pdt, (lat, lon) in hist:
|
|
if d >= pdt.timestamp():
|
|
yield FallbackLocation(
|
|
lat=lat,
|
|
lon=lon,
|
|
accuracy=config.home_accuracy,
|
|
dt=datetime.fromtimestamp(d, timezone.utc),
|
|
datasource='via_home')
|
|
return
|
|
else:
|
|
# I guess the most reasonable is to fallback on the first location
|
|
lat, lon = hist[-1][1]
|
|
yield FallbackLocation(
|
|
lat=lat,
|
|
lon=lon,
|
|
accuracy=config.home_accuracy,
|
|
dt=datetime.fromtimestamp(d, timezone.utc),
|
|
datasource='via_home')
|
|
return
|