Using StrEnum to drop .value

This commit is contained in:
Beverly Nguyen
2025-07-04 17:22:21 -07:00
parent b73b4ac73e
commit 8a41de8f07
6 changed files with 37 additions and 37 deletions

View File

@@ -2,9 +2,9 @@ from app.notify_client import NotifyAdminAPIClient, _attach_current_user
from app.enums import ApiKeyType
# must match key types in notifications-api/app/models.py
KEY_TYPE_NORMAL = ApiKeyType.NORMAL.value
KEY_TYPE_TEAM = ApiKeyType.TEAM.value
KEY_TYPE_TEST = ApiKeyType.TEST.value
KEY_TYPE_NORMAL = ApiKeyType.NORMAL
KEY_TYPE_TEAM = ApiKeyType.TEAM
KEY_TYPE_TEST = ApiKeyType.TEST
class ApiKeyApiClient(NotifyAdminAPIClient):

View File

@@ -9,17 +9,17 @@ from app.enums import JobStatus
class JobApiClient(NotifyAdminAPIClient):
JOB_STATUSES = {
JobStatus.SCHEDULED.value,
JobStatus.PENDING.value,
JobStatus.IN_PROGRESS.value,
JobStatus.FINISHED.value,
JobStatus.CANCELLED.value,
JobStatus.SENDING_LIMITS_EXCEEDED.value,
JobStatus.READY_TO_SEND.value,
JobStatus.SENT_TO_DVLA.value,
JobStatus.SCHEDULED,
JobStatus.PENDING,
JobStatus.IN_PROGRESS,
JobStatus.FINISHED,
JobStatus.CANCELLED,
JobStatus.SENDING_LIMITS_EXCEEDED,
JobStatus.READY_TO_SEND,
JobStatus.SENT_TO_DVLA,
}
SCHEDULED_JOB_STATUS = JobStatus.SCHEDULED.value
CANCELLED_JOB_STATUS = JobStatus.CANCELLED.value
SCHEDULED_JOB_STATUS = JobStatus.SCHEDULED
CANCELLED_JOB_STATUS = JobStatus.CANCELLED
NON_CANCELLED_JOB_STATUSES = JOB_STATUSES - {CANCELLED_JOB_STATUS}
NON_SCHEDULED_JOB_STATUSES = JOB_STATUSES - {
SCHEDULED_JOB_STATUS,