Add rate limiting exceptions.

firebase · Nov 15, 2023 · 4ad0339 · 4ad0339
1 parent c77608e
commit 4ad0339
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/firebase_admin/_auth_utils.py b/firebase_admin/_auth_utils.py
@@ -405,6 +405,20 @@ def __init__(self, message, cause=None, http_response=None):
         exceptions.InvalidArgumentError.__init__(self, message, cause, http_response)
 
 
+class TooManyAttemptsTryLaterError(exceptions.ResourceExhaustedError):
+    """Rate limited because of too many attempts."""
+
+    def __init__(self, message, cause=None, http_response=None):
+        exceptions.ResourceExhaustedError.__init__(self, message, cause, http_response)
+
+
+class ResetPasswordExceedLimitError(exceptions.ResourceExhaustedError):
+    """Reset password emails exceeded their limits."""
+
+    def __init__(self, message, cause=None, http_response=None):
+        exceptions.ResourceExhaustedError.__init__(self, message, cause, http_response)
+
+
 _CODE_TO_EXC_TYPE = {
     'CONFIGURATION_NOT_FOUND': ConfigurationNotFoundError,
     'DUPLICATE_EMAIL': EmailAlreadyExistsError,
@@ -417,6 +431,8 @@ def __init__(self, message, cause=None, http_response=None):
     'PHONE_NUMBER_EXISTS': PhoneNumberAlreadyExistsError,
     'TENANT_NOT_FOUND': TenantNotFoundError,
     'USER_NOT_FOUND': UserNotFoundError,
+    'TOO_MANY_ATTEMPTS_TRY_LATER': TooManyAttemptsTryLaterError,
+    'RESET_PASSWORD_EXCEED_LIMIT': ResetPasswordExceedLimitError,
 }