How to create a postgres connection that uses context manager?

Question

I was trying to create a wrapper class that used enter/exit to allow for a context manager to close a postgres db connection.