"""Subscription Django admin registrations (Unfold)."""
from __future__ import annotations
from typing import ClassVar
from django import forms
from django.contrib import admin
from django.utils.html import format_html
from django.utils.safestring import mark_safe
from django.utils.translation import gettext_lazy as _
try:
from unfold.admin import ModelAdmin as UnfoldModelAdmin
from unfold.admin import TabularInline as UnfoldTabularInline
from unfold.decorators import display
except ImportError:
UnfoldModelAdmin = admin.ModelAdmin # type: ignore[assignment]
UnfoldTabularInline = admin.TabularInline # type: ignore[assignment]
def display(**kwargs): # type: ignore[return-value]
def decorator(fn):
return fn
return decorator
from simorgh.apps.subscription.models import (
Addon,
PlanFeature,
PlatformPlan,
TenantAddonFeature,
TenantSubscription,
)
# Aliases for Phase B
Plan = PlatformPlan
PlanEntitlement = PlanFeature
# ---------------------------------------------------------------------------
# Feature tree widget — hierarchical checkbox selection
# ---------------------------------------------------------------------------
def _build_feature_tree() -> list[dict]:
"""Build Domain → Capability → Feature tree from the catalog."""
from simorgh.apps.modules.models import Capability, Domain, FeatureCatalog
tree: list[dict] = []
domains = Domain.objects.prefetch_related("capabilities__features").order_by("sort_order", "code")
for domain in domains:
capabilities: list[dict] = []
for cap in domain.capabilities.order_by("name"):
features = list(
cap.features.filter(is_always_on=False, is_platform_internal=False)
.order_by("code")
.values_list("code", "label")
)
if features:
capabilities.append({
"name": cap.name,
"slug": cap.slug,
"features": [{"code": f[0], "label": f[1] or f[0]} for f in features],
})
if capabilities:
tree.append({
"id": f"domain-{domain.code}",
"name": domain.name,
"code": domain.code,
"capabilities": capabilities,
})
return tree
class FeatureTreeWidget(forms.CheckboxSelectMultiple):
"""Renders features as a hierarchical checkbox tree: Domain → Capability → Feature."""
template_name = "django/forms/widgets/checkbox_select.html"
option_template_name = "django/forms/widgets/checkbox_option.html"
def render(self, name, value, attrs=None, renderer=None):
"""Render the tree with cascading JavaScript."""
if value is None:
value = []
tree = _build_feature_tree()
html_parts = ['
']
for domain in tree:
domain_id = domain["id"]
html_parts.append(f'