s4-python: Format to PEP8, simplify tests.
[idra/samba.git] / lib / dnspython / dns / rdtypes / IN / AAAA.py
1 # Copyright (C) 2003-2007, 2009, 2010 Nominum, Inc.
2 #
3 # Permission to use, copy, modify, and distribute this software and its
4 # documentation for any purpose with or without fee is hereby granted,
5 # provided that the above copyright notice and this permission notice
6 # appear in all copies.
7 #
8 # THE SOFTWARE IS PROVIDED "AS IS" AND NOMINUM DISCLAIMS ALL WARRANTIES
9 # WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 # MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL NOMINUM BE LIABLE FOR
11 # ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 # WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 # ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
14 # OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15
16 import dns.exception
17 import dns.inet
18 import dns.rdata
19 import dns.tokenizer
20
21 class AAAA(dns.rdata.Rdata):
22     """AAAA record.
23
24     @ivar address: an IPv6 address
25     @type address: string (in the standard IPv6 format)"""
26
27     __slots__ = ['address']
28
29     def __init__(self, rdclass, rdtype, address):
30         super(AAAA, self).__init__(rdclass, rdtype)
31         # check that it's OK
32         junk = dns.inet.inet_pton(dns.inet.AF_INET6, address)
33         self.address = address
34
35     def to_text(self, origin=None, relativize=True, **kw):
36         return self.address
37
38     def from_text(cls, rdclass, rdtype, tok, origin = None, relativize = True):
39         address = tok.get_identifier()
40         tok.get_eol()
41         return cls(rdclass, rdtype, address)
42
43     from_text = classmethod(from_text)
44
45     def to_wire(self, file, compress = None, origin = None):
46         file.write(dns.inet.inet_pton(dns.inet.AF_INET6, self.address))
47
48     def from_wire(cls, rdclass, rdtype, wire, current, rdlen, origin = None):
49         address = dns.inet.inet_ntop(dns.inet.AF_INET6,
50                                      wire[current : current + rdlen])
51         return cls(rdclass, rdtype, address)
52
53     from_wire = classmethod(from_wire)
54
55     def _cmp(self, other):
56         sa = dns.inet.inet_pton(dns.inet.AF_INET6, self.address)
57         oa = dns.inet.inet_pton(dns.inet.AF_INET6, other.address)
58         return cmp(sa, oa)