from django.contrib import admin

from .models import Company, CompanyFollow


@admin.register(CompanyFollow)
class CompanyFollowAdmin(admin.ModelAdmin):
    list_display = ('user', 'company', 'created_at')
    list_filter = ('created_at',)
    search_fields = ('user__email', 'user__username', 'company__name')
    autocomplete_fields = ('user', 'company')


@admin.register(Company)
class CompanyAdmin(admin.ModelAdmin):
    list_display = ('name', 'industry', 'location', 'tier', 'is_public', 'is_online', 'is_featured', 'is_active')
    list_filter = ('tier', 'industry', 'is_public', 'is_online', 'is_featured', 'is_active')
    search_fields = ('name', 'industry', 'location')
    prepopulated_fields = {'slug': ('name',)}