from django.contrib import admin 

from core.apps.counterparty.models import Counterparty, CounterpartyFolder, CounterpartyBalance


class CounterpartyBalanceInline(admin.StackedInline):
    model = CounterpartyBalance
    can_delete = False   
    verbose_name_plural = 'Balance'
    fk_name = 'counterparty'   
    extra = 0
    

@admin.register(Counterparty)
class CounterpartyAdmin(admin.ModelAdmin):
    list_display = ['id', 'name', 'phone', 'type', 'inn']
    inlines = [CounterpartyBalanceInline]

    def get_queryset(self, request):
        return super().get_queryset(request).select_related('balance')


@admin.register(CounterpartyFolder)
class CounterpartyFolderAdmin(admin.ModelAdmin):
    list_display = ['id', 'name']
    list_filter = ['name']