From 3c4c7b2bbc69b2fa3b9c8460434feee429bab4d3 Mon Sep 17 00:00:00 2001 From: Antoine Nguyen Date: Wed, 8 Jun 2016 17:12:39 +0200 Subject: [PATCH] Allow provider list customization. see #879 --- modoboa/admin/management/commands/subcommands/_dnsbl.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/modoboa/admin/management/commands/subcommands/_dnsbl.py b/modoboa/admin/management/commands/subcommands/_dnsbl.py index 79a824586..bb94488b8 100644 --- a/modoboa/admin/management/commands/subcommands/_dnsbl.py +++ b/modoboa/admin/management/commands/subcommands/_dnsbl.py @@ -6,6 +6,7 @@ import dns.resolver import gevent from gevent import socket +from django.conf import settings from django.core.management.base import BaseCommand from modoboa.admin import constants @@ -54,10 +55,13 @@ class CheckDNSBLCommand(BaseCommand): finally: if address is not None: ip_list.append(address) - + if not hasattr(settings, "DNSBL_PROVIDERS"): + providers = constants.DNSBL_PROVIDERS + else: + providers = settings.DNSBL_PROVIDERS jobs = [ gevent.spawn(self.query, ip_list, provider) - for provider in constants.DNSBL_PROVIDERS] + for provider in providers] gevent.joinall(jobs, timeout) for job in jobs: if not job.successful():