[Exchnage oracle] Pass job start/stop frame from Ex oracle to Rec oracle

Author: Marishka17

packages/examples/cvat/exchange-oracle/src/core/annotation_meta.py

@@ -12,6 +12,8 @@ class JobMeta(BaseModel):
     annotation_filename: Path
     annotator_wallet_address: str
     assignment_id: str
+    start_frame: int
+    stop_frame: int
 
 
 class AnnotationMeta(BaseModel):

packages/examples/cvat/exchange-oracle/src/crons/cvat/state_trackers.py

@@ -186,6 +186,8 @@ def track_task_creation(logger: logging.Logger, session: Session) -> None:
                         upload.task_id,
                         upload.task.cvat_project_id,
                         status=JobStatuses(cvat_job.state),
+                        start_frame=cvat_job.start_frame,
+                        stop_frame=cvat_job.stop_frame,
                     )
 
                 completed.append(upload)

packages/examples/cvat/exchange-oracle/src/handlers/completed_escrows.py

@@ -75,6 +75,8 @@ def _export_escrow_annotations(
     jobs = cvat_service.get_jobs_by_escrow_address(session, escrow_address, chain_id)
 
     annotation_format = CVAT_EXPORT_FORMAT_MAPPING[manifest.annotation.type]
+    # FUTURE-TODO: probably can be removed in the future since
+    # these annotations are no longer used in Recording Oracle
     job_annotations = _download_job_annotations(logger, annotation_format, jobs)
 
     if manifest.annotation.type == TaskTypes.image_skeletons_from_boxes.value:

packages/examples/cvat/exchange-oracle/src/handlers/cvat_events.py

@@ -123,6 +123,8 @@ def handle_create_job_event(payload: dict) -> None:
                 payload.job["task_id"],
                 payload.job["project_id"],
                 status=JobStatuses[payload.job["state"]],
+                start_frame=payload.job["start_frame"],
+                stop_frame=payload.job["stop_frame"],
             )
             cvat_service.touch(session, models.Job, [job_id])
 

packages/examples/cvat/exchange-oracle/src/handlers/job_creation.py

@@ -354,7 +354,6 @@ def build(self):
             user_guide=manifest.annotation.user_guide,
         )
 
-        # TODO: setup webhook after task is created and GT job is configured
         # Setup webhooks for a project (update:task, update:job)
         cvat_webhook = cvat_api.create_cvat_webhook(cvat_project.id)
 
@@ -401,7 +400,7 @@ def build(self):
                 )
                 db_service.get_task_by_id(session, task_id, for_update=True)  # lock the row
 
-                # The task is fully created once 'update:task' or 'update:job' webhook is received.
+                # The task is fully created once 'update:task' webhook is received.
                 cvat_api.put_task_data(
                     cvat_task.id,
                     cloud_storage.id,
@@ -2640,7 +2639,6 @@ def _task_params_label_key(ts):
                         # TODO: improve guide handling - split for different points
                     )
 
-                    # FIXME: setup webhook after GT job is created
                     # Setup webhooks for a project (update:task, update:job)
                     cvat_webhook = cvat_api.create_cvat_webhook(cvat_project.id)
 
@@ -2683,8 +2681,7 @@ def _task_params_label_key(ts):
                         )
                         db_service.get_task_by_id(session, task_id, for_update=True)  # lock the row
 
-                        # The task is fully created once 'update:task' or 'update:job'
-                        # webhook is received.
+                        # The task is fully created once 'update:task' webhook is received.
                         cvat_api.put_task_data(
                             cvat_task.id,
                             cvat_cloud_storage.id,

packages/examples/cvat/exchange-oracle/src/handlers/job_export.py

@@ -57,6 +57,8 @@ def prepare_annotation_metafile(
                 annotator_wallet_address=job.latest_assignment.user_wallet_address,
                 assignment_id=job.latest_assignment.id,
                 task_id=job.cvat_task_id,
+                start_frame=job.start_frame,
+                stop_frame=job.stop_frame,
             )
             for job in jobs
         ]

packages/examples/cvat/exchange-oracle/src/models/cvat.py

@@ -194,6 +194,8 @@ class Job(ChildOf[Task]):
     status = Column(String, Enum(JobStatuses), nullable=False)
     created_at = Column(DateTime(timezone=True), server_default=func.now())
     updated_at = Column(DateTime(timezone=True), onupdate=func.now())
+    start_frame = Column(Integer)  # TODO: nullable=False
+    stop_frame = Column(Integer)  # TODO: nullable=False
 
     task: Mapped[Task] = relationship(back_populates="jobs")
     project: Mapped[Project] = relationship(back_populates="jobs")