Merge pull request #24 from YoApp/master

Fix version comparison of Werkzeug
alisaifee · Jan 30, 2015 · f2c44a8 · f2c44a8
2 parents a2eb928 + 8f5fdcb
commit f2c44a8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/flask_limiter/errors.py b/flask_limiter/errors.py
@@ -2,11 +2,13 @@
 errors and exceptions
 """
 
+from distutils.version import LooseVersion
 from werkzeug.exceptions import HTTPException
 
 def _patch_werkzeug():
     import pkg_resources
-    if pkg_resources.get_distribution("werkzeug").version < "0.9":
+    werkzeug_version = pkg_resources.get_distribution("werkzeug").version
+    if LooseVersion(werkzeug_version) < LooseVersion("0.9"):
         # sorry, for touching your internals :).
         import werkzeug._internal # pragma: no cover
         werkzeug._internal.HTTP_STATUS_CODES[429] = 'Too Many Requests' # pragma: no cover