Be more stingy with connections

I think the act of spinning up ~4000 MongoTargets for the Equities caused things to blow up; this way connections will only by created/used when operations are being performed. This is theoretically less efficient, but from what I've read in the documentation Mongo should have some pooling tricks to not sacrifice performance.

metrik/__init__.py

@@ -1,2 +1,2 @@
-__version__ = '0.3.6'
+__version__ = '0.3.7'
 __release__ = __version__

metrik/targets/mongo.py

@@ -1,30 +1,41 @@
 from luigi import Target
 from pymongo import MongoClient
-from metrik.conf import get_config
 from datetime import datetime
+from contextlib import contextmanager
+
+from metrik.conf import get_config
 
 
 class MongoTarget(Target):
 
-    def __init__(self, collection, id):
+    @contextmanager
+    def get_db(self):
         config = get_config()
-        self.connection = MongoClient(
+        client = MongoClient(
             host=config.get('metrik', 'mongo_host'),
-            port=config.getint('metrik', 'mongo_port'))[
+            port=config.getint('metrik', 'mongo_port'))
-            config.get('metrik', 'mongo_database')
+
-        ]
+        yield client[config.get('metrik', 'mongo_database')]
-        self.collection = self.connection[collection]
+
+        client.close()
+
+    def __init__(self, collection, id):
+        self.collection = collection
         self.id = id
 
     def exists(self):
-        return self.collection.find_one({
+        with self.get_db() as db:
+            return db[self.collection].find_one({
                 '_id': self.id
             }) is not None
 
     def persist(self, dict_object):
         id_dict = dict_object
         id_dict['_id'] = self.id
-        return self.collection.insert_one(id_dict).inserted_id
+
+        with self.get_db() as db:
+            return db[self.collection].insert_one(id_dict).inserted_id
 
     def retrieve(self):
-        return self.collection.find_one({'_id': self.id})
+        with self.get_db() as db:
+            return db[self.collection].find_one({'_id': self.id})