Function run

misc/python/materialize/cli/scratch/create.py:201–244

(args: argparse.Namespace)

def run(args: argparse.Namespace) -> None:
    extra_tags = {}
    if args.extra_tags:
        extra_tags = json.loads(args.extra_tags)
        if not isinstance(extra_tags, dict) or not all(
            isinstance(k, str) and isinstance(v, str) for k, v in extra_tags.items()
        ):
            raise RuntimeError(
                "extra-tags must be a JSON dictionary of strings to strings"
            )

    descs = load_machine_descs(
        args.machine,
        instance_type=args.instance_type,
        size_gb=args.size_gb,
    )

    check_required_vars()
    extra_tags["LaunchedBy"] = whoami()

    if args.ssh and len(descs) != 1:
        raise RuntimeError(f"Cannot use `--ssh` with {len(descs)} instances")

    if args.max_age_days <= 0:
        raise RuntimeError(f"max_age_days must be positive, got {args.max_age_days}")
    max_age = datetime.timedelta(days=args.max_age_days)

    instances = launch_cluster(
        descs,
        key_name=args.key_name,
        security_group_name=args.security_group_name,
        instance_profile=args.instance_profile,
        extra_tags=extra_tags,
        delete_after=datetime.datetime.now(datetime.timezone.utc) + max_age,
        git_rev=args.git_rev,
        extra_env={},
    )

    print("Launched instances:")
    print_instances(instances, args.output_format)

    if args.ssh:
        print(f"ssh-ing into: {instances[0].instance_id}")
        mssh(instances[0], "")