fz0x1/HPI
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

my.emfit: expose fake data contextmanager

Browse source
This commit is contained in:
Dima Gerasimov 2020-09-19 16:47:25 +01:00 • committed by karlicoss
parent f02c572cc0
commit cd40fc75c3
1 changed files with 21 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
my/emfit/__init__.py
View file

@ -6,7 +6,7 @@ Consumes data exported by https://github.com/karlicoss/emfitexport
""" """
from datetime import date from datetime import date
from pathlib import Path from pathlib import Path
from typing import Dict, List, Iterable from typing import Dict, List, Iterable, Any
from ..core import get_files from ..core import get_files
from ..core.common import mcachew from ..core.common import mcachew
@ -30,11 +30,12 @@ def dir_hash(path: Path):
# TODO take __file__ into account somehow? # TODO take __file__ into account somehow?
@mcachew(cache_path=cache_dir() / 'emfit.cache', hashf=dir_hash, logger=dal.log) @mcachew(cache_path=cache_dir() / 'emfit.cache', hashf=lambda: dir_hash(config.export_path), logger=dal.log)
def datas(path: Path=config.export_path) -> Iterable[Res[Emfit]]: def datas() -> Iterable[Res[Emfit]]:
import dataclasses import dataclasses
# data from emfit is coming in UTC. There is no way (I think?) to know the 'real' timezone, and local times matter more for sleep analysis # data from emfit is coming in UTC. There is no way (I think?) to know the 'real' timezone, and local times matter more for sleep analysis
# TODO actully this is wrong?? check this..
emfit_tz = config.timezone emfit_tz = config.timezone
for x in dal.sleeps(config.export_path): for x in dal.sleeps(config.export_path):
@ -87,9 +88,10 @@ def pre_dataframe() -> Iterable[Res[Emfit]]:
def dataframe() -> DataFrameT: def dataframe() -> DataFrameT:
from datetime import timedelta from datetime import timedelta
dicts: List[Dict] = [] dicts: List[Dict[str, Any]] = []
last = None last = None
for s in pre_dataframe(): for s in pre_dataframe():
d: Dict[str, Any]
if isinstance(s, Exception): if isinstance(s, Exception):
edt = extract_error_datetime(s) edt = extract_error_datetime(s)
d = { d = {
@ -137,8 +139,22 @@ def stats():
return stat(pre_dataframe) return stat(pre_dataframe)
from contextlib import contextmanager
@contextmanager
def fake_data(nights=500):
from ..core.cfg import override_config
from tempfile import TemporaryDirectory
with override_config(config) as cfg, TemporaryDirectory() as td:
tdir = Path(td)
cfg.export_path = tdir
gen = dal.FakeData()
gen.fill(tdir, count=nights)
yield
# TODO remove/deprecate it? I think used by timeline # TODO remove/deprecate it? I think used by timeline
def get_datas() -> List[Emfit]: def get_datas() -> List[Emfit]:
# todo ugh. run lint properly # todo ugh. run lint properly
return list(sorted(datas(), key=lambda e: e.start)) return list(sorted(datas(), key=lambda e: e.start)) # type: ignore
# TODO move away old entries if there is a diff?? # TODO move away old entries if there is a diff??