added organizationtype to enum

This commit is contained in:
Beverly Nguyen
2025-07-15 23:01:12 -07:00
parent e9f6b93d85
commit 56175bd7aa
4 changed files with 16 additions and 13 deletions

View File

@@ -2,14 +2,15 @@ from collections import OrderedDict
from werkzeug.utils import cached_property
from app.enums import OrganizationType
from app.models import JSONModel, ModelList, SerialisedModelCollection, SortByNameMixin
from app.notify_client.organizations_api_client import organizations_client
class Organization(JSONModel, SortByNameMixin):
TYPE_FEDERAL = "federal"
TYPE_STATE = "state"
TYPE_OTHER = "other"
TYPE_FEDERAL = OrganizationType.FEDERAL
TYPE_STATE = OrganizationType.STATE
TYPE_OTHER = OrganizationType.OTHER
TYPE_LABELS = OrderedDict(
[

View File

@@ -322,7 +322,7 @@ class Service(JSONModel, SortByNameMixin):
@property
def shouldnt_use_govuk_as_sms_sender(self):
return self.organization_type != Organization.TYPE_CENTRAL
return self.organization_type != Organization.TYPE_FEDERAL
@cached_property
def sms_senders(self):