From ca7ed3f51eaebf112ad7474c45e070315c795980 Mon Sep 17 00:00:00 2001
From: mcdallas <mc-dallas@hotmail.com>
Date: Tue, 9 Feb 2016 21:29:49 +0000
Subject: [PATCH] enter and exit methods as per Issue #25

---
 records.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/records.py b/records.py
index 7d5cf12..8de3f16 100644
--- a/records.py
+++ b/records.py
@@ -210,6 +210,12 @@ class Database(object):
         # Enable hstore if it's available.
         self._enable_hstore()
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc, val, traceback):
+        self.db.close()
+
     def _enable_hstore(self):
         try:
             register_hstore(self.db)