diff options
Diffstat (limited to 'ctrack/organisations/views.py')
| -rw-r--r-- | ctrack/organisations/views.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/ctrack/organisations/views.py b/ctrack/organisations/views.py index 45a86bb..82347e6 100644 --- a/ctrack/organisations/views.py +++ b/ctrack/organisations/views.py @@ -145,7 +145,10 @@ class IncidentReportCreateView(FormView): return HttpResponseRedirect(reverse("core:home")) -def inspectors_for_each_mode(lead_type="lead_inspector") -> Dict[str, List[str]]: +def inspectors_for_each_mode(lead_type="lead_inspector") -> Dict[str, Set[str]]: + """ + We want to be able to group lead inspectors by submode. + """ if lead_type not in ["lead_inspector", "deputy_lead_inspector"]: raise ValueError("Can only query for lead_inspector and deputy_lead_inspector attributes.") submodes = Submode.objects.all() @@ -155,6 +158,6 @@ def inspectors_for_each_mode(lead_type="lead_inspector") -> Dict[str, List[str]] orgs = sm.organisation_set.all() for org in orgs: insp.add(getattr(org, lead_type)) - out[sm.descriptor] = list(insp) + out[sm.descriptor] = insp del insp return out |