disconnecting only if there's a connection

cs50 · Jan 13, 2020 · c0b24da · c0b24da
1 parent c73587b
commit c0b24da
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -16,5 +16,5 @@
     package_dir={"": "src"},
     packages=["cs50"],
     url="https://github.com/cs50/python-cs50",
-    version="5.0.1"
+    version="5.0.2"
 )
diff --git a/src/cs50/sql.py b/src/cs50/sql.py
@@ -281,7 +281,8 @@ def execute(self, sql, *args, **kwargs):
                 # Disconnect later
                 @flask.current_app.teardown_appcontext
                 def shutdown_session(exception=None):
-                    flask.g._connection.close()
+                    if hasattr(flask.g, "_connection"):
+                        flask.g._connection.close()
 
             # Use this connection
             connection = flask.g._connection