from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from django.contrib.auth.models import User


class TheStoreAdmin(admin.ModelAdmin):
    empty_value_display = '---'
    exclude = ['created_by', 'modified_by', ]

    def save_model(self, request, obj, form, change):
        if not obj.pk:
            obj.created_by = request.user
        else:
            obj.modified_by = request.user

        obj.save()


UserAdmin.list_display = ['email', 'first_name', 'last_name', 'is_active', 'is_staff', ]
UserAdmin.list_display_links = ['email', ]

admin.site.unregister(User)
admin.site.register(User, UserAdmin)