fork/django-q2
1
0
Fork 0
mirror of https://github.com/django-q2/django-q2.git synced 2026-09-22 21:18:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Moves custom broker class to top

Browse Source
This commit is contained in:
Ilan Steemers
2016-01-27 11:11:29 +01:00
parent b1b4a33c51
commit 3ac7f26de5
1 changed files with 22 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
django_q/brokers/__init__.py
+22 -18
View File
@@ -158,28 +158,32 @@ def get_broker(list_key=Conf.PREFIX):
:type list_key: str
:return:
"""
# disque
if Conf.DISQUE_NODES:
from brokers import disque
return disque.Disque(list_key=list_key)
elif Conf.IRON_MQ:
from brokers import ironmq
return ironmq.IronMQBroker(list_key=list_key)
elif Conf.SQS:
from brokers import aws_sqs
return aws_sqs.Sqs(list_key=list_key)
elif Conf.ORM:
from brokers import orm
return orm.ORM(list_key=list_key)
elif Conf.MONGO:
from brokers import mongo
return mongo.Mongo(list_key=list_key)
elif Conf.BROKER:
# custom
if Conf.BROKER:
module, func = Conf.BROKER.rsplit('.', 1)
m = importlib.import_module(module)
broker = getattr(m, func)
return broker(list_key=list)
# disque
elif Conf.DISQUE_NODES:
from brokers import disque
return disque.Disque(list_key=list_key)
# Iron MQ
elif Conf.IRON_MQ:
from brokers import ironmq
return ironmq.IronMQBroker(list_key=list_key)
# SQS
elif Conf.SQS:
from brokers import aws_sqs
return aws_sqs.Sqs(list_key=list_key)
# ORM
elif Conf.ORM:
from brokers import orm
return orm.ORM(list_key=list_key)
# Mongo
elif Conf.MONGO:
from brokers import mongo
return mongo.Mongo(list_key=list_key)
# default to redis
else:
from brokers import redis_broker