from django.contrib import admin

from apps.investors.models import Beneficiary, Investor, InvestorApplication


@admin.register(InvestorApplication)
class InvestorApplicationAdmin(admin.ModelAdmin):
    list_display = ("email", "investor_type", "status", "full_name", "company_name", "created_at")
    list_filter = ("investor_type", "status")
    search_fields = ("email", "full_name", "company_name")
    readonly_fields = ("created_at", "updated_at")


@admin.register(Investor)
class InvestorAdmin(admin.ModelAdmin):
    list_display = ("account_code", "user", "investor_type", "status", "kyc_status", "country")
    list_filter = ("investor_type", "status", "kyc_status")
    search_fields = ("account_code", "user__email", "company_name")


@admin.register(Beneficiary)
class BeneficiaryAdmin(admin.ModelAdmin):
    list_display = ("display_name", "category", "project", "benefit_type", "status", "is_public")
    list_filter = ("category", "status", "is_public")
    search_fields = ("display_name", "public_label")