monkey/monkey_island/cc/database.py

import gridfs
from flask_pymongo import MongoClient, PyMongo
from pymongo.errors import ServerSelectionTimeoutError

__author__ = 'Barak'

mongo = PyMongo()


class Database:
    def __init__(self):
        self.gridfs = None

    def init(self):
        self.gridfs = gridfs.GridFS(mongo.db)


database = Database()


def is_db_server_up(mongo_url):
    client = MongoClient(mongo_url, serverSelectionTimeoutMS=100)
    try:
        client.server_info()
        return True
    except ServerSelectionTimeoutError:
        return False