radical-api/core/tracker.py

110 lines
3.8 KiB
Python
Raw Normal View History

2021-08-17 02:51:42 +00:00
import json
from datetime import datetime, timedelta
from fastapi.encoders import jsonable_encoder
2021-08-17 02:51:42 +00:00
from pathlib import Path
DATA_DIR = Path.home().joinpath(".radical-api/trackers")
2021-08-17 02:51:42 +00:00
def object_hook(dct):
if "datetime" in dct:
return TrackerPoint(dct["datetime"], dct["value"])
if "date_str" in dct: # old format
return TrackerPoint(dct["date_str"], dct["value"])
if "name" in dct:
return Tracker(dct["name"], dct["points"])
class TrackerPoint:
def __init__(self, date_str: str, value: int) -> None:
self.datetime = datetime.fromisoformat(date_str)
2021-08-17 02:51:42 +00:00
self.value = value
class Tracker:
def __init__(self, name: str, points: list[TrackerPoint]):
2021-08-17 02:51:42 +00:00
self.name = name
self.points = points
def rename(self, name):
"""Rename the Tracker."""
2021-08-17 02:51:42 +00:00
DATA_DIR.joinpath(f"{self.name}.json").rename(
DATA_DIR.joinpath(f"{name}.json"))
self.name = name
self.save()
def delete(self):
"""Delete the Tracker."""
2021-08-17 02:51:42 +00:00
filepath = DATA_DIR.joinpath(f"{self.name}.json")
filepath.unlink()
def save(self):
"""Save the tracker to JSON."""
DATA_DIR.mkdir(exist_ok=True)
filepath = DATA_DIR.joinpath(f"{self.name}.json")
filepath.touch(exist_ok=True)
with filepath.open("w+") as fp:
data = jsonable_encoder(self)
json.dump(data, fp, indent=4)
2021-08-17 02:51:42 +00:00
def modify_point(self, date_str: str, value: int):
2021-08-17 02:51:42 +00:00
"""Modify a point. Change its assigned value to the one given."""
date_time = datetime.fromisoformat(date_str)
2021-08-17 02:51:42 +00:00
for point in self.points:
if point.datetime == date_time:
point.value = value
break
self.save()
def add_point(self, date_str: str, value: int):
2021-08-17 02:51:42 +00:00
"""Add a point to the tracker."""
point = TrackerPoint(date_str, value)
2021-08-17 02:51:42 +00:00
self.points.append(point)
self.save()
def delete_point(self, date_str: str):
2021-08-17 02:51:42 +00:00
"""Remove a point from the tracker."""
date_time = datetime.fromisoformat(date_str)
2021-08-17 02:51:42 +00:00
for point in self.points:
if point.datetime == date_time:
self.points.remove(point)
break
self.save()
def list_points(self, start_date: datetime = None, end_date: datetime = None) -> list:
"""
Return a list of all points that fall in-between the given start_date and end_date.
:param start_date: the date the data returned should start at.
If this is not given, it'll be a week from the given end_date.
:param end_date: the date the data returned should end at. (included in output)
If this is not given, it'll be today.
:return: list of all points that fall in-between the given start_date and end_date.
:raises ValueError: if either date parameter is set in the future, or if the end_date is before the start_date.
"""
point_list = []
if end_date is None:
end_date = datetime.now().date()
if start_date is None:
start_date = (end_date - timedelta(days=7)) # assume we'll start a week prior to today.
if end_date > datetime.now().date() or start_date > datetime.now().date() or end_date < start_date:
raise ValueError(end_date)
for point in self.points:
if start_date <= point.datetime.date() <= end_date:
point_list.append(point)
return point_list
2021-08-17 02:51:42 +00:00
@classmethod
def from_data(cls, name):
"""Load a tracker from the DATA_DIR."""
2021-08-17 02:51:42 +00:00
filepath = DATA_DIR.joinpath(f"{name}.json")
return cls.from_json(filepath.read_text())
@staticmethod
def from_json(json_str):
"""Load a tracker from a JSON string."""
return json.loads(json_str, object_hook=object_hook)