diff --git a/flask_heroku.py b/flask_heroku.py index 1605cb4..ddb5e2b 100644 --- a/flask_heroku.py +++ b/flask_heroku.py @@ -35,4 +35,14 @@ class Heroku(object): url = urlparse.urlparse(redis_url) self.app.config.setdefault('REDIS_HOST', url.hostname) self.app.config.setdefault('REDIS_PORT', url.port) - self.app.config.setdefault('REDIS_PASSWORD', url.password) \ No newline at end of file + self.app.config.setdefault('REDIS_PASSWORD', url.password) + + # Mongolab + mongolab_uri = environ.get('MONGOLAB_URI') + if mongolab_uri: + url = urlparse.urlparse(mongolab_uri) + self.app.config.setdefault('MONGODB_USER', url.username) + self.app.config.setdefault('MONGODB_PASSWORD', url.password) + self.app.config.setdefault('MONGODB_HOST', url.hostname) + self.app.config.setdefault('MONGODB_PORT', url.port) + self.app.config.setdefault('MONGODB_DB', url.path[1:]) \ No newline at end of file