从自定义预测中获取 运行 项目 ID 类

Get running project id from custom prediction classes

我正在寻找获取 GCP 项目 ID 的方法,运行 AI Platform Custom Prediction

我的项目有 3 个项目 dev、stage 和 prod,每个项目都必须从自定义预测内部访问唯一的 URL。自定义预测代码由单个项目控制。

我试过 gcloud 命令 运行 subprocess 但它对我不起作用。

我发现project number和project id有关系,os.environ.get("consumer_project_number")可以在Custom Prediction中得到一个project number的值

这是我获取项目 ID 的最终代码。

if consumer_project_number == "hoge_1":
    return "prd_ID"
elif consumer_project_number == "hoge_2":
    return "stg_ID"
elif consumer_project_number == "hoge_3":
    return "dev_ID"
else:
    raise ValueError("None consumer_project_number detected.")

我有同样的问题,我需要项目名称,使用以下代码

PROJECT_ID = ast.literal_eval(os.getenv('create_version_request')).get('parent').split('/', 2)[1]
print(f'PROJECT_ID={PROJECT_ID}', flush=True)