fix: use the intended state machine in person doc searches (#5204)

* fix: use the intended state machine in person doc searches * chore: apply black to two touched functions
ietf-tools · Feb 24, 2023 · cbe534e · cbe534e
1 parent 91be593
commit cbe534e
Showing 1 changed file with 21 additions and 2 deletions.
diff --git a/ietf/person/models.py b/ietf/person/models.py
@@ -187,12 +187,31 @@ def rfcs(self):
 
     def active_drafts(self):
         from ietf.doc.models import Document
-        return Document.objects.filter(documentauthor__person=self, type='draft', states__slug='active').distinct().order_by('-time')
+
+        return (
+            Document.objects.filter(
+                documentauthor__person=self,
+                type="draft",
+                states__type="draft",
+                states__slug="active",
+            )
+            .distinct()
+            .order_by("-time")
+        )
 
     def expired_drafts(self):
         from ietf.doc.models import Document
-        return Document.objects.filter(documentauthor__person=self, type='draft', states__slug__in=['repl', 'expired', 'auth-rm', 'ietf-rm']).distinct().order_by('-time')
 
+        return (
+            Document.objects.filter(
+                documentauthor__person=self,
+                type="draft",
+                states__type="draft",
+                states__slug__in=["repl", "expired", "auth-rm", "ietf-rm"],
+            )
+            .distinct()
+            .order_by("-time")
+        )
 
     def save(self, *args, **kwargs):
         created = not self.pk