Сценарий DNS Python для изменения файла (ов) зоны, получая ошибку относительности и вставляя escape-символы - PullRequest
Новая
       35

Сценарий DNS Python для изменения файла (ов) зоны, получая ошибку относительности и вставляя escape-символы

1 голос
/ 30 октября 2019

Относительно плохо знаком с python и программированием в целом, но хотел бы автоматизировать миграцию DNS, над которой я работаю для большого количества доменов.

Бесстыдно украл некоторые начальные рамки из Блог AgileTesting .

В текущем состоянии сценарий

  • не удается сохранить с ошибками
  • вставляет управляющие символы в новые записи сервера имен

Я могу отметить rdata.rname = respname в функции soarr(respname), так что я знаю, что там что-то конкретное. Не знаю, как углубиться в проблему, основанную на ошибке.

Для экранирующих персонажей я чувствую, что это что-то простое, но мой мозг мягок, поэтому просто включаю его в качестве незначительной проблемы. 

#!/bin/python3
import re,sys
import dns.zone
from dns.exception import DNSException
from dns.rdataclass import *
from dns.rdatatype import *

script, filename, nameservers = sys.argv
sourcefile = open(filename,"r")

def soarr(respname):
        for (name, ttl, rdata) in zone.iterate_rdatas(SOA):
                serial = rdata.serial
                old_name = rdata.rname
                new_serial = serial + 1
                print ("Changing SOA serial from %d to %d" %(serial, new_serial))
                print ("Changing responsible name from %s to %s" %(old_name, respname))
                rdata.serial = new_serial
                rdata.rname = respname
                rdata.expire = 3600
                print (rdata.rname)

def nsrr(nameserver):
        NS_add = "@"
        target = dns.name.Name((nameserver,))
        print ("Adding record of type NS:", NS_add)
        rdataset = zone.find_rdataset(NS_add, rdtype=NS, create=True)
        rdata = dns.rdtypes.ANY.NS.NS(IN, NS, target)
        rdataset.add(rdata, ttl=3600)
        print (rdata)

def savefile(domain):
        print ("debug",domain)
        new_zone_file = "new.%s.hosts" % domain
        print ("Writing modified zone to file %s" % new_zone_file)
        zone.to_file(new_zone_file,domain)

for domainitem in sourcefile:
        domainitem = domainitem.rstrip()
        print ("Processing %s." % domainitem)
        zone_file = '%s.hosts' % domainitem
        zone = dns.zone.from_file(zone_file, domainitem)

# Updating the SOA record, responsible name, lowering TTL and incrementing serial of the zone file.
        soarr('systems.example.com')

# Adding name servers to the zone file.
        if nameservers == 'customer':
                nsrr('ns01.example.com')
        if nameservers == 'core':
                nsrr("ns01.example2.com")
        if nameservers == 'internal':
                nsrr("ns01.int.example2.com")

# Save the file as a new file.
        savefile(domainitem)

Целью является циклическое переключение списка доменов из файла, открытие соответствующего файла зоны, манипулирование зоной и сохранение изменений во вновь названном файле.

Ошибка при ошибке сохранения. 

Traceback (most recent call last):
  File "./zonefile.py", line 62, in <module>
    savefile(domainitem)
  File "./zonefile.py", line 36, in savefile
    zone.to_file(new_zone_file,domain)
  File "/usr/local/lib/python3.6/site-packages/dns/zone.py", line 531, in to_file
    relativize=relativize)
  File "/usr/local/lib/python3.6/site-packages/dns/node.py", line 51, in to_text
    s.write(rds.to_text(name, **kw))
  File "/usr/local/lib/python3.6/site-packages/dns/rdataset.py", line 218, in to_text
    **kw)))
  File "/usr/local/lib/python3.6/site-packages/dns/rdtypes/ANY/SOA.py", line 62, in to_text
    rname = self.rname.choose_relativity(origin, relativize)
AttributeError: 'str' object has no attribute 'choose_relativity'

Как уже упоминалось, выделение одной строки позволяет сохранить файл. В сохраненном файле в записях NS отображаются escape-символы.

@ 3600 IN NS ns01\.example\.com

...