예제 #1
0
파일: query.py 프로젝트: edmonds/dnsviz
def main(argv):
    try:
        q = DVCommandLine(argv[1:])
        if q.query_and_display():
            sys.exit(0)
        else:
            sys.exit(1)
    except KeyboardInterrupt:
        sys.exit(1)
    finally:
        close_default_dns_transport_handler()
예제 #2
0
파일: probe.py 프로젝트: edmonds/dnsviz
                    a.analyze(names, _flush)
                    fh.write('"_meta._dnsviz.":%s}' % json.dumps(dnsviz_meta, **kwargs))
                    sys.exit(0)

            name_objs = a.analyze(names)

        name_objs = filter(lambda x: x is not None, name_objs)

        if not name_objs:
            sys.exit(4)

        d = collections.OrderedDict()
        for name_obj in name_objs:
            name_obj.serialize(d, meta_only)
        d['_meta._dnsviz.'] = dnsviz_meta

        try:
            fh.write(json.dumps(d, **kwargs))
        except IOError, e:
            logger.error('Error writing analysis: %s' % e)
            sys.exit(3)

    except KeyboardInterrupt:
        logger.error('Interrupted.')
        sys.exit(4)
    finally:
        close_default_dns_transport_handler()

if __name__ == "__main__":
    main(sys.argv)