From be016c67934071319b04b53348d076bf993e5379 Mon Sep 17 00:00:00 2001
From: Sebastian Rittau <srittau@rittau.biz>
Date: Sat, 23 May 2020 22:39:48 +0200
Subject: [PATCH] Add retry_after to TooManyRequests and ServiceUnavailable
 (#4059)

Added in werkzeug 1.0.0

While the implementation accepts any value for retry_after, the docstring
allows only datetime and int. Using any other value could lead to an
invalid HTTP response.
---
 third_party/2and3/werkzeug/exceptions.pyi | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/third_party/2and3/werkzeug/exceptions.pyi b/third_party/2and3/werkzeug/exceptions.pyi
index 36b14ee82..41f8012b5 100644
--- a/third_party/2and3/werkzeug/exceptions.pyi
+++ b/third_party/2and3/werkzeug/exceptions.pyi
@@ -1,3 +1,4 @@
+import datetime
 from typing import Any, Dict, Tuple, List, Text, NoReturn, Optional, Protocol, Type, Union, Iterable
 
 from wsgiref.types import WSGIEnvironment, StartResponse
@@ -124,7 +125,16 @@ class PreconditionRequired(HTTPException):
     code: int
     description: Text
 
-class TooManyRequests(HTTPException):
+class _RetryAfter(HTTPException):
+    retry_after: Union[None, int, datetime.datetime]
+    def __init__(
+        self,
+        description: Optional[Text] = ...,
+        response: Optional[Response] = ...,
+        retry_after: Union[None, int, datetime.datetime] = ...,
+    ) -> None: ...
+
+class TooManyRequests(_RetryAfter):
     code: int
     description: Text
 
@@ -148,7 +158,7 @@ class BadGateway(HTTPException):
     code: int
     description: Text
 
-class ServiceUnavailable(HTTPException):
+class ServiceUnavailable(_RetryAfter):
     code: int
     description: Text