feat(chalice): allow recreate cursor

2022-10-21 16:36:10 +02:00 · 2022-10-21 16:36:10 +02:00 · afeccbbc26
commit afeccbbc26
parent d1da53709f
1 changed files with 1 additions and 0 deletions
--- a/api/chalicelib/utils/pg_client.py
+++ b/api/chalicelib/utils/pg_client.py
@ -117,6 +117,7 @@ class PostgresClient:
    def __enter__(self):
        if self.cursor is None:
            self.cursor = self.connection.cursor(cursor_factory=psycopg2.extras.RealDictCursor)
+            self.cursor.recreate = self.recreate_cursor
        return self.cursor

    def __exit__(self, *args):