Switch everything to Python
This commit is contained in:
228
dns-zone-list
228
dns-zone-list
@@ -1,156 +1,106 @@
|
||||
#!/usr/bin/env bash
|
||||
#!/usr/bin/env python3
|
||||
|
||||
SCRIPT_PATH=$(realpath -s "${0}")
|
||||
SCRIPT_DIR=$(dirname "$SCRIPT_PATH")
|
||||
SCRIPT=$(basename "$SCRIPT_PATH")
|
||||
import argparse
|
||||
import dnsmgr
|
||||
import sys
|
||||
|
||||
usage() {
|
||||
cat <<EOF
|
||||
Usage: $SCRIPT [OPTIONS]... ZONE[@VIEW]
|
||||
from json import dumps
|
||||
|
||||
Show DNS zone content.
|
||||
|
||||
Options:
|
||||
-a, --all do not ignore unsupported record types
|
||||
-c, --config path to config file
|
||||
-h, --help print this help message
|
||||
-i, --interactive interactively ask for missing arguments
|
||||
-j, --json print json format
|
||||
-J, --json-pretty print pretty json format (implies -j)
|
||||
-r, --raw print raw format
|
||||
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 '?'
|
||||
|
||||
EOF
|
||||
exit
|
||||
}
|
||||
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()
|
||||
|
||||
all=false
|
||||
config_file="/etc/dns-manager/config.sh"
|
||||
interactive=false
|
||||
json=false
|
||||
json_pretty=false
|
||||
raw=false
|
||||
zone=""
|
||||
try:
|
||||
manager = dnsmgr.DNSManager(cfgfile=args.config)
|
||||
except RuntimeError as e:
|
||||
dnsmgr.printe(f'config: {e}')
|
||||
sys.exit(100)
|
||||
|
||||
declare -a args=()
|
||||
while [ -n "$1" ]; do
|
||||
opt=$1
|
||||
shift
|
||||
case "$opt" in
|
||||
-a|--all)
|
||||
all=true
|
||||
;;
|
||||
-c|--config)
|
||||
if [ -z "$1" ]; then
|
||||
echo "$SCRIPT: missing argument to option -- '$opt'" >&2
|
||||
exit 1
|
||||
fi
|
||||
config_file=$1
|
||||
shift
|
||||
;;
|
||||
-h|--help)
|
||||
usage
|
||||
;;
|
||||
-i|--interactive)
|
||||
interactive=true
|
||||
;;
|
||||
-j|--json)
|
||||
json=true
|
||||
;;
|
||||
-J|--json-pretty)
|
||||
json=true
|
||||
json_pretty=true
|
||||
;;
|
||||
-r|--raw)
|
||||
raw=true
|
||||
;;
|
||||
-*)
|
||||
echo "$SCRIPT: invalid option -- '$opt'" >&2
|
||||
exit 1
|
||||
;;
|
||||
*)
|
||||
args+=("$opt")
|
||||
if (( ${#args[@]} > 1 )); then
|
||||
echo "$SCRIPT: invalid argument -- '$opt'" >&2
|
||||
exit 1
|
||||
fi
|
||||
;;
|
||||
esac
|
||||
done
|
||||
try:
|
||||
if args.zone is None:
|
||||
zones = manager.select_zones(args.all_zones)
|
||||
else:
|
||||
zones = manager.get_zones(args.zone, args.all_zones)
|
||||
|
||||
if $raw && $json; then
|
||||
echo "$SCRIPT: invalid options: --raw and --json are mutually exclusive" >&2
|
||||
exit 1
|
||||
fi
|
||||
except RuntimeError as e:
|
||||
dnsmgr.printe(e)
|
||||
sys.exit(150)
|
||||
except KeyboardInterrupt:
|
||||
sys.exit(0)
|
||||
|
||||
source "$config_file" || exit 2
|
||||
zones.sort(key=lambda zone: zone.view)
|
||||
|
||||
LIB_DIR=${LIB_DIR:-$SCRIPT_DIR/lib}
|
||||
source "$LIB_DIR"/dns.sh || exit 3
|
||||
source "$LIB_DIR"/output.sh || exit 3
|
||||
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)
|
||||
|
||||
set -- "${args[@]}"
|
||||
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_text(),
|
||||
'name_unicode': name.to_unicode(),
|
||||
'ttl': str(rdataset.ttl),
|
||||
'type': str(rdataset.rdtype.to_text(rdataset.rdtype)),
|
||||
'value': value.to_text(origin=zone.origin, relativize=False)})
|
||||
|
||||
zone=$1
|
||||
if shift; then
|
||||
dns_check_zone_view "$zone" zone view || exit 10
|
||||
elif $interactive; then
|
||||
dns_select_zone zone view || exit 11
|
||||
else
|
||||
echo "$SCRIPT: missing argument -- ZONE[@VIEW]" >&2
|
||||
exit 1
|
||||
fi
|
||||
records.sort(key=lambda r: f'{r["name_unicode"]}{r["type"]}')
|
||||
zone_records[zone.view] = records
|
||||
|
||||
declare -A output
|
||||
if [ "${view}" == "*" ]; then
|
||||
json_array_to_bash views < <(dns_zone_views "$zone")
|
||||
else
|
||||
views=("$view")
|
||||
fi
|
||||
views = sorted(zone_records.keys())
|
||||
|
||||
for view in "${views[@]}"; do
|
||||
output["$view"]=$(dns_zone "$zone" "$view" "$all") || exit 12
|
||||
done
|
||||
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))
|
||||
|
||||
if $raw; then
|
||||
n=0
|
||||
for view in $(printf "%s\n" "${!output[@]}" | sort); do
|
||||
if (( ${#output[@]} > 1 )) || [ "$view" != "$NAMED_DEFAULT_VIEW" ]; then
|
||||
cat <<EOF
|
||||
#
|
||||
# VIEW: $view
|
||||
#
|
||||
else:
|
||||
field_names = ['Name', 'TTL', 'Type', 'Value']
|
||||
for view in views:
|
||||
rows = []
|
||||
for record in zone_records[view]:
|
||||
name = record['name_unicode'] if args.decode else record['name']
|
||||
row = [name, record['ttl'], record['type'], record['value']]
|
||||
rows.append(row)
|
||||
|
||||
EOF
|
||||
fi
|
||||
echo "${output["$view"]}"
|
||||
"$JQ" --raw-output '.[] | "\(.name)\t\(.ttl)\t\(.type)\t\(.value)"' <<<"${output["$view"]}"
|
||||
((n++))
|
||||
(( $n < ${#output[@]} )) && echo
|
||||
done
|
||||
exit
|
||||
fi
|
||||
if len(views) > 1 or view != dnsmgr.NAMED_DEFAULT_VIEW:
|
||||
print(f'View: {view}')
|
||||
print(dnsmgr.prettytable(field_names, rows, truncate=True))
|
||||
print()
|
||||
|
||||
if $json; then
|
||||
jq_opts=""
|
||||
$json_pretty || jq_opts="--compact-output"
|
||||
for view in $(printf "%s\n" "${!output[@]}" | sort); do
|
||||
jq --compact-output --slurp "{ view: \"$view\", records: .[] }" <<<"${output["$view"]}"
|
||||
done | jq --slurp $jq_opts
|
||||
exit
|
||||
fi
|
||||
|
||||
max_value_len=$(( ${TERMINAL_WITH} / 2 - 15))
|
||||
n=0
|
||||
for view in $(printf "%s\n" "${!output[@]}" | sort); do
|
||||
(( ${#output[@]} > 1 )) || [ "$view" != "$NAMED_DEFAULT_VIEW" ] && echo "View: $view"
|
||||
{
|
||||
(( $max_value_len <= 0 )) && max_value_len=1
|
||||
while read -r name ttl rtype value; do
|
||||
name=$("$IDN2" --decode <<<"$name")
|
||||
(( ${#value} > $max_value_len )) && value="${value:0:$max_value_len} ...[TRUNCATED]"
|
||||
echo "$name$TAB$ttl$TAB$rtype$TAB$value"
|
||||
done < <("$JQ" --raw-output '.[] | "\(.name)\t\(.ttl)\t\(.type)\t\(.value)"' <<<"${output["$view"]}")
|
||||
} | table_output "NAME" "TTL" "TYPE" "VALUE"
|
||||
((n++))
|
||||
(( $n < ${#output[@]} )) && echo
|
||||
done
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user