From 6ec42453740d54df2df045fc6db3e5392e0ef4eb Mon Sep 17 00:00:00 2001 From: Lyndsy Simon Date: Sun, 3 Mar 2013 16:24:37 -0600 Subject: [PATCH] Adds support for resource URIs for MongoHQ and Mongolab. Old-style MONGODB_* configuration is left in place, but Flask-PyMongo at least uses the MONGO_URI in preference. At some point in the future, we should remove the MONGODB_* selectors if it is possible to do so. --- flask_heroku.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/flask_heroku.py b/flask_heroku.py index 902da23..97b878f 100644 --- a/flask_heroku.py +++ b/flask_heroku.py @@ -58,6 +58,7 @@ class Heroku(object): mongolab_uri = environ.get('MONGOLAB_URI') if mongolab_uri: url = urlparse.urlparse(mongolab_uri) + app.config.setdefault('MONGO_URI', mongolab_uri) app.config.setdefault('MONGODB_USER', url.username) app.config.setdefault('MONGODB_PASSWORD', url.password) app.config.setdefault('MONGODB_HOST', url.hostname) @@ -68,6 +69,7 @@ class Heroku(object): mongohq_uri = environ.get('MONGOHQ_URL') if mongohq_uri: url = urlparse.urlparse(mongohq_uri) + app.config.setdefault('MONGO_URI', mongohq_uri) app.config.setdefault('MONGODB_USER', url.username) app.config.setdefault('MONGODB_PASSWORD', url.password) app.config.setdefault('MONGODB_HOST', url.hostname)