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