如何使用Python从K8s集群内部的Pod中获取当前集群名称

方法1：从Pod内部获取集群名称 #

Kubernetes默认不会将集群名称注入到Pod的运行环境中，AKS也没有默认提供该内置变量，你可以通过两种方式拿到集群名称：

• 调用AKS实例元数据服务（IMDS）获取
  AKS节点默认运行元数据服务，Pod内可以直接调用该接口拿到所属集群名称，无需额外配置权限，Python示例代码如下：

  import requests
  
  def get_aks_cluster_name():
      url = "http://169.254.169.254/metadata/instance/compute/aksClusterName?api-version=2021-02-01&format=text"
      headers = {"Metadata": "true"}
      try:
          resp = requests.get(url, headers=headers, timeout=5)
          resp.raise_for_status()
          return resp.text
      except Exception as e:
          # 本地开发环境调用失败返回默认值
          return "dev-my-cluster"
  

• 部署阶段注入集群名称到环境变量
  你可以在CI/CD流水线或者Helm部署模板中，提前配置不同集群对应的集群名称参数，部署时直接注入到Pod的环境变量中，Python代码直接读取即可：

  import os
  cluster_name = os.getenv("CLUSTER_NAME", "dev-my-cluster")
  

更优的实现方案 #

不建议通过集群名称转换运行环境的方案，更推荐直接在部署阶段注入环境标识变量，方案优势和实现如下：

• 代码逻辑更简洁，不需要处理集群名称匹配、接口调用异常等问题，直接读取环境变量即可
• 扩展性更强，后续新增集群无需修改代码里的集群名匹配规则，直接新增环境标识即可
• 安全性更高，不需要调用元数据接口或者集群API，没有额外权限依赖

实现示例：

1. 在Deployment配置中注入环境变量，不同集群部署时替换对应值即可：

spec:
  containers:
  - name: your-app
    image: your-app-image:v1
    env:
    - name: ENV_TYPE
      value: "stg" # 生产集群部署时改为prod

2. Python代码直接读取变量加载对应配置：

import os

env_type = os.getenv("ENV_TYPE", "dev")
if env_type == "stg":
    # 加载测试环境配置
    config = load_stg_config()
elif env_type == "prod":
    # 加载生产环境配置
    config = load_prod_config()
else:
    # 加载本地开发配置
    config = load_dev_config()

内容的提问来源于stack exchange，提问作者torxx666