20 lines
860 B
Python
20 lines
860 B
Python
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):
|
|
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) |