import argparse
import time
import dataclasses
import git
import datetime
from collections.abc import Iterator, Sequence
def parse_arguments():
parser = argparse.ArgumentParser()
parser.add_argument("repo")
return parser.parse_args()
@dataclasses.dataclass
class WorkSample:
registered_at: datetime.datetime
labels: Sequence[str]
def get_samples_from_project(repo: git.Repo) -> Iterator[WorkSample]:
labels = []
labels.append('project:git_time_tracker')
for commit in repo.iter_commits('main'):
assert commit.authored_date == commit.committed_date, 'Not yet supported'
committed_at = datetime.datetime.fromtimestamp(commit.authored_date, tz=datetime.UTC)
yield WorkSample(committed_at, tuple(labels))
def main():
args = parse_arguments()
repo = git.Repo(args.repo)
print(repo)
# TODO: Branch on main or master or default
print(list(get_samples_from_project(repo)))