greendns: Reduce code redundancy
This list of module dependencies are already present in the code three times and I intend to make a change that will require the same dependencies reused even more. This patch is contributed by Smarkets Limited.
This commit is contained in:
@@ -42,23 +42,21 @@ from eventlet.green import select
|
|||||||
from eventlet.support import six
|
from eventlet.support import six
|
||||||
|
|
||||||
|
|
||||||
dns = patcher.import_patched('dns',
|
def import_patched(module_name):
|
||||||
select=select,
|
return patcher.import_patched(module_name,
|
||||||
time=time,
|
select=select,
|
||||||
os=os,
|
time=time,
|
||||||
socket=_socket_nodns)
|
os=os,
|
||||||
dns.resolver = patcher.import_patched('dns.resolver',
|
socket=_socket_nodns)
|
||||||
select=select,
|
|
||||||
time=time,
|
|
||||||
os=os,
|
dns = import_patched('dns')
|
||||||
socket=_socket_nodns)
|
dns.resolver = import_patched('dns.resolver')
|
||||||
|
|
||||||
for pkg in ('dns.entropy', 'dns.inet', 'dns.query'):
|
for pkg in ('dns.entropy', 'dns.inet', 'dns.query'):
|
||||||
setattr(dns, pkg.split('.')[1], patcher.import_patched(pkg,
|
setattr(dns, pkg.split('.')[1], import_patched(pkg))
|
||||||
select=select,
|
del import_patched
|
||||||
time=time,
|
|
||||||
os=os,
|
|
||||||
socket=_socket_nodns))
|
|
||||||
import dns.rdtypes
|
import dns.rdtypes
|
||||||
for pkg in ['dns.rdtypes.IN', 'dns.rdtypes.ANY']:
|
for pkg in ['dns.rdtypes.IN', 'dns.rdtypes.ANY']:
|
||||||
setattr(dns.rdtypes, pkg.split('.')[-1], patcher.import_patched(pkg))
|
setattr(dns.rdtypes, pkg.split('.')[-1], patcher.import_patched(pkg))
|
||||||
|
Reference in New Issue
Block a user