From 3a07adba103b415e5c56fe2c562a824e8525b6e0 Mon Sep 17 00:00:00 2001 From: Justin Riley Date: Wed, 28 Sep 2016 11:07:12 -0400 Subject: [PATCH] resolver: add zone kwarg to {add,remove}_host methods --- onedns/resolver.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/onedns/resolver.py b/onedns/resolver.py index 5d5ef48..f57204e 100644 --- a/onedns/resolver.py +++ b/onedns/resolver.py @@ -62,12 +62,14 @@ class DynamicResolver(server.BaseResolver): self.zone = zone @synchronized(_lock) - def add_host(self, name, ip): - self.zone.add_host(name, ip) + def add_host(self, name, ip, zone=None): + z = zone or self.zone + z.add_host(name, ip) @synchronized(_lock) - def remove_host(self, name, ip): - self.zone.remove_host(name, ip) + def remove_host(self, name, ip, zone=None): + z = zone or self.zone + z.remove_host(name, ip) def start(self, dns_address='0.0.0.0', dns_port=53, api_address='127.0.0.1', api_port=8000, tcp=False, udplen=0,