2024-07-09 22:15:45 +00:00
|
|
|
"""# Git-based Time Tracker.
|
|
|
|
|
|
|
|
Quick and dirty time tracker on git histories.
|
|
|
|
|
|
|
|
Uses the simple heuristics that each commit takes precisely one hour of work.
|
|
|
|
It will automatically trim commits below one hour if another commit occurred
|
|
|
|
less than an hour ago.
|
|
|
|
|
|
|
|
Usage:
|
|
|
|
|
|
|
|
```
|
|
|
|
python -m git_time_tracker REPO1 REPO2...
|
|
|
|
```
|
|
|
|
|
|
|
|
# Obligatory
|
|
|
|
|
|
|
|
This tool reports:
|
|
|
|
|
|
|
|
```
|
|
|
|
project Jmaa/git-time-tracker.git 3h 33m (2024)
|
|
|
|
```
|
|
|
|
|
|
|
|
And the ([Hamster](https://github.com/projecthamster/hamster)) manual time tracker reports:
|
|
|
|
|
|
|
|
![](docs/obligatory-hamster.png)
|
|
|
|
"""
|
|
|
|
|
2024-06-03 20:42:05 +00:00
|
|
|
import argparse
|
2024-06-03 21:01:45 +00:00
|
|
|
import datetime
|
2024-06-08 12:43:44 +00:00
|
|
|
import logging
|
|
|
|
import sys
|
2024-08-25 22:14:12 +00:00
|
|
|
from collections.abc import Iterator
|
2024-06-03 21:13:16 +00:00
|
|
|
from pathlib import Path
|
2024-06-03 20:42:05 +00:00
|
|
|
|
2024-08-25 22:57:51 +00:00
|
|
|
from .data import (
|
|
|
|
HIDDEN_LABEL_PREFIX,
|
|
|
|
HIDDEN_LABEL_TOTAL,
|
|
|
|
RealizedWorkSample,
|
|
|
|
WorkSample,
|
|
|
|
)
|
2024-08-25 22:26:04 +00:00
|
|
|
from .format import cli, icalendar
|
2024-08-25 21:39:40 +00:00
|
|
|
from .source import git_repo
|
2024-06-03 22:15:18 +00:00
|
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
2024-06-08 12:43:44 +00:00
|
|
|
|
2024-08-25 22:14:12 +00:00
|
|
|
DEFAULT_ESTIMATED_DURATION = datetime.timedelta(hours=1)
|
2024-06-08 12:43:44 +00:00
|
|
|
ZERO_DURATION = datetime.timedelta(seconds=0)
|
|
|
|
HOUR = datetime.timedelta(hours=1)
|
2024-06-08 12:49:26 +00:00
|
|
|
MINUTE = datetime.timedelta(minutes=1)
|
2024-06-08 12:43:44 +00:00
|
|
|
|
2024-06-03 21:49:42 +00:00
|
|
|
|
2024-08-25 22:57:51 +00:00
|
|
|
def filter_samples(
|
|
|
|
samples: list[WorkSample], sample_filter: set[str],
|
|
|
|
) -> list[WorkSample]:
|
2024-08-25 22:14:12 +00:00
|
|
|
assert len(sample_filter) > 0
|
2024-08-25 22:26:04 +00:00
|
|
|
return [s for s in samples if set(s.labels).intersection(sample_filter)]
|
2024-08-25 22:14:12 +00:00
|
|
|
|
2024-08-25 22:57:51 +00:00
|
|
|
|
|
|
|
def heuristically_realize_samples(
|
|
|
|
samples: list[WorkSample],
|
|
|
|
) -> Iterator[RealizedWorkSample]:
|
2024-08-25 22:14:12 +00:00
|
|
|
"""Secret sauce.
|
|
|
|
|
|
|
|
Guarentees that:
|
|
|
|
* No samples overlap.
|
|
|
|
"""
|
|
|
|
|
|
|
|
previous_sample_end = datetime.datetime.fromtimestamp(0, datetime.UTC)
|
|
|
|
for sample in samples:
|
|
|
|
end_at = sample.end_at
|
|
|
|
|
|
|
|
assert previous_sample_end <= end_at, 'Iterating in incorrect order'
|
|
|
|
|
|
|
|
# TODO: Allow end_at is None
|
|
|
|
|
|
|
|
start_at = sample.start_at
|
|
|
|
if start_at is None:
|
|
|
|
estimated_duration: datetime.timedelta = DEFAULT_ESTIMATED_DURATION
|
|
|
|
start_at = max(previous_sample_end, end_at - estimated_duration)
|
|
|
|
del estimated_duration
|
|
|
|
|
|
|
|
yield RealizedWorkSample(labels=sample.labels, end_at=end_at, start_at=start_at)
|
|
|
|
|
|
|
|
previous_sample_end = sample.end_at
|
|
|
|
del sample
|
|
|
|
|
2024-08-25 22:57:51 +00:00
|
|
|
|
2024-06-08 13:17:04 +00:00
|
|
|
def parse_arguments():
|
|
|
|
parser = argparse.ArgumentParser()
|
2024-08-25 21:41:42 +00:00
|
|
|
parser.add_argument(
|
2024-08-25 21:46:12 +00:00
|
|
|
'--git-repo',
|
|
|
|
action='extend',
|
|
|
|
nargs='+',
|
|
|
|
type=Path,
|
|
|
|
dest='repositories',
|
2024-08-25 21:41:42 +00:00
|
|
|
)
|
|
|
|
parser.add_argument(
|
|
|
|
'--filter',
|
|
|
|
action='extend',
|
|
|
|
nargs='+',
|
|
|
|
type=str,
|
|
|
|
dest='sample_filter',
|
|
|
|
default=[],
|
|
|
|
)
|
2024-08-25 22:26:04 +00:00
|
|
|
parser.add_argument(
|
|
|
|
'--format',
|
|
|
|
action='store',
|
|
|
|
type=str,
|
|
|
|
dest='format_mode',
|
|
|
|
default='cli_report',
|
|
|
|
choices=['cli_report', 'icalendar'],
|
|
|
|
)
|
2024-06-08 13:17:04 +00:00
|
|
|
return parser.parse_args()
|
|
|
|
|
|
|
|
|
2024-06-03 20:42:05 +00:00
|
|
|
def main():
|
2024-06-03 22:15:18 +00:00
|
|
|
logging.basicConfig()
|
|
|
|
|
2024-06-03 20:42:05 +00:00
|
|
|
args = parse_arguments()
|
|
|
|
|
2024-08-25 22:14:12 +00:00
|
|
|
shared_time_stamps_set: set[WorkSample] = set()
|
2024-06-03 21:13:16 +00:00
|
|
|
for repo_path in args.repositories:
|
2024-06-03 22:15:18 +00:00
|
|
|
logger.warning('Visit %s', repo_path)
|
2024-08-25 22:14:12 +00:00
|
|
|
shared_time_stamps_set |= set(
|
2024-08-25 21:41:42 +00:00
|
|
|
git_repo.iterate_samples_from_git_repository(repo_path),
|
|
|
|
)
|
2024-06-03 21:49:42 +00:00
|
|
|
|
2024-08-25 22:57:51 +00:00
|
|
|
shared_time_stamps = sorted(shared_time_stamps_set, key=lambda s: s.end_at)
|
2024-08-25 22:14:12 +00:00
|
|
|
del shared_time_stamps_set
|
|
|
|
|
|
|
|
sample_filter = args.sample_filter
|
|
|
|
if len(sample_filter) != 0:
|
|
|
|
logger.warning('Filtering %s samples', len(shared_time_stamps))
|
|
|
|
shared_time_stamps = filter_samples(shared_time_stamps, sample_filter)
|
|
|
|
logger.warning('Filtered down to %s samples', len(shared_time_stamps))
|
|
|
|
|
|
|
|
logger.warning('Realizing %s samples', len(shared_time_stamps))
|
2024-08-25 22:26:04 +00:00
|
|
|
shared_time_stamps = list(heuristically_realize_samples(shared_time_stamps))
|
2024-06-03 21:49:42 +00:00
|
|
|
|
2024-08-25 22:26:04 +00:00
|
|
|
if args.format_mode == 'cli_report':
|
|
|
|
for t in cli.generate_report(shared_time_stamps):
|
|
|
|
sys.stdout.write(t)
|
|
|
|
elif args.format_mode == 'icalendar':
|
2024-08-25 22:57:51 +00:00
|
|
|
icalendar.generate_icalendar_file(
|
|
|
|
shared_time_stamps, file='./output/samples.ics',
|
|
|
|
)
|