zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-08-06 01:35:22 +02:00
Code Issues Projects Releases Packages Wiki Activity

updated domain json query to handle expired logic

Browse source
This commit is contained in:
David Kennedy 2024-06-20 15:01:27 -04:00
parent 8e50ac82f2
commit b263c7e144
No known key found for this signature in database
GPG key ID: 6528A5386E66B96B
1 changed files with 22 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

23
src/registrar/views/domains_json.py
View file

@ -1,3 +1,4 @@
from datetime import timezone
from django.http import JsonResponse
from django.core.paginator import Paginator
from registrar.models import UserDomainRole, Domain
@ -30,7 +31,27 @@ def get_domains_json(request):
status_param = request.GET.get("status")
if status_param:
status_list = status_param.split(',')
objects = objects.filter(state__in=status_list)
# Split the status list into normal states and custom states
normal_states = [state for state in status_list if state in Domain.State.values]
custom_states = [state for state in status_list if state == 'expired']
# Construct Q objects for normal states that can be queried through ORM
state_query = Q()
if normal_states:
state_query |= Q(state__in=normal_states)
# Handle custom states in Python, as expired can not be queried through ORM
if 'expired' in custom_states:
expired_domains = [domain.id for domain in objects if domain.state_display() == 'Expired']
state_query |= Q(id__in=expired_domains)
# Apply the combined query
objects = objects.filter(state_query)
# NOTE: when a domain has a state of 'ready' and is_expired(), a search for
# status=ready will include the domain, even though the domain's state_display
# is Expired.
if sort_by == "state_display":
# Fetch the objects and sort them in Python