Skip to content

Commit a94a927

Browse files
committed
Add PROJECT_COMPLETED_STATUSES to account for deleted status
1 parent 4b5a2e1 commit a94a927

File tree

2 files changed

+9
-6
lines changed

2 files changed

+9
-6
lines changed

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

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
from src.endpoints.serializers import (
2929
ASSIGNMENT_PROJECT_VALIDATION_STATUSES,
3030
PROJECT_ACTIVE_STATUSES,
31+
PROJECT_COMPLETED_STATUSES,
3132
serialize_assignment,
3233
serialize_job,
3334
)
@@ -146,9 +147,7 @@ async def list_jobs(
146147
cvat_service.Project.status == cvat_service.ProjectStatuses.canceled
147148
)
148149
case JobStatuses.completed:
149-
query = query.filter(
150-
cvat_service.Project.status == cvat_service.ProjectStatuses.recorded
151-
)
150+
query = query.filter(cvat_service.Project.status.in_(PROJECT_COMPLETED_STATUSES))
152151
case _:
153152
raise NotImplementedError(f"Unsupported status {status}")
154153

packages/examples/cvat/exchange-oracle/src/endpoints/serializers.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,15 @@
1212
from src.schemas import exchange as service_api
1313
from src.utils.assignments import compose_assignment_url, parse_manifest
1414

15-
PROJECT_ACTIVE_STATUSES = [
15+
PROJECT_ACTIVE_STATUSES = {
1616
ProjectStatuses.annotation,
1717
ProjectStatuses.completed,
1818
ProjectStatuses.validation,
19-
]
19+
}
20+
PROJECT_COMPLETED_STATUSES = {
21+
ProjectStatuses.recorded,
22+
ProjectStatuses.deleted,
23+
}
2024

2125

2226
def serialize_job(
@@ -48,7 +52,7 @@ def serialize_job(
4852
api_status = service_api.JobStatuses.canceled
4953
elif project.status in PROJECT_ACTIVE_STATUSES:
5054
api_status = service_api.JobStatuses.active
51-
elif project.status == ProjectStatuses.recorded:
55+
elif project.status in PROJECT_COMPLETED_STATUSES:
5256
api_status = service_api.JobStatuses.completed
5357
else:
5458
raise NotImplementedError(f"Unknown status {project.status}")

0 commit comments

Comments
 (0)