97 lines
3.5 KiB
Python
Executable File
97 lines
3.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import dns.rdataclass
|
|
import dns.rdataset
|
|
import dnsmgr
|
|
import sys
|
|
|
|
|
|
def main():
|
|
preparser = argparse.ArgumentParser(add_help=False)
|
|
preparser.add_argument('-b', '--batch', action='store_true')
|
|
preargs, args = preparser.parse_known_args()
|
|
nargs = None if preargs.batch else '?'
|
|
nvalueargs = '+' if preargs.batch else '*'
|
|
|
|
parser = argparse.ArgumentParser(description='Add DNS records.')
|
|
parser.add_argument('-a', '--all-zones', help='allow zones that are not managed', action='store_true')
|
|
parser.add_argument('-A', '--all-types', help='allow unsupported record types', action='store_true')
|
|
parser.add_argument('-b', '--batch', help='run in batch mode (no user input)', action='store_true')
|
|
parser.add_argument('-c', '--config', help='path to config file', default=dnsmgr.DEFAULT_CFGFILE)
|
|
parser.add_argument('zone', metavar='ZONE[@VIEWS]', nargs=nargs, help='DNS zone name and optional list of views (comma separated or asterisk to select all views)', default=None)
|
|
parser.add_argument('name', metavar='NAME', nargs=nargs, help='DNS record name', default=None)
|
|
parser.add_argument('ttl', metavar='TTL', nargs=nargs, help='DNS record TTL in seconds', type=int, default=None)
|
|
parser.add_argument('type', metavar='TYPE', nargs=nargs, help='DNS record type', default=None)
|
|
parser.add_argument('value', metavar='VALUE', nargs=nvalueargs, help='DNS record value, multiple values are choined by a space character', default=None)
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
manager = dnsmgr.DNSManager(cfgfile=args.config)
|
|
except RuntimeError as e:
|
|
dnsmgr.printe(f'config: {e}')
|
|
sys.exit(100)
|
|
|
|
try:
|
|
if args.zone is None:
|
|
zones = manager.select_zones(args.all_zones)
|
|
else:
|
|
zones = manager.get_zones(args.zone, args.all_zones)
|
|
|
|
origin = zones[0].origin
|
|
|
|
if args.name is None:
|
|
name = dnsmgr.input_name(origin, prompt='Record name')
|
|
else:
|
|
name = dnsmgr.name_from_text(args.name, origin)
|
|
|
|
if args.ttl is None:
|
|
ttl = dnsmgr.input_ttl()
|
|
else:
|
|
ttl = dnsmgr.ttl_from_text(args.ttl)
|
|
|
|
if args.type is None:
|
|
rdtype = dnsmgr.select_type(args.all_types)
|
|
else:
|
|
rdtype = dnsmgr.type_from_text(args.type, args.all_types)
|
|
|
|
if not args.value:
|
|
rdata = dnsmgr.input_rdata(rdtype, origin)
|
|
else:
|
|
rdata = dnsmgr.rdata_from_text(rdtype, ' '.join(args.value), origin)
|
|
|
|
except RuntimeError as e:
|
|
dnsmgr.printe(e)
|
|
sys.exit(150)
|
|
except KeyboardInterrupt:
|
|
sys.exit(0)
|
|
|
|
rdataset = dns.rdataset.Rdataset(dns.rdataclass.IN, rdtype, ttl=ttl)
|
|
rdataset.add(rdata)
|
|
|
|
if not args.batch:
|
|
for zone in zones:
|
|
text = rdataset.to_text(origin=zone.origin, relativize=False)
|
|
print(f'View: {zone.view}')
|
|
print(f'\033[32m+ {name} {text}\033[0m\n')
|
|
|
|
if not dnsmgr.input_yes_no():
|
|
sys.exit(0)
|
|
|
|
for zone in zones:
|
|
origin = zone.origin.to_text(omit_final_dot=True)
|
|
if len(zones) > 1 or zone.view != dnsmgr.NAMED_DEFAULT_VIEW:
|
|
origin = f'{origin}@{zone.view}'
|
|
print(f"Sending DDNS updates for '{origin}'... ", end='')
|
|
|
|
try:
|
|
manager.add_zone_record(zone, name, rdataset)
|
|
print('OK')
|
|
except RuntimeError as e:
|
|
dnsmgr.printe(e)
|
|
sys.exit(160)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|