HPI/my/bluemaestro/__init__.py
2019-12-19 20:14:26 +00:00

99 lines
2.9 KiB
Python
Executable file

#!/usr/bin/python3
import logging
import sqlite3
from datetime import datetime
from itertools import chain, islice
from pathlib import Path
from typing import Any, Dict, Iterable, NamedTuple, Set
from cachew import cachew
# TODO move to common??
from kython import dictify
# TODO vendorize in my. pkg? It's quite handy...
from kython.klogging import LazyLogger
from mycfg import paths
logger = LazyLogger('bluemaestro', level=logging.DEBUG)
def get_backup_files():
return list(sorted(chain.from_iterable(d.glob('*.db') for d in paths.bluemaestro.export_paths)))
class Point(NamedTuple):
dt: datetime
temp: float
# TODO hmm, does cachew have py.typed?
@cachew(cache_path=paths.bluemaestro.cache)
def iter_points(dbs) -> Iterable[Point]:
# I guess we can affort keeping them in sorted order
points: Set[Point] = set()
# TODO do some sanity check??
for f in dbs:
# err = f'{f}: mismatch: {v} vs {value}'
# if abs(v - value) > 0.4:
# logger.warning(err)
# # TODO mm. dunno how to mark errors properly..
# # raise AssertionError(err)
# else:
# pass
with sqlite3.connect(str(f)) as db:
datas = list(db.execute('select * from data'))
for _, tss, temp, hum, pres, dew in datas:
# TODO is that utc???
tss = tss.replace('Juli', 'Jul').replace('Aug.', 'Aug')
dt = datetime.strptime(tss, '%Y-%b-%d %H:%M')
p = Point(
dt=dt,
temp=temp,
# TODO use pressure and humidity as well
)
if p in points:
continue
points.add(p)
for p in sorted(points, key=lambda p: p.dt):
yield p
# logger.info('total items: %d', len(merged))
# TODO assert frequency?
# for k, v in merged.items():
# # TODO shit. quite a few of them have varying values... how is that freaking possible????
# # most of them are within 0.5 degree though... so just ignore?
# if isinstance(v, set) and len(v) > 1:
# print(k, v)
# for k, v in merged.items():
# yield Point(dt=k, temp=v) # meh?
# TODO does it even have to be a dict?
# @dictify(key=lambda p: p.dt)
def get_temperature(backups=get_backup_files()): # TODO misleading name
return list(iter_points(backups))
def get_dataframe():
"""
%matplotlib gtk
from my.bluemaestro import get_dataframe
get_dataframe().plot()
"""
import pandas as pd # type: ignore
return pd.DataFrame(p._asdict() for p in get_temperature()).set_index('dt')
def test():
print(get_temperature(get_backup_files()[-1:]))
def main():
ll = list(iter_points(get_backup_files()))
print(len(ll))
# print(get_temperature(get_backup_files()[-1:]))
# print(type(t))
if __name__ == '__main__':
main()