getdc.py

#!/usr/bin/env python3
import sys
import argparse
import logging
import adenum
from ad.dc import get_domain_controllers_by_dns
from net.name import get_fqdn_by_addr

logger = logging.getLogger(adenum.__name__)

parser = argparse.ArgumentParser()
parser.add_argument('-d', '--domain', help='domain')
parser.add_argument('--debug', action='store_true', help='enable debug output')
parser.add_argument('-n', '--name-server', dest='name_server', help='name server')
args = parser.parse_args()

if args.debug:
    h = logging.StreamHandler()
    h.setFormatter(logging.Formatter('[%(levelname)s] %(filename)s:%(lineno)s %(message)s'))
    for n in [__name__, 'plugins', 'lib']:
        l = logging.getLogger(n)
        l.setLevel(logging.DEBUG)
        l.addHandler(h)

if not args.domain:
    if args.name_server:
        name = get_fqdn_by_addr(args.name_server, args.name_server)
        if name:
            args.domain = name.split('.', maxsplit=1)[-1]

if not args.domain:
    print('Error: must specify a domain with -d')
    sys.exit()

for dc in get_domain_controllers_by_dns(args.domain, args.name_server):
    print(dc['address'], '\t', dc['hostname'])