#313 don't use werkzeug, flask 2.0.0 and use Response instead of BaseResponse

Author: hbusul

flask_restx/api.py

@@ -34,7 +34,13 @@
     NotAcceptable,
     InternalServerError,
 )
-from werkzeug.wrappers import BaseResponse
+
+from werkzeug import __version__ as werkzeug_version
+
+if werkzeug_version.split('.')[0] >= '2':
+    from werkzeug.wrappers import Response as BaseResponse
+else:
+    from werkzeug.wrappers import BaseResponse
 
 from . import apidoc
 from .mask import ParseError, MaskError

flask_restx/resource.py

@@ -3,7 +3,12 @@
 
 from flask import request
 from flask.views import MethodView
-from werkzeug.wrappers import BaseResponse
+from werkzeug import __version__ as werkzeug_version
+
+if werkzeug_version.split('.')[0] >= '2':
+    from werkzeug.wrappers import Response as BaseResponse
+else:
+    from werkzeug.wrappers import BaseResponse
 
 from .model import ModelBase
 

requirements/install.pip

@@ -1,8 +1,8 @@
 aniso8601==8.0.0; python_version < '3.5'
 aniso8601>=0.82; python_version >= '3.5'
 jsonschema
-Flask>=0.8, <2.0.0
-werkzeug <2.0.0
+Flask>=0.8, !=2.0.0
+werkzeug !=2.0.0
 pytz
 six>=1.3.0
 enum34; python_version < '3.4'