112 lines
4.4 KiB
Python
Executable File
112 lines
4.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import dnsmgr
|
|
import re
|
|
import sys
|
|
|
|
from dns.reversename import ipv4_reverse_domain, ipv6_reverse_domain
|
|
from json import dumps
|
|
|
|
|
|
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 '?'
|
|
|
|
parser = argparse.ArgumentParser(description='Show DNS zone records.')
|
|
parser.add_argument('-a', '--all-zones', help='do not ignore zones that are not managed', action='store_true')
|
|
parser.add_argument('-A', '--all-records', help='do not ignore 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('-d', '--decode', help='decode internationalized domain names (IDN)', action='store_true')
|
|
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)
|
|
output = parser.add_mutually_exclusive_group()
|
|
output.add_argument('-j', '--json', help='print json format', action='store_true')
|
|
output.add_argument('-J', '--json-pretty', help='print pretty json format', action='store_true')
|
|
output.add_argument('-r', '--raw', help='print raw format', action='store_true')
|
|
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)
|
|
|
|
except RuntimeError as e:
|
|
dnsmgr.printe(e)
|
|
sys.exit(150)
|
|
except KeyboardInterrupt:
|
|
sys.exit(0)
|
|
|
|
zones.sort(key=lambda zone: zone.view)
|
|
|
|
zone_records = {}
|
|
for zone in zones:
|
|
try:
|
|
manager.get_zone_content(zone)
|
|
except RuntimeError as e:
|
|
dnsmgr.printe(f"zone transfer of '{zone.origin.to_text(True)}@{zone.view}': {e}")
|
|
sys.exit(160)
|
|
|
|
records = []
|
|
for name, node in zone.items():
|
|
for rdataset in node:
|
|
if not args.all_records and rdataset.rdtype not in dnsmgr.RECORD_TYPES:
|
|
continue
|
|
for value in rdataset:
|
|
records.append({
|
|
'name': name.to_unicode() if args.decode else name.to_text(),
|
|
'ttl': str(rdataset.ttl),
|
|
'type': str(rdataset.rdtype.to_text(rdataset.rdtype)),
|
|
'value': value.to_text(origin=zone.origin, relativize=False)})
|
|
|
|
if zone.origin.is_subdomain(ipv4_reverse_domain):
|
|
records.sort(key=lambda r: int(re.findall(r'\d+|$', r['name'])[0] or 0))
|
|
elif zone.origin.is_subdomain(ipv6_reverse_domain):
|
|
records.sort(key=lambda r: int(''.join(re.findall(r'\d+', r['name'])) or 0))
|
|
else:
|
|
records.sort(key=lambda r: f'{r["name"]}{r["type"]}')
|
|
zone_records[zone.view] = records
|
|
|
|
views = sorted(zone_records.keys())
|
|
|
|
if args.raw:
|
|
for view in views:
|
|
print(f';\n; View: {view}\n;\n')
|
|
for record in zone_records[view]:
|
|
print('\t'.join([record['name'], record['ttl'], record['type'], record['value']]))
|
|
print(f'\n;\n; End of view: {view}\n;\n')
|
|
elif args.json or args.json_pretty:
|
|
json_output = []
|
|
for view in views:
|
|
json_output.append({'view': view, 'records': zone_records[view]})
|
|
if args.json_pretty:
|
|
print(dumps(json_output, indent=2))
|
|
else:
|
|
print(dumps(json_output))
|
|
|
|
else:
|
|
field_names = ['Name', 'TTL', 'Type', 'Value']
|
|
for view in views:
|
|
rows = []
|
|
for record in zone_records[view]:
|
|
row = [record['name'], record['ttl'], record['type'], record['value']]
|
|
rows.append(row)
|
|
|
|
if len(views) > 1 or view != dnsmgr.NAMED_DEFAULT_VIEW:
|
|
print(f'View: {view}')
|
|
print(dnsmgr.prettytable(field_names, rows, truncate=True))
|
|
print()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|