admin.py

from django.contrib import admin

from shared.admin import MarkdownxGuardedModelAdmin

from .forms import ContractAdminForm
from .models import (
    Contract,
    Contractee,
    ContracteeRepresentative,
    ContracteeSignature,
    ContractFilingArea,
    ContractIntent,
    ContractIssue,
    ContractSubtype,
    Signee,
    SigneeRepresentative,
    SigneeSignature,
)


class IndexHiddenModelAdmin(MarkdownxGuardedModelAdmin):
    def has_module_permission(self, request):
        return False


class ContractAdmin(MarkdownxGuardedModelAdmin):
    form = ContractAdminForm

    fields = (
        "type",
        "subtype",
        "signee_signature",
        "contractee_signatures",
        "valid_start_date",
        "valid_end_date",
        "legal_state",
        "public_state",
        "paper_form_state",
        "publishing_rejection_comment",
        "tender_url",
        "identifier",
        "issues",
        "notes",
        "summary",
        "anonymized_contract_file",
        "original_contract_file",
        "primary_contract",
        "expected_cost_total",
        "expected_cost_year",
        "expected_cost_month",
        "expected_cost_hour",
        "agreement_url",
        "filing_area",
    )


for model in (
    SigneeRepresentative,
    SigneeSignature,
    ContracteeRepresentative,
    ContracteeSignature,
    ContractSubtype,
    ContractIntent,
):
    admin.site.register(model, IndexHiddenModelAdmin)

for model in (
    Signee,
    Contractee,
    ContractIssue,
    ContractFilingArea,
):
    admin.site.register(model, MarkdownxGuardedModelAdmin)

admin.site.register(Contract, ContractAdmin)