Add submitted at and started at to job dataclass

Author: lukasbindreiter

tilebox-workflows/tests/tasks_data.py

@@ -4,15 +4,17 @@
 
 import json
 import string
-from datetime import timedelta
+from datetime import datetime, timedelta
 
 from hypothesis.strategies import (
     DrawFn,
     booleans,
     composite,
+    datetimes,
     dictionaries,
     floats,
     integers,
+    just,
     lists,
     none,
     one_of,
@@ -123,12 +125,16 @@ def jobs(draw: DrawFn, canceled: bool | None = None) -> Job:
     name = draw(alphanumerical_text())
     trace_parent = draw(alphanumerical_text())
     state = draw(sampled_from(JobState))
+    submitted_at = draw(datetimes(min_value=datetime(1990, 1, 1), max_value=datetime(2024, 1, 1), timezones=just(None)))
+    started_at = draw(
+        one_of(none(), datetimes(min_value=submitted_at, max_value=datetime(2025, 1, 1), timezones=just(None)))
+    )
     if canceled is None:
         canceled = draw(booleans())
 
     progress = draw(lists(progress_bars(), min_size=0, max_size=3))
 
-    return Job(job_id, name, trace_parent, state, canceled, progress)
+    return Job(job_id, name, trace_parent, state, submitted_at, started_at, canceled, progress)
 
 
 @composite

tilebox-workflows/tilebox/workflows/data.py

@@ -201,6 +201,8 @@ class Job:
     name: str
     trace_parent: str
     state: JobState
+    submitted_at: datetime
+    started_at: datetime | None
     canceled: bool
     progress_bars: list[ProgressBar]
 
@@ -212,6 +214,8 @@ def from_message(cls, job: core_pb2.Job) -> "Job":  # lets use typing.Self once
             name=job.name,
             trace_parent=job.trace_parent,
             state=_JOB_STATES[job.state],
+            submitted_at=timestamp_to_datetime(job.submitted_at),
+            started_at=timestamp_to_datetime(job.started_at) if job.started_at else None,
             canceled=job.canceled,
             progress_bars=[ProgressBar.from_message(progress_bar) for progress_bar in job.progress_bars],
         )
@@ -223,6 +227,8 @@ def to_message(self) -> core_pb2.Job:
             name=self.name,
             trace_parent=self.trace_parent,
             state=f"JOB_STATE_{self.state.name}",
+            submitted_at=datetime_to_timestamp(self.submitted_at),
+            started_at=datetime_to_timestamp(self.started_at) if self.started_at else None,
             canceled=self.canceled,
             progress_bars=[progress_bar.to_message() for progress_bar in self.progress_bars],
         )