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 VCTC = "vctc", "vctc" 

17 

18 

19@dataclass 

20class AgencyIndex: 

21 headline: str 

22 

23 def dict(self): 

24 return asdict(self) 

25 

26 

27agency_index = { 

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

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

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

31 AgencySlug.NEVCO.value: AgencyIndex( 

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

33 ), 

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

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

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

37 AgencySlug.VCTC.value: AgencyIndex( 

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

39 ), 

40}