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

my.emfit: add missing properties to fake data generator

Browse source
This commit is contained in:
karlicoss 2023-11-10 02:00:37 +00:00
parent ac5f71c68b
commit 65c617ed94
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
my/emfit/__init__.py
View file

@ -170,6 +170,8 @@ def fake_data(nights: int = 500) -> Iterator:
from my.core.cfg import tmp_config from my.core.cfg import tmp_config
from tempfile import TemporaryDirectory from tempfile import TemporaryDirectory
import pytz
with TemporaryDirectory() as td: with TemporaryDirectory() as td:
tdir = Path(td) tdir = Path(td)
gen = dal.FakeData() gen = dal.FakeData()
@ -178,6 +180,8 @@ def fake_data(nights: int = 500) -> Iterator:
class override: class override:
class emfit: class emfit:
export_path = tdir export_path = tdir
excluded_sids = ()
timezone = pytz.timezone('Europe/London') # meh
with tmp_config(modules=__name__, config=override) as cfg: with tmp_config(modules=__name__, config=override) as cfg:
yield cfg yield cfg