From 9c12df10c408e2a644a7b729cf36f7dda26ccccf Mon Sep 17 00:00:00 2001 From: Jeffrey Gelens Date: Thu, 18 Jul 2013 09:43:09 +0200 Subject: [PATCH] Fixed bug when initializing SSLify without 'app' argument and using init_app(app) to set it later. --- flask_sslify.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/flask_sslify.py b/flask_sslify.py index e747889..8559bc4 100644 --- a/flask_sslify.py +++ b/flask_sslify.py @@ -8,7 +8,7 @@ YEAR_IN_SECS = 31536000 class SSLify(object): """Secures your Flask App.""" - def __init__(self, app, age=YEAR_IN_SECS, subdomains=False, permanent=False): + def __init__(self, app=None, age=YEAR_IN_SECS, subdomains=False, permanent=False): if app is not None: self.app = app self.hsts_age = age @@ -21,6 +21,8 @@ class SSLify(object): def init_app(self, app): """Configures the configured Flask app to enforce SSL.""" + + self.app = app app.before_request(self.redirect_to_ssl) app.after_request(self.set_hsts_header)