69 lines
2.9 KiB
Python
69 lines
2.9 KiB
Python
import struct
|
|
from toggl.TogglPy import Toggl
|
|
from zei.ZeiDelegate import ZeiDelegate
|
|
from datetime import datetime, timezone
|
|
from Mapping import Mapping
|
|
|
|
import dateutil.parser
|
|
import logging
|
|
|
|
_log = logging.getLogger(__name__)
|
|
_log.addHandler(logging.StreamHandler())
|
|
_log.setLevel(logging.INFO)
|
|
|
|
class TogglDelegate(ZeiDelegate):
|
|
|
|
def __init__(self, periph, config):
|
|
self.config = config
|
|
self.toggl = Toggl()
|
|
self.toggl.setAPIKey(self.config['toggl']['settings']['token'])
|
|
self._populateProjects()
|
|
self._populateMappings(self.config['mappings'])
|
|
super().__init__(periph)
|
|
|
|
def handleNotification(self, cHandle, data):
|
|
if cHandle == 38: # Side Change Notification
|
|
side = struct.unpack('B', data)[0]
|
|
self._trackProjectByMapping(self.mappings[side])
|
|
else:
|
|
_log.info("Notification from hndl: %s - %r", cHandle, data)
|
|
|
|
def _trackProjectByMapping(self, mapping: Mapping):
|
|
self._trackProject(description=mapping.description, pid=mapping.id, tags=mapping.tags)
|
|
|
|
def _trackProject(self, description: str, pid: int, tags: list):
|
|
current = self.toggl.currentRunningTimeEntry()['data']
|
|
|
|
if current is not None:
|
|
if (datetime.now(timezone.utc) - dateutil.parser.isoparse(current['start'])).total_seconds() < 20:
|
|
# Delete entry if not older than 20s
|
|
_log.info("Abort currently running entry")
|
|
self.toggl.deleteTimeEntry(current['id'])
|
|
else:
|
|
_log.info("Stopping currently running entry")
|
|
self.toggl.stopTimeEntry(current['id'])
|
|
|
|
|
|
_log.info("Now tracking project %s: %s (%s)", self.projects[pid]['name'], description, ', '.join(tags if tags else []))
|
|
if pid == 0:
|
|
return
|
|
|
|
self.toggl.startTimeEntry(description, pid=pid, tags=tags)
|
|
|
|
def _populateMappings(self, mappings: dict):
|
|
self.mappings = { 0: Mapping(0, 0)}
|
|
for i in mappings:
|
|
self.mappings[int(i)] = Mapping(int(i), int(mappings[i]["id"]))
|
|
if "description" in mappings[i]:
|
|
self.mappings[int(i)].description = mappings[i]["description"]
|
|
if "tags" in mappings[i]:
|
|
self.mappings[int(i)].tags = mappings[i]["tags"]
|
|
|
|
def _populateProjects(self):
|
|
self.projects = {}
|
|
proj = (self.toggl.getWorkspaceProjects(self.config['toggl']['settings']['workspace_id']))
|
|
NoneProj = {'id': 0, 'wid': int(self.config['toggl']['settings']['workspace_id']), 'name': 'None', 'billable': False, 'is_private': True, 'active': True, 'template': False, 'at': '2020-06-09T04:02:38+00:00', 'created_at': '2019-12-09T16:36:28+00:00', 'color': '9', 'auto_estimates': False, 'actual_hours': 0, 'hex_color': '#990099'}
|
|
|
|
self.projects[0] = NoneProj
|
|
for i in proj:
|
|
self.projects[i['id']] = i |