1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39
|
#!/usr/bin/env python3
import argparse
import asyncio
from .resolver import STSResolver
from . import utils
def parse_args():
parser = argparse.ArgumentParser(
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument("-v", "--verbosity",
help="logging verbosity",
type=utils.check_loglevel,
choices=utils.LogLevel,
default=utils.LogLevel.info)
parser.add_argument("domain",
help="domain to fetch MTA-STS policy from")
parser.add_argument("known_version",
nargs="?",
default=None,
help="latest known version")
return parser.parse_args()
def main(): # pragma: no cover
args = parse_args()
with utils.AsyncLoggingHandler(None) as log_handler:
utils.setup_logger('RES', args.verbosity, log_handler)
loop = asyncio.get_event_loop()
resolver = STSResolver(loop=loop)
result = loop.run_until_complete(resolver.resolve(args.domain, args.known_version))
print(result)
if __name__ == '__main__': # pragma: no cover
main()
|