zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-05-19 10:59:21 +02:00
Code Issues Projects Releases Packages Wiki Activity

Migrations, etc

Browse source
This commit is contained in:
zandercymatics 2024-04-30 10:52:53 -06:00
parent f52a7e08fe
commit 5e2ec5d1c8
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
8 changed files with 103 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/registrar/admin.py
View file

@ -15,7 +15,9 @@ from django.contrib.contenttypes.models import ContentType
from django.urls import reverse
from dateutil.relativedelta import relativedelta # type: ignore
from epplibwrapper.errors import ErrorCode, RegistryError
from waffle.admin import FlagAdmin
from registrar.models import Contact, Domain, DomainRequest, DraftDomain, User, Website
from waffle.models import Sample, Switch
from registrar.utility.errors import FSMDomainRequestError, FSMErrorCodes
from registrar.views.utility.mixins import OrderableFieldsMixin
from django.contrib.admin.views.main import ORDER_VAR
@ -2157,8 +2159,19 @@ class UserGroupAdmin(AuditedAdmin):
def user_group(self, obj):
return obj.name
class WaffleFlagAdmin(FlagAdmin):
class Meta:
"""Contains meta information about this class"""
model = models.WaffleFlag
fields = "__all__"
admin.site.unregister(LogEntry) # Unregister the default registration
# Unregister samples and switches from django-waffle, as we currently don't use these.
# TODO - address this
admin.site.unregister(Sample)
admin.site.unregister(Switch)
admin.site.register(LogEntry, CustomLogEntryAdmin)
admin.site.register(models.User, MyUserAdmin)
# Unregister the built-in Group model
@ -2180,3 +2193,6 @@ admin.site.register(models.PublicContact, PublicContactAdmin)
admin.site.register(models.DomainRequest, DomainRequestAdmin)
admin.site.register(models.TransitionDomain, TransitionDomainAdmin)
admin.site.register(models.VerifiedByStaff, VerifiedByStaffAdmin)
# Register our custom waffle flag implementation
admin.site.register(models.WaffleFlag, WaffleFlagAdmin)