from django.contrib import admin
from django.contrib.auth.admin import UserAdmin as BaseUserAdmin

from main.models import Pill, Pharmacy, Company
from main.models import User


class UserAdmin(BaseUserAdmin):
    list_display = ('first_name', 'last_name', 'email', 'phone', 'company')
    fieldsets = (
        (None, {'fields': ('first_name', 'password',)}),
        ('Personal info', {'fields': ('first_name', 'last_name', 'email', 'company',)}),
        ('Permissions', {'fields': ('is_active', 'is_staff', 'is_superuser', 'groups', 'user_permissions',)}),
        ('MyField', {'fields': ('photo', 'phone',)}),
    )
    add_fieldsets = (
        (None, {
            'classes': ('wide',),
            'fields': ('phone', 'email', 'first_name', 'last_name',)}
         ),
    )
    search_fields = ('id', 'phone', 'email', 'first_name', 'last_name',)
    ordering = ('last_name', 'first_name',)

    class Meta:
        model = User


admin.site.register(User, UserAdmin)