Upload mongoDB.py

utils/mongoDB.py

@@ -0,0 +1,22 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+
+import pymongo
+import json
+
+
+class MongoDBConnector():
+    def __init__(self, str_conn):
+        self.str_connnection = json.loads(str_conn)['conn_string']
+        self.db_name = json.loads(str_conn)['db_name']
+
+    def get_mongo_connection_string(self):
+        if not self.str_connnection.startswith("mongodb://"):
+            self.str_connnection = "mongodb://" + self.str_connnection
+        return self.str_connnection
+
+    def get_mongo_client(self):
+        conn_string = self.get_mongo_connection_string()
+        return pymongo.MongoClient(host=conn_string)
+