38 lines
951 B
Python
38 lines
951 B
Python
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)))
|