class DentistAdmin(admin.ModelAdmin):
    list_display = ['dentist_id',
                    'first_name',
                    'last_name',
                    'id',
                     'state',
                     'city',
                     'zipcode_zip',
                    ]

    raw_id_fields = ['dentist_category', 'zipcode']
    # list_filter = ['no_google', 'no_rating', 'no_reviews', 'no_listing', 'is_registered']
    readonly_fields = ['dentist_search_field',]
    search_fields = ['first_name',
                     'last_name',
                     'address',
                     'city_text_zip',
                     'state_text_zip',
                     'zipcode_zip',
                     'zone_id',
                     'email', ]

    fieldsets = (
        ('Personal info', {'fields': ('dentist_id', 'zone_id', 'slug', 'first_name', 'middle_name', 'last_name', 'second_name', 'photo', 'email')}),
        ('Practice', {'fields': ('practice', 'title', 'city', 'county', 'state', 'address', 'address_second')}),
        ('Zip text', {'fields': ('state_text_zip', 'metro_text_zip', 'city_text_zip', 'zipcode_zip')}),
        ('Text', {'fields': ('specialty', 'website', 'description', 'phone', 'fax', 'license')}),
        ('Registered', {'fields': ('is_registered',)}),
        ('Additional info', {'fields': ('num_reviews', 'rank_overall', 'no_reviews', 'no_listing', 'watch', 'view_subscription','change_fee', 'sub_cost_3',
                                        'sub_cost_6', 'sub_cost_12', 'subscription_cost', 'refused_subscription', 'current_subscription', 'non_participating')}),
    )

    def get_search_results(self, request, queryset, search_term):
        queryset, use_distinct = super(DentistAdmin, self).get_search_results(request, queryset, search_term)
        if search_term.isdigit():
            queryset = self.model.objects.filter(Q(id=search_term) | Q(dentist_id=search_term))
        return queryset, use_distinct



class OogleUserAdmin(admin.ModelAdmin):

    form = OogleUserAdminForm

    list_display = ['id', '__unicode__', 'username', 'is_dentist', 'dentist', 'duplicated', 'date_joined', 'login_as', 'is_staff', 'is_superuser', 'is_active',]
    raw_id_fields = ['dentist', 'user_permissions', 'groups', 'login_dentist']
    list_filter = ['duplicated', 'is_dentist', 'role', 'is_superuser', 'is_staff', 'is_active']
    search_fields = ['email', 'username', 'nickname',]

    def __init__(self, *args, **kwargs):
        super(OogleUserAdmin, self).__init__(*args, **kwargs)

    def login_as(self, obj):
        return mark_safe('<a href="/switch_user/?user={}" class="default" type="button" >login as</a>'.format(obj.user_secret_switch_key()))

    def get_search_results(self, request, queryset, search_term):
        queryset, use_distinct = super(OogleUserAdmin, self).get_search_results(request, queryset, search_term)
        if search_term.isdigit():
            queryset = self.model.objects.filter(Q(id=search_term) | Q(dentist__dentist_id=search_term) | Q(login_dentist__dentist_id=search_term))
        return queryset, use_distinct