Switch everything to Python
This commit is contained in:
@@ -1,203 +1,138 @@
|
||||
#!/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 dns.rdataclass
|
||||
import dnsmgr
|
||||
import sys
|
||||
|
||||
usage() {
|
||||
cat <<EOF
|
||||
Usage: $SCRIPT [OPTIONS]... ZONE[@VIEW] NAME TYPE [VALUE]
|
||||
|
||||
Delete DNS records.
|
||||
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 '?'
|
||||
|
||||
Options:
|
||||
-c, --config path to config file
|
||||
-h, --help print this help message
|
||||
-f, --force delete records without confirmation prompt
|
||||
-i, --interactive interactively ask for missing arguments
|
||||
parser = argparse.ArgumentParser(description='Delete 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('type', metavar='TYPE', nargs=nargs, help='DNS record type', default=None)
|
||||
parser.add_argument('value', metavar='VALUE', nargs='*', help='DNS record value, multiple values are choined by a space character', default=None)
|
||||
args = parser.parse_args()
|
||||
|
||||
EOF
|
||||
exit
|
||||
}
|
||||
try:
|
||||
manager = dnsmgr.DNSManager(cfgfile=args.config)
|
||||
except RuntimeError as e:
|
||||
dnsmgr.printe(f'config: {e}')
|
||||
sys.exit(100)
|
||||
|
||||
config_file="/etc/dns-manager/config.sh"
|
||||
force=false
|
||||
interactive=false
|
||||
try:
|
||||
if args.zone is None:
|
||||
zones = manager.select_zones(args.all_zones)
|
||||
else:
|
||||
zones = manager.get_zones(args.zone, args.all_zones)
|
||||
|
||||
declare -a args=()
|
||||
while [ -n "$1" ]; do
|
||||
opt=$1
|
||||
shift
|
||||
case "$opt" in
|
||||
-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
|
||||
;;
|
||||
-f|--force)
|
||||
force=true
|
||||
;;
|
||||
-*)
|
||||
echo "$SCRIPT: invalid option -- '$opt'" >&2
|
||||
exit 1
|
||||
;;
|
||||
*)
|
||||
args+=("$opt")
|
||||
if (( ${#args[@]} > 5 )); then
|
||||
echo "$SCRIPT: invalid argument -- '$opt'" >&2
|
||||
exit 1
|
||||
fi
|
||||
;;
|
||||
esac
|
||||
done
|
||||
for zone in zones:
|
||||
manager.get_zone_content(zone)
|
||||
|
||||
source "$config_file" || exit 2
|
||||
origin = zones[0].origin
|
||||
|
||||
LIB_DIR=${LIB_DIR:-$SCRIPT_DIR/lib}
|
||||
source "$LIB_DIR"/dns.sh || exit 3
|
||||
if args.name is None:
|
||||
names = sorted(set([name.to_unicode() for name in zone for zone in zones]))
|
||||
rows = [[name] for name in names]
|
||||
index = dnsmgr.prettyselect(['Record name'], rows, prompt='Select record name')
|
||||
args.name = names[index]
|
||||
|
||||
set -- "${args[@]}"
|
||||
name = dnsmgr.name_from_text(args.name, origin)
|
||||
|
||||
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
|
||||
for zone in zones:
|
||||
zone.filter_by_name(name, origin)
|
||||
|
||||
declare -A zone_data
|
||||
zones = list(filter(lambda zone: zone.nodes, zones))
|
||||
if not zones:
|
||||
raise RuntimeError(f"No such DNS record -- '{name.to_text(True)}'")
|
||||
|
||||
if [ "${view}" == "*" ]; then
|
||||
json_array_to_bash views < <(dns_zone_views "$zone")
|
||||
else
|
||||
views=("$view")
|
||||
fi
|
||||
if args.type is None:
|
||||
rdtypes = sorted(set([rdataset.rdtype for rdataset in zone.get_node(name) for zone in zones]))
|
||||
if not args.all_types:
|
||||
rdtypes = list(filter(lambda rdtype: rdtype in dnsmgr.RECORD_TYPES, rdtypes))
|
||||
rdtypes = [rdtype.to_text(rdtype) for rdtype in rdtypes]
|
||||
rows = [[rdtype] for rdtype in rdtypes]
|
||||
index = dnsmgr.prettyselect(['Record type'], rows, prompt='Select record type')
|
||||
args.type = rdtypes[index]
|
||||
|
||||
records=$(
|
||||
for view in "${views[@]}"; do
|
||||
dns_zone "$zone" "$view" || exit 22
|
||||
done | "$JQ" --compact-output --slurp 'add'
|
||||
)
|
||||
rdtype = dnsmgr.type_from_text(args.type, args.all_types)
|
||||
|
||||
name=$1
|
||||
if shift; then
|
||||
dns_check_record_name "$name" name || exit 21
|
||||
elif $interactive; then
|
||||
if (( ${#zone_data[@]} > 1 )); then
|
||||
dns_read_record_name name || exit 10
|
||||
else
|
||||
json_array_to_bash names < <("$JQ" --compact-output '[ .[] | .name ] | sort | unique' <<<"$records")
|
||||
COLUMNS=30
|
||||
echo -e "Select record name:\n"
|
||||
select name in "${names[@]}"; do
|
||||
[ -n "$name" ] && break
|
||||
done
|
||||
[ -z "$name" ] && echo "ERROR: record name selection failed" >&2 && exit 11
|
||||
echo
|
||||
fi
|
||||
else
|
||||
echo "$SCRIPT: missing argument -- NAME" >&2
|
||||
exit 1
|
||||
fi
|
||||
for zone in zones:
|
||||
zone.filter_by_rdtype(rdtype)
|
||||
|
||||
records=$("$JQ" --compact-output --arg name "$name" '[ .[] | select(.name == $name) ]' <<<"$records")
|
||||
[ "$records" == "[]" ] && echo "ERROR: no such record -- '$name'" >&2 && exit 5
|
||||
zones = list(filter(lambda zone: zone.nodes, zones))
|
||||
if not zones:
|
||||
raise RuntimeError(f"No such {rdtype.to_text(rdtype)} record -- '{name.to_text(True)}'")
|
||||
|
||||
json_array_to_bash rtypes < <("$JQ" --compact-output '[ .[] | .type ] | sort | unique' <<<"$records")
|
||||
rdata = None
|
||||
if not args.value and not args.batch and not dnsmgr.input_yes_no(f'Delete all {rdtype.to_text(rdtype)}-records?'):
|
||||
values = []
|
||||
for zone in zones:
|
||||
for rdataset in zone.get_node(name):
|
||||
for rdata in rdataset:
|
||||
values.append(rdata.to_text(origin=zone.origin, relativize=False))
|
||||
values = sorted(set(values))
|
||||
rows = [[value] for value in values]
|
||||
index = dnsmgr.prettyselect(['Record value'], rows, prompt='Select record value', truncate=True)
|
||||
args.value = [values[index]]
|
||||
|
||||
rtype=${1^^}
|
||||
if shift; then
|
||||
dns_check_record_type "$rtype" || exit 23
|
||||
elif $interactive; then
|
||||
if [ "$view" == "*" ]; then
|
||||
dns_select_record_type rtype || exit 10
|
||||
else
|
||||
echo -e "Select record type:\n"
|
||||
select rtype in "${rtypes[@]}"; do
|
||||
[ -n "$rtype" ] && break
|
||||
done
|
||||
[ -z "$name" ] && echo "ERROR: record type selection failed" >&2 && exit 11
|
||||
echo
|
||||
fi
|
||||
else
|
||||
echo "$SCRIPT: missing argument -- TYPE" >&2
|
||||
exit 1
|
||||
fi
|
||||
if args.value:
|
||||
value = ' '.join(args.value)
|
||||
rdata = dnsmgr.rdata_from_text(rdtype, value, origin)
|
||||
for zone in zones:
|
||||
zone.filter_by_rdata(rdata)
|
||||
|
||||
records=$("$JQ" --compact-output --arg rtype "$rtype" '[ .[] | select(.type == $rtype) ]' <<<"$records")
|
||||
[ "$records" == "[]" ] && echo "ERROR: no ${rtype} record found" >&2 && exit 5
|
||||
zones = list(filter(lambda zone: zone.nodes, zones))
|
||||
if not zones:
|
||||
raise RuntimeError(f"No such DNS record found -- {name.to_text(True)} IN {rdtype.to_text(rdtype)} {value}")
|
||||
|
||||
json_array_to_bash values < <("$JQ" --compact-output '[ .[] | .value ] | sort | unique' <<<"$records")
|
||||
except RuntimeError as e:
|
||||
dnsmgr.printe(e)
|
||||
sys.exit(150)
|
||||
except KeyboardInterrupt:
|
||||
sys.exit(0)
|
||||
|
||||
value=$1
|
||||
if shift; then
|
||||
dns_check_record_value "$rtype" "$value" value || exit 24
|
||||
elif $interactive; then
|
||||
if [ "$view" == "*" ]; then
|
||||
dns_read_record_value "$rtype" value || exit 10
|
||||
else
|
||||
if ! yes_no "Delete all ${rtype} records?"; then
|
||||
echo -e "\nSelect value:\n"
|
||||
select value in "${values[@]}"; do
|
||||
[ -n "$value" ] && break
|
||||
done
|
||||
fi
|
||||
[ -z "$name" ] && echo "ERROR: invalid answer" >&2 && exit 11
|
||||
echo
|
||||
fi
|
||||
fi
|
||||
zones.sort(key=lambda zone: zone.view)
|
||||
|
||||
if [ -n "$value" ]; then
|
||||
decoded_value=$(dns_decode_txt_value "$value")
|
||||
records=$("$JQ" --compact-output --arg value "$decoded_value" '[ .[] | select(.decoded_value == $value) ]' <<<"$records")
|
||||
[ "$records" == "[]" ] && echo "ERROR: no $rtype record matches value" >&2 && exit 6
|
||||
fi
|
||||
if not args.batch:
|
||||
for zone in zones:
|
||||
print(f'View: {zone.view}')
|
||||
node = zone.find_node(name)
|
||||
rdataset = node.find_rdataset(dns.rdataclass.IN, rdtype)
|
||||
rdclassstr = rdataset.rdclass.to_text(rdataset.rdclass)
|
||||
rdtypestr = rdataset.rdtype.to_text(rdataset.rdtype)
|
||||
for rdata in rdataset:
|
||||
text = rdata.to_text(origin=zone.origin, relativize=False)
|
||||
print(f'\033[31m- {name} {rdataset.ttl} {rdclassstr} {rdtypestr} {text}\033[0m\n')
|
||||
|
||||
json_array_to_bash views < <("$JQ" --compact-output '[ .[] | .view ] | sort | unique' <<<"$records")
|
||||
if not dnsmgr.input_yes_no():
|
||||
sys.exit(0)
|
||||
|
||||
if ! $force; then
|
||||
for view in "${views[@]}"; do
|
||||
echo "View: $view"
|
||||
json_array_to_bash values < <("$JQ" --compact-output '[ .[] | .value ] | sort | unique' <<<"$records")
|
||||
for value in "${values[@]}"; do
|
||||
output=$(dns_record_delete "true" "$zone" "$view" "$name" "$rtype" "$value" 2>&1)
|
||||
if (( $? == 0 )); then
|
||||
echo -n -e "\e[31m- $TAB"
|
||||
echo -n -e "$output\e[0m" | grep --color=never -v -E '^(Outgoing update query:|;.*)?$'
|
||||
else
|
||||
echo -e "\e[31mERROR:\n" >&2
|
||||
echo -e "$output\e[0m" >&2
|
||||
exit 30
|
||||
fi
|
||||
done
|
||||
done
|
||||
echo
|
||||
! yes_no "Proceed?" && echo -e "Aborted" && exit
|
||||
echo
|
||||
fi
|
||||
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='')
|
||||
|
||||
echo -n "Sending DDNS update(s)... "
|
||||
for view in "${views[@]}"; do
|
||||
json_array_to_bash values < <("$JQ" --compact-output '[ .[] | .value ] | sort | unique' <<<"$records")
|
||||
for value in "${values[@]}"; do
|
||||
output=$(dns_record_delete "false" "$zone" "$view" "$name" "$rtype" "$value" 2>&1)
|
||||
if (( $? != 0 )); then
|
||||
echo -e "ERROR updating view -- '$view'\n" >&2
|
||||
echo "$output" >&2
|
||||
exit 31
|
||||
fi
|
||||
done
|
||||
done
|
||||
echo "OK"
|
||||
node = zone.find_node(name)
|
||||
rdataset = node.find_rdataset(dns.rdataclass.IN, rdtype)
|
||||
|
||||
try:
|
||||
manager.delete_zone_record(zone, name, rdataset)
|
||||
print('OK')
|
||||
except RuntimeError as e:
|
||||
dnsmgr.printe(e)
|
||||
sys.exit(160)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user