fix bug introduced in 2.0

Author: thibaut

flask_replicated.py

@@ -2,22 +2,23 @@
 
 from flask import current_app, g, request
 
+
 class FlaskReplicated(object):
     READONLY_METHODS = set(['GET', 'HEAD'])
 
     def __init__(self, app=None):
         if app is not None:
             self.init_app(app)
+        self.AUTO_SLAVE = app.config.get('AUTO_READ_ON_SLAVE', True)
 
     def init_app(self, app):
         assert hasattr(app, 'extensions')
         assert 'sqlalchemy' in app.extensions
         if 'replicated' not in app.extensions:
             app.extensions['replicated'] = self
             binds = app.config.get('SQLALCHEMY_BINDS') or {}
-            auto_slave = app.config.get('AUTO_READ_ON_SLAVE', True)
             if 'slave' in binds:
-                app.before_request(self._pick_database_replica, auto_slave)
+                app.before_request(self._pick_database_replica)
                 db = app.extensions['sqlalchemy'].db
                 get_engine_vanilla = db.get_engine
 
@@ -30,11 +31,11 @@ def get_replicated_engine(app=app, bind=None):
                     return get_engine_vanilla(app, bind)
                 db.get_engine = get_replicated_engine
 
-    def _pick_database_replica(self, auto_slave = True):
+    def _pick_database_replica(self):
         func = current_app.view_functions.get(request.endpoint)
         if getattr(func, 'use_master_database', False):
             g.use_master = True
-        g.use_slave = ( request.method in self.READONLY_METHODS and auto_slave ) or getattr(func, 'use_slave_database', False)
+        g.use_slave = ( request.method in self.READONLY_METHODS and self.AUTO_SLAVE ) or getattr(func, 'use_slave_database', False)
 
 def use_master_database(func):
     func.use_master_database = True
@@ -43,4 +44,4 @@ def use_master_database(func):
 
 def use_slave_database(func):
     func.use_slave_database = True
-    return func
+    return func

setup.py

@@ -28,6 +28,6 @@
         'Programming Language :: Python :: 3.6',
         'Programming Language :: Python :: 3.7',
     ],
-    version='2.0',
+    version='2.1',
     py_modules=['flask_replicated'],
 )