def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("-r", "--region", dest="regions", action="append",
                        help="optional list of regions")
    parser.add_argument("-q", "--quiet", action="store_true",
                        help="Supress logging messages")
    parser.add_argument("-j", "--concurrency", type=int, default=4)

    args = parser.parse_args()

    logging.basicConfig(format="%(asctime)s - %(levelname)s - %(message)s")
    if not args.quiet:
        log.setLevel(logging.DEBUG)
    else:
        log.setLevel(logging.ERROR)
        logging.getLogger("boto").setLevel(logging.CRITICAL)

    if not args.regions:
        args.regions = DEFAULT_REGIONS

    instances_to_tag = []
    for r in args.regions:
        instances_to_tag.extend(get_instances_to_tag(r))
        if instances_to_tag:
            populate_spot_requests_cache(
                r, [i.spot_instance_request_id for i in instances_to_tag])
            for i in instances_to_tag:
                log.debug("tagging %s", i)
                copy_spot_request_tags(i)
Esempio n. 2
0
def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("-r",
                        "--region",
                        dest="regions",
                        action="append",
                        help="optional list of regions")
    parser.add_argument("-q",
                        "--quiet",
                        action="store_true",
                        help="Supress logging messages")
    parser.add_argument("-j", "--concurrency", type=int, default=4)

    args = parser.parse_args()

    logging.basicConfig(format="%(asctime)s - %(levelname)s - %(message)s")
    if not args.quiet:
        log.setLevel(logging.DEBUG)
    else:
        log.setLevel(logging.ERROR)
        logging.getLogger("boto").setLevel(logging.CRITICAL)

    if not args.regions:
        args.regions = DEFAULT_REGIONS

    instances_to_tag = []
    for r in args.regions:
        instances_to_tag.extend(get_instances_to_tag(r))
        if instances_to_tag:
            populate_spot_requests_cache(
                r, [i.spot_instance_request_id for i in instances_to_tag])
            for i in instances_to_tag:
                log.debug("tagging %s", i)
                copy_spot_request_tags(i)
def test_some_tags(m_get_spot_instances):
    i = mock.Mock()
    i.tags = {
        "Name1": "n1",
        "FQDN": "fqdn1",
        "moz-type": "t1",
        "moz-state": "s1"
    }
    m_get_spot_instances.return_value = [i]
    assert get_instances_to_tag("r-1") == [i]
def test_some_tags(m_get_spot_instances):
    i = mock.Mock()
    i.tags = {"Name1": "n1", "FQDN": "fqdn1", "moz-type": "t1",
              "moz-state": "s1"}
    m_get_spot_instances.return_value = [i]
    assert get_instances_to_tag("r-1") == [i]