Rename module files

src/dnsmgr/zone_list.py

@@ -0,0 +1,109 @@
+import argparse
+import re
+import sys
+
+from . import NAMED_DEFAULT_VIEW, RECORD_TYPES, DEFAULT_CFGFILE, DNSManager, printe, prettytable
+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=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 = DNSManager(cfgfile=args.config)
+    except RuntimeError as e:
+        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:
+        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:
+            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 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 != NAMED_DEFAULT_VIEW:
+                print(f'View: {view}')
+            print(prettytable(field_names, rows, truncate=True))
+            print()
+
+
+if __name__ == '__main__':
+    main()