from django.contrib import admin
from apps.core.models import (
    User,
    CompanyInfo,
    UserActivation,
    ForgetPassword,
)


@admin.register(ForgetPassword)
class ForgetPasswordAdmin(admin.ModelAdmin):
    pass


@admin.register(User)
class UserAdmin(admin.ModelAdmin):
    list_display = ('id', 'email', 'username', 'is_staff', 'is_active', 'active_company', 'companies_count')
    list_display_links = ('id', 'email', 'username')
    search_fields = ('email', 'username')
    list_filter = ('is_staff', 'is_active', 'active_company')
    list_per_page = 25
    filter_horizontal = ('companies',)
    
    def companies_count(self, obj):
        return obj.companies.count()
    companies_count.short_description = 'Companies Count'


@admin.register(CompanyInfo)
class CompanyInfoAdmin(admin.ModelAdmin):
    pass


@admin.register(UserActivation)
class UserActivationAdmin(admin.ModelAdmin):
    pass