从python创建GCP实例时复制文件

from google.cloud import compute_v1

# 创建GCP实例
def create_instance(project_id, zone, instance_name, machine_type, image_family):
    compute_client = compute_v1.InstancesClient()

    project_zone = f"projects/{project_id}/zones/{zone}"
    source_image = f"projects/{project_id}/global/images/{image_family}"

    config = {
        "name": instance_name,
        "machine_type": f"zones/{zone}/machineTypes/{machine_type}",
        "disks": [
            {
                "boot": True,
                "auto_delete": True,
                "initialize_params": {
                    "source_image": source_image,
                },
            }
        ],
    }

    operation = compute_client.insert(project=project_id, zone=zone, instance_resource=config)
    operation.result()

    print(f"Instance {instance_name} created.")

# 复制文件到GCP实例
def copy_file_to_instance(project_id, zone, instance_name, source_file, destination_file):
    compute_client = compute_v1.InstancesClient()

    project_zone = f"projects/{project_id}/zones/{zone}"

    instance = compute_client.get(project=project_id, zone=zone, instance=instance_name)

    # 获取实例的外部IP地址
    external_ip = instance.network_interfaces[0].access_configs[0].nat_ip

    # 使用SCP命令复制文件到实例
    import subprocess
    subprocess.run(["scp", source_file, f"{external_ip}:{destination_file}"])

    print(f"File {source_file} copied to {instance_name}.")

# 示例用法
project_id = "your-project-id"
zone = "us-central1-a"
instance_name = "your-instance-name"
machine_type = "n1-standard-1"
image_family = "debian-10"

source_file = "path/to/source/file"
destination_file = "path/to/destination/file"

create_instance(project_id, zone, instance_name, machine_type, image_family)
copy_file_to_instance(project_id, zone, instance_name, source_file, destination_file)