from django.contrib import admin
from home import models


@admin.register(models.CompanyProfile)
class CompanyProfileAdmin(admin.ModelAdmin):
    list_display = ['title', 'abbreviation']


@admin.register(models.Profile)
class ProfileAdmin(admin.ModelAdmin):
    list_display = ('user', 'role', 'profile_image')
    search_fields = ('user__username', 'user__first_name', 'user__last_name', 'role')
    list_filter = ('role',)

    def profile_image(self, obj):
        if obj.image:
            return f"<img src='{obj.image.url}' width='40' height='40' style='border-radius: 50%; object-fit: cover;' />"
        return "-"
    profile_image.allow_tags = True
    profile_image.short_description = "Image"