Switch everything to Python
This commit is contained in:
249
dns-zone-delete
249
dns-zone-delete
@@ -1,156 +1,111 @@
|
||||
#!/usr/bin/env bash
|
||||
#!/usr/bin/env python3
|
||||
|
||||
SCRIPT_PATH=$(realpath -s "${0}")
|
||||
SCRIPT_DIR=$(dirname "$SCRIPT_PATH")
|
||||
SCRIPT=$(basename "$SCRIPT_PATH")
|
||||
|
||||
usage() {
|
||||
cat <<EOF
|
||||
Usage: $SCRIPT [OPTIONS]... ZONE[@VIEW]
|
||||
|
||||
Delete DNS zones.
|
||||
|
||||
Options:
|
||||
-c, --config path to config file
|
||||
-h, --help print this help message
|
||||
-f, --force delete zones without confirmation prompt
|
||||
-i, --interactive interactively ask for missing arguments
|
||||
EOF
|
||||
exit
|
||||
}
|
||||
|
||||
config_file="/etc/dns-manager/config.sh"
|
||||
force=false
|
||||
interactive=false
|
||||
zone=""
|
||||
|
||||
declare -a args=()
|
||||
while [ -n "$1" ]; do
|
||||
opt=$1
|
||||
shift
|
||||
case "$opt" in
|
||||
-c|--config)
|
||||
config_file=$1
|
||||
if ! shift; then
|
||||
echo "$SCRIPT: missing argument to option -- '$opt'" >&2
|
||||
exit 1
|
||||
fi
|
||||
;;
|
||||
-h|--help)
|
||||
usage
|
||||
;;
|
||||
-f|--force)
|
||||
force=true
|
||||
;;
|
||||
-i|--interactive)
|
||||
interactive=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
|
||||
|
||||
source "$config_file" || exit 2
|
||||
|
||||
LIB_DIR=${LIB_DIR:-$SCRIPT_DIR/lib}
|
||||
source "$LIB_DIR"/dns.sh || exit 3
|
||||
source "$LIB_DIR"/output.sh || exit 3
|
||||
|
||||
set -- "${args[@]}"
|
||||
|
||||
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
|
||||
|
||||
declare -A output
|
||||
if [ "${view}" == "*" ]; then
|
||||
json_array_to_bash views < <(dns_zone_views)
|
||||
else
|
||||
views=("$view")
|
||||
fi
|
||||
|
||||
for view in "${views[@]}"; do
|
||||
dns_get_base_config "$view" zone_dir conf_dir conf_file || exit 10
|
||||
|
||||
zone_conf_file="$conf_dir/$zone.conf"
|
||||
! [ -f "$zone_conf_file" ] && echo "ERROR: zone exists in DNS but no config file found, zone not managed by DNS-Manager -- '$zone_conf_file'" >&2 && exit 14
|
||||
|
||||
zone_file="$zone_dir/$zone.zone"
|
||||
! [ -f "$zone_file" ] && echo "ERROR: zone exists in DNS but no zone file found, zone not managed by DNS-Manager -- '$zone_file'" >&2 && exit 15
|
||||
done
|
||||
import argparse
|
||||
import dns.rdataclass
|
||||
import dns.rdataset
|
||||
import dns.rdatatype
|
||||
import dnsmgr
|
||||
import sys
|
||||
|
||||
|
||||
if ! $force; then
|
||||
for view in "${views[@]}"; do
|
||||
echo "View: $view"
|
||||
echo -e "\e[31m- $TAB$zone\e[0m"
|
||||
done
|
||||
echo
|
||||
! yes_no "Proceed?" && echo -e "Aborted" && exit
|
||||
echo
|
||||
fi
|
||||
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 '?'
|
||||
|
||||
echo -n "Deleting zone from config... "
|
||||
for view in "${views[@]}"; do
|
||||
dns_get_base_config "$view" zone_dir conf_dir conf_file || exit 10
|
||||
parser = argparse.ArgumentParser(description='Delete DNS zones.')
|
||||
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)
|
||||
args = parser.parse_args()
|
||||
|
||||
zone_conf_file="$conf_dir/$zone.conf"
|
||||
! rm "$zone_conf_file" && echo "ERROR: unable to delete config file -- '$zone_conf_file'" >&2 && exit 14
|
||||
|
||||
tmp=$(mktemp)
|
||||
cat >"$tmp" <<EOF
|
||||
/*
|
||||
* This file was generated by DNS-Manager.
|
||||
* DO NOT EDIT, YOUR CHANGES WILL BE OVERWRITTEN!
|
||||
*/
|
||||
EOF
|
||||
while IFS=$NEWLINE read -r file; do
|
||||
if ! cat "$file" >>"$tmp"; then
|
||||
echo "ERROR: unable to write to temp file -- '$tmp'" >&2
|
||||
rm "$tmp"
|
||||
exit 23
|
||||
fi
|
||||
done < <(find "$conf_dir" -maxdepth 1 -type f -name '*.conf')
|
||||
if ! cat "$tmp" > "$conf_file"; then
|
||||
echo "ERROR: unable to write config file -- '$conf_file'" >&2
|
||||
rm "$tmp"
|
||||
exit 24
|
||||
fi
|
||||
rm "$tmp"
|
||||
done
|
||||
echo "Ok"
|
||||
try:
|
||||
manager = dnsmgr.DNSManager(cfgfile=args.config)
|
||||
except RuntimeError as e:
|
||||
dnsmgr.printe(f'config: {e}')
|
||||
sys.exit(100)
|
||||
|
||||
echo -n "Reload Bind config... "
|
||||
rndc_args=""
|
||||
[ -n "$CONTROL_KEY" ] && rndc_args="-k $CONTROL_KEY"
|
||||
! "$RNDC" $rndc_args reconfig && echo "ERROR: rndc reconfig failed" >&2 && exit 25
|
||||
echo "Ok"
|
||||
try:
|
||||
if args.zone is None:
|
||||
zones = manager.select_zones()
|
||||
else:
|
||||
zones = manager.get_zones(args.zone)
|
||||
except RuntimeError as e:
|
||||
dnsmgr.printe(e)
|
||||
sys.exit(150)
|
||||
except KeyboardInterrupt:
|
||||
sys.exit(0)
|
||||
|
||||
error=false
|
||||
echo -n "Deleting zone files... "
|
||||
for view in "${views[@]}"; do
|
||||
dns_get_base_config "$view" zone_dir || exit 10
|
||||
if not args.batch:
|
||||
for zone in zones:
|
||||
origin = zone.origin.to_text(omit_final_dot=True)
|
||||
print(f'View: {zone.view}')
|
||||
print(f'\033[31m- {origin}\033[0m\n')
|
||||
|
||||
zone_file="$zone_dir/$zone.zone"
|
||||
! rm "$zone_file" && echo "ERROR: unable to delete zone file -- '$zone_file'" >&2 && error=true
|
||||
if not dnsmgr.input_yes_no():
|
||||
sys.exit(0)
|
||||
|
||||
while IFS=$NEWLINE read -r file; do
|
||||
! rm "$file" && echo "ERROR: unable to delete zone related file -- '$zone_file'" >&2 && error=true
|
||||
done < <(find "$zone_dir" -maxdepth 1 -type f -name "$zone.zone.*")
|
||||
done
|
||||
! $error && echo "Ok"
|
||||
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}'
|
||||
|
||||
try:
|
||||
catalog_zone_name = manager.config.zones_config[zone.view].catalog_zone
|
||||
if catalog_zone_name:
|
||||
try:
|
||||
catalog_zones = manager.get_zones(catalog_zone_name, all_zones=True)
|
||||
except RuntimeError as e:
|
||||
raise RuntimeError(f'catalog zone of view \'{zone.view}\': {e}')
|
||||
|
||||
for catalog_zone in catalog_zones:
|
||||
manager.get_zone_content(catalog_zone)
|
||||
|
||||
rdname = dns.name.from_text(zone.nfz() + '.zones', catalog_zone.origin)
|
||||
node = catalog_zone.get_node(rdname)
|
||||
if not node:
|
||||
continue
|
||||
|
||||
rdataset = node.get_rdataset(dns.rdataclass.IN, dns.rdatatype.PTR)
|
||||
if not rdataset:
|
||||
continue
|
||||
|
||||
catalog_zone_origin = catalog_zone.origin.to_text(omit_final_dot=True)
|
||||
if catalog_zone.view != dnsmgr.NAMED_DEFAULT_VIEW:
|
||||
catalog_zone_origin += f'@{catalog_zone.view}'
|
||||
print(f'Removing zone \'{origin}\' from catalog zone \'{catalog_zone_origin}\'... ', end='')
|
||||
manager.delete_zone_record(catalog_zone, rdname, rdataset)
|
||||
print('OK')
|
||||
|
||||
print(f"Deleting config of zone '{origin}'... ", end='')
|
||||
manager.delete_zone(zone)
|
||||
print('OK')
|
||||
except RuntimeError as e:
|
||||
dnsmgr.printe(e)
|
||||
sys.exit(160)
|
||||
|
||||
try:
|
||||
print('Reloading named... ', end='')
|
||||
manager.named_reload()
|
||||
print('OK')
|
||||
except RuntimeError as e:
|
||||
dnsmgr.printe(e)
|
||||
sys.exit(170)
|
||||
|
||||
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"Cleanup zone files of zone '{origin}'... ", end='')
|
||||
|
||||
try:
|
||||
manager.cleanup_zone(zone)
|
||||
print('OK')
|
||||
except Exception as e:
|
||||
dnsmgr.printe(e)
|
||||
sys.exit(180)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user