from django.contrib import admin

from .models import Company, CompanyUser


@admin.register(Company)
class CompanyAdmin(admin.ModelAdmin):
    list_display = ('company_name', 'date_created', 'last_updated')
    search_fields = ('company_name',)


@admin.register(CompanyUser)
class CompanyUserAdmin(admin.ModelAdmin):
    list_display = ('user', 'company', 'user_type', 'work_email_address', 'phone_number', 'profile_pic')
    list_filter = ('user_type', 'company')
    search_fields = ('user__username', 'user__email', 'company__company_name', 'work_email_address', 'phone_number')