435b0392669fcaea56b6039c5c4a45f80becf705
[zonemaker.git] / zonemaker / zone.py
1 import ipaddress, re
2
3 second = 1
4 minute = 60*second
5 hour = 60*minute
6 day = 24*hour
7
8 def hostname(name):
9     # check hostname for validity
10     label = r'[a-zA-Z90-9]([a-zA-Z90-9-]{0,61}[a-zA-Z90-9])?' # must not start or end with hyphen
11     pattern = r'^{0}(\.{0})*\.?'.format(label)
12     print(pattern)
13     if re.match(pattern, name):
14         return name
15     raise Exception(name+" is not a valid hostname")
16
17 class Address:
18     def __init__(self, IPv4 = None, IPv6 = None):
19         self._IPv4 = None if IPv4 is None else ipaddress.IPv4Address(IPv4)
20         self._IPv6 = None if IPv6 is None else ipaddress.IPv6Address(IPv6)
21     
22     def IPv4(self):
23         return Address(IPv4 = self._IPv4)
24     
25     def IPv6(self):
26         return Address(IPv6 = self._IPv6)
27
28 class Name:
29     def __init__(self, address = None, MX = None, TCP = None, UDP = None):
30         self._address = address
31
32 class Service:
33     def __init__(self, SRV = None, TLSA=None):
34         self._SRV = SRV
35         self._TLSA = TLSA
36
37 class CName:
38     def __init__(self, name):
39         self._name = name
40
41 class Delegation():
42     def __init__(self, NS, DS = None):
43         pass
44
45 class Zone:
46     def __init__(self, name, mail, NS,
47                  secondary_refresh, secondary_retry, secondary_discard,
48                  NX_TTL = None, A_TTL = None, other_TTL = None,
49                  domains = []):
50         self._name = hostname(name)
51         if not mail.endswith('.'): raise Exception("Mail must be absolute, end with a dot")
52         atpos = mail.find('@')
53         if atpos < 0 or atpos > mail.find('.'): raise Exception("Mail must contain an @ before the first dot")
54         self._mail = hostname(mail.replace('@', '.', 1))
55         self._NS = list(map(hostname, NS))
56         
57         self._secondary_refresh = secondary_refresh
58         self._secondary_retry = secondary_retry
59         self._secondary_discard = secondary_discard
60         
61         assert other_TTL is not None
62         self._NX_TTL = other_TTL if NX_TTL is None else NX_TTL
63         self._A_TTL = other_TTL if A_TTL is None else A_TTL
64         self._other_TTL = other_TTL
65     
66     def write(self, file):
67         raise NotImplementedError()