def add_default_records(self, endpoint=None): # sanity check if not self.domain_id: raise Exception("Cannot add default records when domain_id is not set") # don't duplicate SOA record try: existing_soa = Record.get(Record.domain_id == self.domain_id, Record.type == RecordType().set("SOA")) except DoesNotExist: # create SOA record try: default_soa = DefaultRecord.get(DefaultRecord.type == RecordType().set("SOA")) soa = Record() soa.domain_id = self.domain_id soa.host = default_soa.host.replace("DOMAIN", self.domain) soa.val = default_soa.val soa.ttl = default_soa.ttl soa.type = default_soa.type # replace uses of DOMAIN soa.save() if endpoint is not None: endpoint.dns_log(soa.domain_id, "added soa") except DoesNotExist: # no default SOA record set! pass # create all other records default_records = DefaultRecord.select().where(DefaultRecord.type != RecordType().set("SOA")) for record in default_records: new = Record() new.domain_id = self.domain_id new.distance = record.distance new.host = record.host.replace("DOMAIN", self.domain) new.val = record.val.replace("DOMAIN", self.domain) new.distance = record.distance new.port = record.port new.ttl = record.ttl new.type = record.type new.weight = record.weight new.save() if endpoint is not None: endpoint.dns_log( new.domain_id, ("added " + RecordType().get(new.type) + " with host " + new.host + " and value " + new.val), )
def test_mx_record(self): model = Record() model.type = 'M' model.host = 'vegadns.ubuntu' model.val = 'mail.vegadns.ubuntu' model.distance = 10 model.ttl = '3600' self.assertEquals(self.export.data_line_from_model(model), '@vegadns.ubuntu::mail.vegadns.ubuntu:10:3600\n')
def test_mx_record(self): model = Record() model.type = 'M' model.host = 'vegadns.ubuntu' model.val = 'mail.vegadns.ubuntu' model.distance = 10 model.ttl = '3600' self.assertEquals( self.export.data_line_from_model(model), '@vegadns.ubuntu::mail.vegadns.ubuntu:10:3600\n' )
def add_default_records(self, endpoint=None, skipSoa=False): # sanity check if not self.domain_id: raise Exception( "Cannot add default records when domain_id is not set") if not skipSoa: try: soa = self.add_default_soa_record(endpoint) except DoesNotExist: # no default SOA record set! pass # create all other records default_records = DefaultRecord.select().where( DefaultRecord.type != RecordType().set("SOA")) for record in default_records: new = Record() new.domain_id = self.domain_id new.distance = record.distance new.host = record.host.replace("DOMAIN", self.domain) new.val = record.val.replace("DOMAIN", self.domain) new.distance = record.distance new.port = record.port new.ttl = record.ttl new.type = record.type new.weight = record.weight new.save() if endpoint is not None: endpoint.dns_log( new.domain_id, ("added " + RecordType().get(new.type) + " with host " + new.host + " and value " + new.val))
def test_srv_record(self): model = Record() model.type = 'V' model.host = '_xmpp-client._tcp.example.com.' model.val = 'xmpp.example.com' model.ttl = '3600' model.distance = 0 model.weight = 10 model.port = 5222 expected = (":_xmpp-client._tcp.example.com." ":33" r":\000\000\000\012\024\146\004xmpp\007example\003com\000" ":3600\n") self.assertEquals(self.export.data_line_from_model(model), expected)
def test_srv_record(self): model = Record() model.type = 'V' model.host = '_xmpp-client._tcp.example.com.' model.val = 'xmpp.example.com' model.ttl = '3600' model.distance = 0 model.weight = 10 model.port = 5222 expected = ( ":_xmpp-client._tcp.example.com." ":33" r":\000\000\000\012\024\146\004xmpp\007example\003com\000" ":3600\n" ) self.assertEquals( self.export.data_line_from_model(model), expected )