From d276f58e2ad5e60d80cd17224e69aee65fdbb8b3 Mon Sep 17 00:00:00 2001 From: awsum Date: Fri, 20 Apr 2012 00:10:44 +0300 Subject: [PATCH] Mongolab --- flask_heroku.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) 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