Merge pull request #6 from Sponsorlytix-Ltd/chore/mongo_client

Chore/mongo client

Author: popmeup

setup.py

@@ -15,7 +15,7 @@
 setup(
     name='sponsorlytix_api_utils',
     description='Sponsorlytix code utils',
-    version='1.1',
+    version='1.2',
     author='Sponsorlytix Team',
     install_requires=requirements,
     setup_requires=['pytest-runner'],

sponsorlytix_api_utils/sponsorlytix_mongo/client.py

@@ -1,20 +1,16 @@
-import os
-
 from pymongo import MongoClient
 
 
 class SponsorlytixMongoClient:
 
-    def __init__(self, mongodb_config, db_name='sponsorlytix'):
-        host, port = mongodb_config.get('host'), mongodb_config.get('port')
+    def __init__(self, host, port, db_name):
         client = MongoClient(host, port)
-        self.client = client[db_name]
+        self.client = client.get_database(db_name)
 
     def get_collection(self, collection_name):
-        return self.client[collection_name]
+        return self.client.get_collection(collection_name)
 
     @classmethod
-    def get_client(cls, db_name='sponsorlytix'):
-        mongo_url = os.environ.get('MONGO_URL')
-        mongo_conn = MongoClient(mongo_url, 27017)
-        return mongo_conn[db_name]
+    def get_client(cls, host, port, db_name):
+        client = MongoClient(host, port)
+        return client.get_database(db_name)