"""
Admin registrations
"""
from django.contrib import admin
from django.contrib.auth.models import Group

from ..models import User, Note, Session, Protocol, Category
from .auth import UserAdmin
from .auth import GroupAdmin


# Re-register UserAdmin
admin.site.register(User, UserAdmin)

# Re-register GroupAdmin
admin.site.unregister(Group)
admin.site.register(Group, GroupAdmin)

admin.site.register(Note)
admin.site.register(Session)
admin.site.register(Protocol)
admin.site.register(Category)