58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
def ui(program, args):
|
|
from .db import RespoDB
|
|
from .dns import main
|
|
from .structs import Options
|
|
from argparse import ArgumentParser
|
|
from asyncio import run
|
|
import respodns.checks as chk
|
|
|
|
name = "respodns6"
|
|
parser = ArgumentParser(name,
|
|
description=name + ": test and log DNS records")
|
|
|
|
# TODO: support multiple paths. nargs="+", iterate with pooling?
|
|
parser.add_argument(
|
|
"path", metavar="file-path",
|
|
help="a path to a file containing IPv4 addresses which host DNS servers")
|
|
|
|
parser.add_argument(
|
|
"--database",
|
|
help="specify database for logging")
|
|
|
|
a = parser.parse_args(args)
|
|
|
|
checks = []
|
|
checks += chk.first
|
|
#checks += chk.new
|
|
checks += chk.likely
|
|
#checks += chk.unlikely
|
|
#checks += chk.top100
|
|
|
|
opts = Options()
|
|
opts.dry = a.database is None
|
|
opts.early_stopping = opts.dry
|
|
|
|
if a.database is not None:
|
|
if a.database.startswith("sqlite:"):
|
|
uri = a.database
|
|
else:
|
|
uri = "sqlite:///" + a.database
|
|
|
|
def runwrap(db, debug=False):
|
|
if debug:
|
|
import logging
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
run(main(db, a.path, checks, opts), debug=True)
|
|
else:
|
|
run(main(db, a.path, checks, opts))
|
|
|
|
if opts.dry:
|
|
runwrap(None)
|
|
else:
|
|
# log to a database.
|
|
db = RespoDB(uri, create=True)
|
|
with db: # TODO: .open and .close methods for manual invocation.
|
|
with db.execution as execution: # TODO: clean up this interface.
|
|
opts.execution = execution
|
|
runwrap(db)
|