from django.contrib import admin

from apps.investments.models import Investment, InvestmentAllocation, InvestmentCertificate


class AllocationInline(admin.TabularInline):
    model = InvestmentAllocation
    extra = 0


@admin.register(Investment)
class InvestmentAdmin(admin.ModelAdmin):
    list_display = ("reference", "agreement_code", "investor", "project", "status", "committed_amount", "participation_percent")
    list_filter = ("status", "currency", "source")
    search_fields = ("reference", "agreement_code", "investor__account_code", "investor__user__email")
    inlines = [AllocationInline]


@admin.register(InvestmentCertificate)
class InvestmentCertificateAdmin(admin.ModelAdmin):
    list_display = ("certificate_number", "investment", "issued_at")