from django.contrib import admin
from . import models
from safesite import models as s_models
# Register your models here.
# admin.site.register(models.GroupUser)


@admin.register(models.Group)
class GroupAdmin(admin.ModelAdmin):
    def formfield_for_manytomany(self, db_field, request, **kwargs):
        if db_field.name == "members":
            kwargs["queryset"] = s_models.Partment.objects.filter(iscompany=1,deletemark=1)
        return super().formfield_for_manytomany(db_field, request, **kwargs)

@admin.register(models.Groupmember)
class GroupmemberAdmin(admin.ModelAdmin):
    list_display=['group','member']
    def formfield_for_foreignkey(self, db_field, request, **kwargs):
        if db_field.name == "member":
            kwargs["queryset"] = s_models.Partment.objects.filter(iscompany=1,deletemark=1)
        return super().formfield_for_foreignkey(db_field, request, **kwargs)

@admin.register(models.GroupUser)
class GroupuserAdmin(admin.ModelAdmin):
    list_display=['username','group']
    exclude = ('user',)