71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
from django.utils.translation import gettext_lazy as _
|
|
|
|
from cms.plugin_base import CMSPluginBase
|
|
from cms.plugin_pool import plugin_pool
|
|
|
|
from .models import (
|
|
PLUGIN_TEMPLATES,
|
|
GovernmentPlansCMSPlugin,
|
|
GovernmentPlanSection,
|
|
GovernmentPlanSectionsCMSPlugin,
|
|
GovernmentPlanUpdatesCMSPlugin,
|
|
PlanStatus,
|
|
)
|
|
|
|
|
|
@plugin_pool.register_plugin
|
|
class GovernmentPlansPlugin(CMSPluginBase):
|
|
name = _("Government plans")
|
|
model = GovernmentPlansCMSPlugin
|
|
filter_horizontal = ("categories",)
|
|
cache = True
|
|
|
|
def get_render_template(self, context, instance, placeholder):
|
|
return instance.template or PLUGIN_TEMPLATES[0][0]
|
|
|
|
def render(self, context, instance, placeholder):
|
|
context = super().render(context, instance, placeholder)
|
|
context["plugin"] = instance
|
|
context["status_list"] = PlanStatus.choices
|
|
context["object_list"] = instance.get_plans(
|
|
context["request"], published_only=False
|
|
)
|
|
return context
|
|
|
|
|
|
@plugin_pool.register_plugin
|
|
class GovernmentPlanSectionsPlugin(CMSPluginBase):
|
|
name = _("Government plan sections")
|
|
model = GovernmentPlanSectionsCMSPlugin
|
|
render_template = "froide_govplan/plugins/sections.html"
|
|
|
|
def render(self, context, instance, placeholder):
|
|
context = super().render(context, instance, placeholder)
|
|
|
|
if instance.government_id:
|
|
sections = GovernmentPlanSection.objects.filter(
|
|
government_id=instance.government_id
|
|
)
|
|
else:
|
|
sections = GovernmentPlanSection.objects.all()
|
|
|
|
sections = sections.select_related("government")
|
|
|
|
context["sections"] = sections
|
|
|
|
return context
|
|
|
|
|
|
@plugin_pool.register_plugin
|
|
class GovernmentPlanUpdatesPlugin(CMSPluginBase):
|
|
name = _("Government plan updates")
|
|
model = GovernmentPlanUpdatesCMSPlugin
|
|
render_template = "froide_govplan/plugins/updates.html"
|
|
|
|
def render(self, context, instance, placeholder):
|
|
context = super().render(context, instance, placeholder)
|
|
context["updates"] = instance.get_updates(
|
|
context["request"], published_only=False
|
|
)
|
|
context["show_context"] = True
|
|
return context
|