1
0
This commit is contained in:
Jon Michael Aanes 2024-08-26 00:57:51 +02:00
parent 7ab46bc48e
commit a1c71aa1b5
Signed by: Jmaa
SSH Key Fingerprint: SHA256:Ab0GfHGCblESJx7JRE4fj4bFy/KRpeLhi41y4pF3sNA
4 changed files with 36 additions and 25 deletions

View File

@ -32,7 +32,12 @@ import sys
from collections.abc import Iterator
from pathlib import Path
from .data import HIDDEN_LABEL_PREFIX, HIDDEN_LABEL_TOTAL, WorkSample, RealizedWorkSample
from .data import (
HIDDEN_LABEL_PREFIX,
HIDDEN_LABEL_TOTAL,
RealizedWorkSample,
WorkSample,
)
from .format import cli, icalendar
from .source import git_repo
@ -45,11 +50,16 @@ HOUR = datetime.timedelta(hours=1)
MINUTE = datetime.timedelta(minutes=1)
def filter_samples(samples: list[WorkSample], sample_filter: set[str]) -> list[WorkSample]:
def filter_samples(
samples: list[WorkSample], sample_filter: set[str],
) -> list[WorkSample]:
assert len(sample_filter) > 0
return [s for s in samples if set(s.labels).intersection(sample_filter)]
def heuristically_realize_samples(samples: list[WorkSample]) -> Iterator[RealizedWorkSample]:
def heuristically_realize_samples(
samples: list[WorkSample],
) -> Iterator[RealizedWorkSample]:
"""Secret sauce.
Guarentees that:
@ -75,6 +85,7 @@ def heuristically_realize_samples(samples: list[WorkSample]) -> Iterator[Realize
previous_sample_end = sample.end_at
del sample
def parse_arguments():
parser = argparse.ArgumentParser()
parser.add_argument(
@ -131,4 +142,6 @@ def main():
for t in cli.generate_report(shared_time_stamps):
sys.stdout.write(t)
elif args.format_mode == 'icalendar':
icalendar.generate_icalendar_file(shared_time_stamps, file='./output/samples.ics')
icalendar.generate_icalendar_file(
shared_time_stamps, file='./output/samples.ics',
)

View File

@ -12,6 +12,7 @@ class WorkSample:
start_at: datetime.datetime | None
end_at: datetime.datetime | None
@dataclasses.dataclass(frozen=True, order=True)
class RealizedWorkSample(WorkSample):
start_at: datetime.datetime

View File

@ -1,15 +1,8 @@
import datetime
from collections.abc import Iterator
import argparse
import datetime
import urllib.parse
import icalendar
from personal_data.util import load_csv_file
from ..data import HIDDEN_LABEL_PREFIX, HIDDEN_LABEL_TOTAL, RealizedWorkSample
from ..data import HIDDEN_LABEL_PREFIX, RealizedWorkSample
ZERO_DURATION = datetime.timedelta(seconds=0)
HOUR = datetime.timedelta(hours=1)
@ -37,7 +30,6 @@ def generate_calendar(
cal.add('version', '2.0')
for sample in samples:
title = create_title(sample)
description = ''
@ -52,7 +44,9 @@ def generate_calendar(
for label_and_type in sample.labels:
if label_and_type.startswith('author:'):
event.add('organizer', 'mailto:'+label_and_type.removeprefix('author:'))
event.add(
'organizer', 'mailto:' + label_and_type.removeprefix('author:'),
)
cal.add_component(event)
del event
@ -64,7 +58,6 @@ def generate_icalendar_file(
samples: list[RealizedWorkSample],
file: str,
) -> None:
calendar = generate_calendar(samples)
with open(file, 'wb') as f:

View File

@ -38,8 +38,12 @@ def get_samples_from_project(repo: git.Repo) -> Iterator[WorkSample]:
labels.append('project:' + project_name)
labels.append('author:' + commit.author.email)
authored_date = datetime.datetime.fromtimestamp(commit.authored_date, tz=datetime.UTC)
committed_date = datetime.datetime.fromtimestamp(commit.committed_date, tz=datetime.UTC)
authored_date = datetime.datetime.fromtimestamp(
commit.authored_date, tz=datetime.UTC,
)
committed_date = datetime.datetime.fromtimestamp(
commit.committed_date, tz=datetime.UTC,
)
yield WorkSample(
labels=tuple(labels),