from django.contrib import admin
from .models import CustomGroup, GroupSelect
from django.contrib.auth.models import User


class CustomGroupAdmin(admin.ModelAdmin):
    fields = ('name', 'group_users', 'admin', 'share', 'users_delete')
    list_display = (
        'name',
        'admin'
    )

admin.site.register(CustomGroup, CustomGroupAdmin)

class GroupSelectAdmin(admin.ModelAdmin):
    fields = ('user', 'group')

admin.site.register(GroupSelect, GroupSelectAdmin)