diff foodtruck/bcryptfallback.py @ 640:59968ee07a07

admin: Don't require `bcrypt` for running FoodTruck with `chef`.
author Ludovic Chabant <ludovic@chabant.com>
date Sun, 14 Feb 2016 22:06:32 -0800
parents
children 3885421c29a3
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/foodtruck/bcryptfallback.py	Sun Feb 14 22:06:32 2016 -0800
@@ -0,0 +1,47 @@
+import hashlib
+import logging
+
+
+print_warning = False
+logger = logging.getLogger(__name__)
+
+
+try:
+    from bcrypt import hashpw, gensalt
+except ImportError:
+    print_warning = True
+
+    def hashpw(password, *args, **kwargs):
+        return hashlib.sha512(password).hexdigest().encode('utf8')
+
+    def gensalt(*args, **kwargs):
+        return b''
+
+
+try:
+    from flask.ext.bcrypt import Bcrypt
+except ImportError:
+    print_warning = True
+
+    def generate_password_hash(password):
+        return hashlib.sha512(password.encode('utf8')).hexdigest()
+
+    def check_password_hash(reference, check):
+        check_hash = hashlib.sha512(check.encode('utf8')).hexdigest()
+        return check_hash == reference
+
+    class SHA512Fallback(object):
+        is_fallback_bcrypt = True
+
+        def __init__(self, app=None):
+            self.generate_password_hash = generate_password_hash
+            self.check_password_hash = check_password_hash
+
+    Bcrypt = SHA512Fallback
+
+
+if print_warning:
+    logging.warning("Bcrypt not available... falling back to SHA512.")
+    logging.warning("Run `pip install Flask-Bcrypt` for more secure "
+                    "password hashing.")
+