blob: 6cc973a9d06534dc86cab581fb0a953419d2ba2f (
plain) (
tree)
|
|
from typing import Dict, Set
from django.contrib.auth.models import User
from django.db.models import QuerySet, Q
from ctrack.organisations.models import Submode
def filter_private_events(events: QuerySet, user: User):
"""
Given a QuerySet containing SingleDateTimeEvent objects,
ensure that any objects whose user==user and private==True
are filtered out. This supports OrganisationDetailView, which
lists all events for an organisation but which must hide private
events for the logged-in user.
"""
return events.exclude(~Q(user=user) & Q(private=True))
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()
out = {}
for sm in submodes:
insp = set()
orgs = sm.organisation_set.all()
for org in orgs:
insp.add(getattr(org, lead_type))
insp = {x for x in insp if x is not None}
out[sm.descriptor] = insp
del insp
return out
|
