adds support for celery with rediscloud

cburmeister · cburmeister · commit 9230210b14c1 · 2013-11-06T11:47:23.000-08:00
diff --git a/flask_heroku.py b/flask_heroku.py
@@ -25,7 +25,12 @@ def init_app(self, app):
         app.config.setdefault('GOOGLE_DOMAIN', environ.get('GOOGLE_DOMAIN'))
 
         # Celery w/ RabbitMQ
-        app.config.setdefault('BROKER_URL', environ.get('RABBITMQ_URL'))
+        if 'RABBITMQ_URL' in environ:
+            app.config.setdefault('BROKER_URL', environ.get('RABBITMQ_URL'))
+        # Celery w/ RedisCloud
+        elif 'REDISCLOUD_URL' in environ:
+            app.config.setdefault('BROKER_URL', environ.get('REDISCLOUD_URL'))
+            app.config.setdefault('BROKER_BACKEND', environ.get('REDISCLOUD_URL'))
 
         # Mailgun
         if 'MAILGUN_SMTP_SERVER' in environ:
