from django.contrib import admin
from .models import UserModel, Location


class UserModelAdmin(admin.ModelAdmin):
    list_display = ['user', 'refresh_token', 'gmb_account_id']


class LocationsAdmin(admin.ModelAdmin):
    list_display = ['location_id','care_name', 'location_name', 'website_url', 'display_name']


admin.site.register(Location, LocationsAdmin)

admin.site.register(UserModel, UserModelAdmin)