def setup_mongo(self): try: self.mongo = Db(self.mongo_conf) self.mongo.setup() for key in schemas.keys(): log("Adding schema for collection: {0}".format(key)) self.mongo.add_schema(key, schemas[key]) self.mongo.client.admin.command("ping") log("Set up Mongo successfully.") except Exception as e: log("Unable to set up Mongo: {0}".format(e), logging.ERROR)