diff options
Diffstat (limited to 'dns/rdtypes/ANY/AMTRELAY.py')
| -rw-r--r-- | dns/rdtypes/ANY/AMTRELAY.py | 21 |
1 files changed, 13 insertions, 8 deletions
diff --git a/dns/rdtypes/ANY/AMTRELAY.py b/dns/rdtypes/ANY/AMTRELAY.py index 5a7eb91..9f093de 100644 --- a/dns/rdtypes/ANY/AMTRELAY.py +++ b/dns/rdtypes/ANY/AMTRELAY.py @@ -25,6 +25,11 @@ import dns.rdtypes.util class Relay(dns.rdtypes.util.Gateway): name = 'AMTRELAY relay' + @property + def relay(self): + return self.gateway + + @dns.immutable.immutable class AMTRELAY(dns.rdata.Rdata): @@ -37,11 +42,11 @@ class AMTRELAY(dns.rdata.Rdata): def __init__(self, rdclass, rdtype, precedence, discovery_optional, relay_type, relay): super().__init__(rdclass, rdtype) - Relay(relay_type, relay).check() + relay = Relay(relay_type, relay) self.precedence = self._as_uint8(precedence) self.discovery_optional = self._as_bool(discovery_optional) - self.relay_type = self._as_uint8(relay_type) - self.relay = relay + self.relay_type = relay.type + self.relay = relay.relay def to_text(self, origin=None, relativize=True, **kw): relay = Relay(self.relay_type, self.relay).to_text(origin, relativize) @@ -59,10 +64,10 @@ class AMTRELAY(dns.rdata.Rdata): relay_type = tok.get_uint8() if relay_type > 0x7f: raise dns.exception.SyntaxError('expecting an integer <= 127') - relay = Relay(relay_type).from_text(tok, origin, relativize, - relativize_to) + relay = Relay.from_text(relay_type, tok, origin, relativize, + relativize_to) return cls(rdclass, rdtype, precedence, discovery_optional, relay_type, - relay) + relay.relay) def _to_wire(self, file, compress=None, origin=None, canonicalize=False): relay_type = self.relay_type | (self.discovery_optional << 7) @@ -76,6 +81,6 @@ class AMTRELAY(dns.rdata.Rdata): (precedence, relay_type) = parser.get_struct('!BB') discovery_optional = bool(relay_type >> 7) relay_type &= 0x7f - relay = Relay(relay_type).from_wire_parser(parser, origin) + relay = Relay.from_wire_parser(relay_type, parser, origin) return cls(rdclass, rdtype, precedence, discovery_optional, relay_type, - relay) + relay.relay) |