add label function for feb purpose choices

This commit is contained in:
matthewswspence 2025-03-17 11:09:29 -05:00
parent 4c089ff904
commit bbcd2ce08d
No known key found for this signature in database
GPG key ID: FB458202A7852BA4

View file

@ -59,6 +59,16 @@ class DomainRequest(TimeStampedModel):
REDIRECT = "redirect"
OTHER = "other"
@classmethod
def get_purpose_label(cls, purpose_name: str):
"""Returns the associated label for a given purpose name"""
if purpose_name == cls.WEBSITE:
return "Used for a new website"
elif purpose_name == cls.REDIRECT:
return "Used as a redirect for an existing website"
else:
return "Not for a website"
class StateTerritoryChoices(models.TextChoices):
ALABAMA = "AL", "Alabama (AL)"
ALASKA = "AK", "Alaska (AK)"