Changes regarding to this Flask change: pallets/flask#4682

Author: kashcode

examples/app.py

@@ -1,19 +1,26 @@
+import logging
+
 from flask import Flask
 from flask_postgresql import PostgreSQL
 
+logging.basicConfig(
+    level=logging.DEBUG,
+    format='%(asctime)s %(levelname)-8s %(name)s %(threadName)s : %(message)s'
+)
+
 app = Flask(__name__)
 app.config["PG_HOST"] = "192.168.99.103"
 app.config["PG_USERNAME"] = "test"
 app.config["PG_PASSWORD"] = "test"
 app.config["PG_DB"] = "test"
 app.config["PG_PORT"] = "5432"
 
-db = PostgreSQL(app)
+postgres = PostgreSQL(app)
 
 
 @app.route("/")
 def users():
-    connection = db.connection
+    connection = postgres.connection
 
     with connection:
         with connection.cursor() as curs:

flask_postgresql/postgresql.py

@@ -1,7 +1,7 @@
 import logging
 
 import psycopg2
-from flask import current_app, _app_ctx_stack
+from flask import current_app, g
 
 logger = logging.getLogger(__name__)
 
@@ -44,16 +44,14 @@ def connect(self):
                 logger.debug('Connection opened successfully')
 
     def teardown(self, exception):
-        ctx = _app_ctx_stack.top
-        if hasattr(ctx, 'postgresql_db'):
-            if ctx.postgresql_db is not None:
-                ctx.postgresql_db.close()
-                logger.debug('Connection closed')
+        db = getattr(g, '_postgres_db', None)
+        if db is not None:
+            db.close()
+            logger.debug('Connection closed')
 
     @property
     def connection(self):
-        ctx = _app_ctx_stack.top
-        if ctx is not None:
-            if not hasattr(ctx, 'postgresql_db'):
-                ctx.postgresql_db = self.connect()
-            return ctx.postgresql_db
+        db = getattr(g, '_postgres_db', None)
+        if db is None:
+            db = g._postgres_db = self.connect()
+        return db

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name="Flask-PostgreSQL",
-    version="1.0.0",
+    version="1.1.0",
     author="Aleksandrs Kašs",
     author_email="[email protected]",
     description="Flask extension for PostgreSQL",