43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
from camps.mixins import CampViewMixin
|
|
from django.shortcuts import get_object_or_404
|
|
|
|
from .models import Facility, FacilityType
|
|
|
|
|
|
class FacilityTypeViewMixin(CampViewMixin):
|
|
"""
|
|
A mixin to get the FacilityType object based on facility_type_slug in url kwargs
|
|
"""
|
|
|
|
def setup(self, *args, **kwargs):
|
|
super().setup(*args, **kwargs)
|
|
self.facility_type = get_object_or_404(
|
|
FacilityType,
|
|
responsible_team__camp=self.camp,
|
|
slug=self.kwargs["facility_type_slug"],
|
|
)
|
|
|
|
def get_context_data(self, *args, **kwargs):
|
|
context = super().get_context_data(*args, **kwargs)
|
|
context["facilitytype"] = self.facility_type
|
|
return context
|
|
|
|
|
|
class FacilityViewMixin(FacilityTypeViewMixin):
|
|
"""
|
|
A mixin to get the Facility object based on facility_uuid in url kwargs
|
|
"""
|
|
|
|
def setup(self, *args, **kwargs):
|
|
super().setup(*args, **kwargs)
|
|
self.facility = get_object_or_404(
|
|
Facility,
|
|
facility_type=self.facility_type,
|
|
uuid=self.kwargs["facility_uuid"],
|
|
)
|
|
|
|
def get_context_data(self, *args, **kwargs):
|
|
context = super().get_context_data(*args, **kwargs)
|
|
context["facility"] = self.facility
|
|
return context
|