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

fix ruci

Browse source
This commit is contained in:
Dima Gerasimov 2019-05-08 21:17:22 +01:00
parent 7b2ff3627c
commit 71eb70eaa8
3 changed files with 29 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
check
View file

@ -3,7 +3,13 @@ from datetime import datetime, timedelta
from rescuetime import get_rescuetime from rescuetime import get_rescuetime
rs = get_rescuetime(latest=1)
latest_dt = rs[-1].dt
assert (datetime.now() - latest_dt) < timedelta(days=1) def main():
rs = get_rescuetime(latest=1)
latest_dt = rs[-1].dt
assert (datetime.now() - latest_dt) < timedelta(days=1)
if __name__ == '__main__':
main()

12
rescuetime/__init__.py
View file

@ -2,7 +2,7 @@ import logging
from pathlib import Path from pathlib import Path
import json import json
from datetime import datetime, timedelta from datetime import datetime, timedelta
from typing import NamedTuple, Dict, List, Set from typing import NamedTuple, Dict, List, Set, Optional
from functools import lru_cache from functools import lru_cache
@ -47,11 +47,15 @@ class Entry(NamedTuple):
return Entry(dt=dt, duration_s=dur, activity=activity) return Entry(dt=dt, duration_s=dur, activity=activity)
@lru_cache() @lru_cache(1)
def get_rescuetime(latest=None): def get_rescuetime(latest: Optional[int]=None):
if latest is None:
latest = 0
entries: Set[Entry] = set() entries: Set[Entry] = set()
for fp in list(sorted(_PATH.glob('*.json')))[(0 if latest is None else -latest):]: # pylint: disable=invalid-unary-operand-type
for fp in list(sorted(_PATH.glob('*.json')))[-latest:]:
j = try_load(fp) j = try_load(fp)
if j is None: if j is None:
continue continue

21
rescuetime/__main__.py
View file

@ -2,15 +2,18 @@ from kython.klogging import setup_logzero
from . import get_logger, get_groups, get_rescuetime, fill_influxdb from . import get_logger, get_groups, get_rescuetime, fill_influxdb
logger = get_logger() def main():
setup_logzero(logger) logger = get_logger()
setup_logzero(logger)
# for gr in get_groups(): # for gr in get_groups():
# print(f"{gr[0].dt}--{gr[-1].dt}") # print(f"{gr[0].dt}--{gr[-1].dt}")
# for e in get_rescuetime(latest=2): # for e in get_rescuetime(latest=2):
# print(e) # print(e)
fill_influxdb() fill_influxdb()
# TODO merged db? # TODO merged db?
# TODO ok, it summarises my sleep intervals pretty well. I guess should adjust it for the fact I don't sleep during the day, and it would be ok! # TODO ok, it summarises my sleep intervals pretty well. I guess should adjust it for the fact I don't sleep during the day, and it would be ok!
if __name__ == '__main__':
main()