Add __mkfqdn()helper function to main.Zone

This commit is contained in:
minecraftchest1@outlook.com 2024-11-12 07:45:08 -06:00
parent c0b9b0ba05
commit 315331a5c3

12
main.py
View file

@ -99,14 +99,18 @@ class Zone:
return zone return zone
def __mkfqdn(self, name):
if name[-1] != '.':
return name + '.' + self.origin
else:
return name
def new_soa(self, mname: str = 'ns1.example.com', rname: str = 'admin.example.com', serial: int = int(time.time()), refresh: int = 86400, retry: int = 7200, expire: int = 15552000, ttl: int = 21700): def new_soa(self, mname: str = 'ns1.example.com', rname: str = 'admin.example.com', serial: int = int(time.time()), refresh: int = 86400, retry: int = 7200, expire: int = 15552000, ttl: int = 21700):
if mname[-1] != '.': mname = self.__mkfqdn(name)
mname = mname + '.' + self.origin
self.add(SOA(mname=mname, rname=rname, serial=serial, refresh=refresh, retry=retry, expire=expire, ttl=ttl)) self.add(SOA(mname=mname, rname=rname, serial=serial, refresh=refresh, retry=retry, expire=expire, ttl=ttl))
def new_record(self, name: str = '@', ttl: str = 3600, rtype: str = 'A', data: str = '0.0.0.0'): def new_record(self, name: str = '@', ttl: str = 3600, rtype: str = 'A', data: str = '0.0.0.0'):
if name[-1] != '.': name = self.__mkfqdn(name)
name = name + '.' + self.origin
self.add(name=name, ttl=ttl, rtype=rtype, data=data) self.add(name=name, ttl=ttl, rtype=rtype, data=data)
def add(self, record: Record): def add(self, record: Record):