Add DomainGroup and Suborg models

This commit is contained in:
Erin 2024-06-20 14:19:59 -07:00
parent fa502151a2
commit 2995792375
No known key found for this signature in database
GPG key ID: 1CAD275313C62460
6 changed files with 129 additions and 0 deletions

View file

@ -2655,6 +2655,27 @@ class WaffleFlagAdmin(FlagAdmin):
model = models.WaffleFlag
fields = "__all__"
class DomainGroupResource(resources.ModelResource):
"""defines how each field in the referenced model should be mapped to the corresponding fields in the
import/export file"""
class Meta:
model = models.DomainGroup
class DomainGroupAdmin(ListHeaderAdmin, ImportExportModelAdmin):
resource_classes = [DomainGroupResource]
list_display = ["name", "portfolio"]
class SuborganizationResource(resources.ModelResource):
"""defines how each field in the referenced model should be mapped to the corresponding fields in the
import/export file"""
class Meta:
model = models.Suborganization
class SuborganizationAdmin(ListHeaderAdmin, ImportExportModelAdmin):
resource_classes = [SuborganizationResource]
list_display = ["name", "portfolio"]
admin.site.unregister(LogEntry) # Unregister the default registration
@ -2679,6 +2700,8 @@ admin.site.register(models.DomainRequest, DomainRequestAdmin)
admin.site.register(models.TransitionDomain, TransitionDomainAdmin)
admin.site.register(models.VerifiedByStaff, VerifiedByStaffAdmin)
admin.site.register(models.Portfolio, PortfolioAdmin)
admin.site.register(models.DomainGroup, DomainGroupAdmin)
admin.site.register(models.Suborganization, SuborganizationAdmin)
# Register our custom waffle implementations
admin.site.register(models.WaffleFlag, WaffleFlagAdmin)