Rename module files

src/dnsmgr/record_add.py

@@ -0,0 +1,109 @@
+import argparse
+import dns.rdataclass
+import dns.rdataset
+import sys
+
+from . import (
+    DEFAULT_CFGFILE,
+    NAMED_DEFAULT_VIEW,
+    DNSManager,
+    input_name,
+    input_rdata,
+    input_ttl,
+    input_yes_no,
+    name_from_text,
+    printe,
+    rdata_from_text,
+    select_type,
+    ttl_from_text,
+    type_from_text
+)
+
+
+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=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 = 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)
+
+        origin = zones[0].origin
+
+        if args.name is None:
+            name = input_name(origin, prompt='Record name')
+        else:
+            name = name_from_text(args.name, origin)
+
+        if args.ttl is None:
+            ttl = input_ttl()
+        else:
+            ttl = ttl_from_text(args.ttl)
+
+        if args.type is None:
+            rdtype = select_type(args.all_types)
+        else:
+            rdtype = type_from_text(args.type, args.all_types)
+
+        if not args.value:
+            rdata = input_rdata(rdtype, origin)
+        else:
+            rdata = rdata_from_text(rdtype, ' '.join(args.value), origin)
+
+    except RuntimeError as e:
+        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 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 != 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:
+            printe(e)
+            sys.exit(160)
+
+
+if __name__ == '__main__':
+    main()