From 5343dff5adf3397c88915e963c90e72d5dc32a8f Mon Sep 17 00:00:00 2001
From: FelipeV34 <felipevanegas11@hotmail.com>
Date: Thu, 19 Aug 2021 00:39:03 -0500
Subject: [PATCH] Create contextmanagers.py

Context managers for a connection, a cursor given a connection, and a cursor given the db connection kwargs
---
 lib/contextmanagers.py | 49 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 49 insertions(+)
 create mode 100644 lib/contextmanagers.py

diff --git a/lib/contextmanagers.py b/lib/contextmanagers.py
new file mode 100644
index 00000000..1bec607f
--- /dev/null
+++ b/lib/contextmanagers.py
@@ -0,0 +1,49 @@
+"""psycopg context managers for connections, cursors, and pools."""
+
+
+from contextlib import contextmanager
+from extensions import connect, cursor
+
+
+@contextmanager
+def connection(**kwargs):
+    """Context manager to yield and close a db connection."""
+    connection = connect(**kwargs)
+
+    try:
+        yield connection
+
+    finally:
+        connection.close()
+
+
+@contextmanager
+def cursor(connection):
+    """
+    Context manager to yield and close a cursor given an existing db
+    connection.
+    """
+    cursor = connection.cursor()
+
+    try:
+        yield cursor
+
+    finally:
+        cursor.close()
+
+
+@contextmanager
+def connection_cursor(**kwargs):
+    """
+    Context manager to yield a cursor and close it and its connection for
+    purposes or performing a single db call.
+    """
+    connection = connect(**kwargs)
+    cursor = connection.cursor()
+
+    try:
+        yield cursor
+
+    finally:
+        cursor.close()
+        connection.close()