Switch everything to Python
This commit is contained in:
214
dns-record-add
214
dns-record-add
@@ -1,152 +1,96 @@
|
||||
#!/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 dns.rdataset
|
||||
import dnsmgr
|
||||
import sys
|
||||
|
||||
usage() {
|
||||
cat <<EOF
|
||||
Usage: $SCRIPT [OPTIONS]... ZONE[@VIEW] NAME TTL TYPE VALUE
|
||||
|
||||
Add new records to a DNS zone.
|
||||
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 '*'
|
||||
|
||||
Options:
|
||||
-c, --config path to config file
|
||||
-h, --help print this help message
|
||||
-f, --force add records without confirmation prompt
|
||||
-i, --interactive interactively ask for missing arguments
|
||||
EOF
|
||||
exit
|
||||
}
|
||||
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=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('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()
|
||||
|
||||
config_file="/etc/dns-manager/config.sh"
|
||||
force=false
|
||||
interactive=false
|
||||
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
|
||||
-c|--config)
|
||||
if [ -z "$1" ]; then
|
||||
echo "$SCRIPT: missing argument to option -- '$opt'" >&2
|
||||
exit 1
|
||||
fi
|
||||
config_file=$1
|
||||
shift
|
||||
;;
|
||||
-h|--help)
|
||||
usage
|
||||
;;
|
||||
-f|--force)
|
||||
force=true
|
||||
;;
|
||||
-i|--interactive)
|
||||
interactive=true
|
||||
;;
|
||||
-*)
|
||||
echo "$SCRIPT: invalid option -- '$opt'" >&2
|
||||
exit 1
|
||||
;;
|
||||
*)
|
||||
args+=("$opt")
|
||||
if (( ${#args[@]} > 6 )); 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)
|
||||
|
||||
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:
|
||||
name = dnsmgr.input_name(origin, prompt='Record name')
|
||||
else:
|
||||
name = dnsmgr.name_from_text(args.name, origin)
|
||||
|
||||
set -- "${args[@]}"
|
||||
if args.ttl is None:
|
||||
ttl = dnsmgr.input_ttl()
|
||||
else:
|
||||
ttl = dnsmgr.ttl_from_text(args.ttl)
|
||||
|
||||
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
|
||||
if args.type is None:
|
||||
rdtype = dnsmgr.select_type(args.all_types)
|
||||
else:
|
||||
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
|
||||
dns_read_record_name name || exit 10
|
||||
else
|
||||
echo "$SCRIPT: missing argument -- NAME" >&2
|
||||
exit 1
|
||||
fi
|
||||
if args.value is None:
|
||||
rdata = dnsmgr.input_rdata(rdtype, origin)
|
||||
else:
|
||||
rdata = dnsmgr.rdata_from_text(rdtype, ' '.join(args.value), origin)
|
||||
|
||||
ttl=$1
|
||||
if shift; then
|
||||
dns_check_ttl "$ttl" || exit 22
|
||||
elif $interactive; then
|
||||
dns_read_ttl ttl || exit 10
|
||||
else
|
||||
echo "$SCRIPT: missing argument -- TTL" >&2
|
||||
exit 1
|
||||
fi
|
||||
except RuntimeError as e:
|
||||
dnsmgr.printe(e)
|
||||
sys.exit(150)
|
||||
except KeyboardInterrupt:
|
||||
sys.exit(0)
|
||||
|
||||
rtype=$1
|
||||
if shift; then
|
||||
dns_check_record_type "$rtype" || exit 23
|
||||
elif $interactive; then
|
||||
dns_select_record_type rtype || exit 10
|
||||
else
|
||||
echo "$SCRIPT: missing argument -- TYPE" >&2
|
||||
exit 1
|
||||
fi
|
||||
rdataset = dns.rdataset.Rdataset(dns.rdataclass.IN, rdtype, ttl=ttl)
|
||||
rdataset.add(rdata)
|
||||
|
||||
value=$1
|
||||
if shift; then
|
||||
dns_check_record_value "$rtype" "$value" value || exit 24
|
||||
elif $interactive; then
|
||||
dns_read_record_value "$rtype" value || exit 10
|
||||
else
|
||||
echo "$SCRIPT: missing argument -- VALUE" >&2
|
||||
exit 1
|
||||
fi
|
||||
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 [ "${view}" == "*" ]; then
|
||||
json_array_to_bash views < <(dns_zone_views "$zone")
|
||||
else
|
||||
views=("$view")
|
||||
fi
|
||||
if not dnsmgr.input_yes_no():
|
||||
sys.exit(0)
|
||||
|
||||
if ! $force; then
|
||||
for view in "${views[@]}"; do
|
||||
echo "View: $view"
|
||||
output=$(dns_record_add "true" "$zone" "$view" "$name" "$ttl" "$rtype" "$value" 2>&1)
|
||||
if (( $? == 0 )); then
|
||||
echo -n -e "\e[32m+ $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
|
||||
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
|
||||
output=$(dns_record_add "false" "$zone" "$view" "$name" "$ttl" "$rtype" "$value" 2>&1)
|
||||
if (( $? != 0 )); then
|
||||
echo -e "ERROR\n" >&2
|
||||
echo "$output" >&2
|
||||
exit 31
|
||||
fi
|
||||
done
|
||||
echo "OK"
|
||||
try:
|
||||
manager.add_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