altf4arnold/datatracker
1
0
Fork 0
Code Issues Pull requests Actions 3 Packages Projects Releases Wiki Activity

test: Use QuerySetAny alias for QuerySet type checks

Browse source
This commit is contained in:
Jennifer Richards 2023-05-31 11:05:56 -03:00
parent 1922a6066e
commit 3015359323
No known key found for this signature in database
GPG key ID: 9B2BF5C5ADDA6A6E
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
ietf/doc/utils.py
View file

@ -17,7 +17,6 @@ from zoneinfo import ZoneInfo
from django.conf import settings from django.conf import settings
from django.contrib import messages from django.contrib import messages
from django.db.models import QuerySet
from django.forms import ValidationError from django.forms import ValidationError
from django.http import Http404 from django.http import Http404
from django.template.loader import render_to_string from django.template.loader import render_to_string
@ -25,6 +24,7 @@ from django.utils import timezone
from django.utils.html import escape from django.utils.html import escape
from django.urls import reverse as urlreverse from django.urls import reverse as urlreverse
from django_stubs_ext import QuerySetAny
import debug # pyflakes:ignore import debug # pyflakes:ignore
from ietf.community.models import CommunityList from ietf.community.models import CommunityList
@ -345,7 +345,7 @@ def augment_events_with_revision(doc, events):
"""Take a set of events for doc and add a .rev attribute with the """Take a set of events for doc and add a .rev attribute with the
revision they refer to by checking NewRevisionDocEvents.""" revision they refer to by checking NewRevisionDocEvents."""
if isinstance(events, QuerySet): if isinstance(events, QuerySetAny):
qs = events.filter(newrevisiondocevent__isnull=False) qs = events.filter(newrevisiondocevent__isnull=False)
else: else:
qs = NewRevisionDocEvent.objects.filter(doc=doc) qs = NewRevisionDocEvent.objects.filter(doc=doc)
@ -353,7 +353,7 @@ def augment_events_with_revision(doc, events):
if doc.type_id == "draft" and doc.get_state_slug() == "rfc": if doc.type_id == "draft" and doc.get_state_slug() == "rfc":
# add fake "RFC" revision # add fake "RFC" revision
if isinstance(events, QuerySet): if isinstance(events, QuerySetAny):
e = events.filter(type="published_rfc").order_by('time').last() e = events.filter(type="published_rfc").order_by('time').last()
else: else:
e = doc.latest_event(type="published_rfc") e = doc.latest_event(type="published_rfc")