From ed77684427336876075ae3f1beeb900c57ea5464 Mon Sep 17 00:00:00 2001 From: Zixuan James Li Date: Wed, 27 Jul 2022 16:04:14 -0400 Subject: [PATCH] models: Add RealmReactivationStatus to ConfirmObjT. We added RealmReactivationStatus as a possible confirmation object in #22584. We also need to add this type to ConfirmationObjT. Signed-off-by: Zixuan James Li --- confirmation/models.py | 18 ++++++++++++++++-- 1 file changed, 16 insertions(+), 2 deletions(-) diff --git a/confirmation/models.py b/confirmation/models.py index d1420e81d3..6288f87ad5 100644 --- a/confirmation/models.py +++ b/confirmation/models.py @@ -19,7 +19,14 @@ from django.utils.timezone import now as timezone_now from confirmation import settings as confirmation_settings from zerver.lib.types import UnspecifiedValue -from zerver.models import EmailChangeStatus, MultiuseInvite, PreregistrationUser, Realm, UserProfile +from zerver.models import ( + EmailChangeStatus, + MultiuseInvite, + PreregistrationUser, + Realm, + RealmReactivationStatus, + UserProfile, +) class ConfirmationKeyException(Exception): @@ -47,7 +54,14 @@ def generate_key() -> str: return b32encode(secrets.token_bytes(15)).decode().lower() -ConfirmationObjT = Union[MultiuseInvite, PreregistrationUser, EmailChangeStatus, UserProfile, Realm] +ConfirmationObjT = Union[ + MultiuseInvite, + PreregistrationUser, + EmailChangeStatus, + UserProfile, + Realm, + RealmReactivationStatus, +] def get_object_from_key(