from django.contrib import admin
from flats.models import Building, Floor, Flat


class FloorInline(admin.TabularInline):
    model = Floor
    extra = 1


@admin.register(Building)
class BuildingAdmin(admin.ModelAdmin):
    list_display = ("name", "project", "total_floors")
    list_filter = ("project",)
    inlines = [FloorInline]


@admin.register(Floor)
class FloorAdmin(admin.ModelAdmin):
    list_display = ("building", "floor_name", "floor_number")
    list_filter = ("building",)


@admin.register(Flat)
class FlatAdmin(admin.ModelAdmin):
    list_display = ("__str__", "owner", "share_percent", "size_sqft", "parking_count",
                     "construction_cost", "paid_amount", "due_amount", "status")
    list_filter = ("status", "floor__building")
    search_fields = ("flat_no", "owner__full_name")