from django.contrib import admin
from django.contrib.auth.admin import UserAdmin
from .models import Gateway, User, Config


class CustomUserAdmin(UserAdmin):
    fieldsets = UserAdmin.fieldsets + (
        (None, {'fields': ('custom_connection_gateway', 'custom_connection_gateway_token')}),
    )


admin.site.register(User, CustomUserAdmin)
admin.site.register(Config)
admin.site.register(Gateway)