Coverage for benefits / core / context / agency.py: 100%

1from dataclasses import dataclass, asdict 

2 

3from django.db import models 

4from django.utils.translation import gettext_lazy as _ 

5 

6 

7class AgencySlug(models.TextChoices): 

8 # raw value, display value 

9 CST = "cst", "cst" 

10 MST = "mst", "mst" 

11 EDCTA = "edcta", "edcta" 

12 NEVCO = "nevco", "nevco" 

13 RABA = "raba", "raba" 

14 SACRT = "sacrt", "sacrt" 

15 SBMTD = "sbmtd", "sbmtd" 

16 SLORTA = "slorta", "slorta" 

17 VCTC = "vctc", "vctc" 

18 

19 

20@dataclass 

21class AgencyIndex: 

22 headline: str 

23 

24 def dict(self): 

25 return asdict(self) 

26 

27 

28agency_index = { 

29 AgencySlug.CST.value: AgencyIndex(headline=_("Get your reduced fare on CST public transit when you tap to ride")), 

30 AgencySlug.EDCTA.value: AgencyIndex(headline=_("Get your reduced fare on EDCTA public transit when you tap to ride")), 

31 AgencySlug.MST.value: AgencyIndex(headline=_("Get your reduced fare on MST public transit when you tap to ride")), 

32 AgencySlug.NEVCO.value: AgencyIndex( 

33 headline=_("Get your reduced fare on Nevada County Connects public transit when you tap to ride") 

34 ), 

35 AgencySlug.RABA.value: AgencyIndex(headline=_("Get your reduced fare on RABA public transit when you tap to ride")), 

36 AgencySlug.SACRT.value: AgencyIndex(headline=_("Get your reduced fare on SacRT buses when you tap to ride")), 

37 AgencySlug.SBMTD.value: AgencyIndex(headline=_("Get your reduced fare on Santa Barbara MTD buses when you tap to ride")), 

38 AgencySlug.SLORTA.value: AgencyIndex( 

39 headline=_("Get your reduced fare on San Luis Obispo RTA buses when you tap to ride") 

40 ), 

41 AgencySlug.VCTC.value: AgencyIndex( 

42 headline=_("Get your reduced fare on Ventura County Transportation Commission buses when you tap to ride") 

43 ), 

44}