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)
|