website/hub/services/admin.py

from django.contrib import admin
from django.utils.html import format_html
from .models import (
    Category,
    CloudProvider,
    ConsultingPartner,
    Currency,
    ExternalLink,
    Plan,
    PlanPrice,
    Service,
    ServiceOffering,
    Term,
    Lead,
)


@admin.register(Category)
class CategoryAdmin(admin.ModelAdmin):
    list_display = ("name", "slug", "parent", "order")
    list_filter = ("parent",)
    search_fields = ("name", "description")
    prepopulated_fields = {"slug": ("name",)}
    ordering = ("order", "name")


@admin.register(Currency)
class CurrencyAdmin(admin.ModelAdmin):
    list_display = ("code", "name", "symbol")
    search_fields = ("code", "name")
    ordering = ("code",)


@admin.register(Term)
class TermAdmin(admin.ModelAdmin):
    list_display = ("name", "order")
    ordering = ("order", "name")
    search_fields = ("name", "description")


@admin.register(CloudProvider)
class CloudProviderAdmin(admin.ModelAdmin):
    list_display = ("name", "slug", "logo_preview")
    search_fields = ("name", "description")
    prepopulated_fields = {"slug": ("name",)}

    def logo_preview(self, obj):
        if obj.logo:
            return format_html(
                '<img src="{}" style="max-height: 50px;"/>', obj.logo.url
            )
        return "No logo"

    logo_preview.short_description = "Logo"


class ExternalLinkInline(admin.TabularInline):
    model = ExternalLink
    extra = 1
    fields = ("description", "url", "order")
    ordering = ("order", "description")


@admin.register(Service)
class ServiceAdmin(admin.ModelAdmin):
    list_display = ("name", "logo_preview", "category_list", "partner_list")
    list_filter = ("categories",)
    search_fields = ("name", "description", "slug")
    prepopulated_fields = {"slug": ("name",)}
    filter_horizontal = ("categories",)
    inlines = [ExternalLinkInline]

    def logo_preview(self, obj):
        if obj.logo:
            return format_html(
                '<img src="{}" style="max-height: 50px;"/>', obj.logo.url
            )
        return "No logo"

    logo_preview.short_description = "Logo"

    def category_list(self, obj):
        return ", ".join([cat.name for cat in obj.categories.all()])

    category_list.short_description = "Categories"

    def partner_list(self, obj):
        return ", ".join([partner.name for partner in obj.consulting_partners.all()])

    partner_list.short_description = "Consulting Partners"


@admin.register(PlanPrice)
class PlanPriceAdmin(admin.ModelAdmin):
    model = PlanPrice


class PlanInline(admin.StackedInline):
    model = Plan
    extra = 1
    fieldsets = (
        (None, {"fields": ("name", "description", "is_default", "features", "order")}),
        (
            "Pricing",
            {
                "fields": ("get_prices",),
            },
        ),
    )
    readonly_fields = ("get_prices",)

    def get_prices(self, obj):
        if not obj:
            return "Save the plan first to add prices"

        html = ['<table class="price-table">']
        html.append(
            "<tr><th>Currency</th><th>Term</th><th>Price</th><th>Actions</th></tr>"
        )

        for price in obj.prices.all():
            html.append(
                f"""
                <tr>
                    <td>{price.currency}</td>
                    <td>{price.term}</td>
                    <td>{price.price}</td>
                    <td>
                        <a href="/admin/services/planprice/{price.id}/change/" class="button" target="_blank">Edit</a>
                    </td>
                </tr>
            """
            )

        html.append("</table>")
        html.append(
            f'<a href="/admin/services/planprice/add/?plan={obj.id}" class="button" target="_blank">Add Price</a>'
        )
        return format_html("".join(html))

    get_prices.short_description = "Plan Prices"


@admin.register(ServiceOffering)
class ServiceOfferingAdmin(admin.ModelAdmin):
    list_display = ("service", "cloud_provider", "status")
    list_filter = ("service", "cloud_provider")
    search_fields = ("service__name", "cloud_provider__name", "description")
    inlines = [PlanInline]

    class Media:
        css = {"all": ("admin/css/service_offering.css",)}


@admin.register(ConsultingPartner)
class ConsultingPartnerAdmin(admin.ModelAdmin):
    list_display = ("name", "website", "logo_preview")
    search_fields = ("name", "description")
    prepopulated_fields = {"slug": ("name",)}
    filter_horizontal = ("services", "cloud_providers")

    def logo_preview(self, obj):
        if obj.logo:
            return format_html(
                '<img src="{}" style="max-height: 50px;"/>', obj.logo.url
            )
        return "No logo"

    logo_preview.short_description = "Logo"


@admin.register(Lead)
class LeadAdmin(admin.ModelAdmin):
    list_display = ("name", "company", "created_at", "odoo_lead_id")
    search_fields = ("name", "company")