如何在一个函数中通过for循环将变量传递给另一个函数

Question

我想创建一个数组，用来收集三个变量的值。这些变量的值是从我的Azure Keyvault获取的。

下面是脚本：

CREDENTIAL = DefaultAzureCredential(exclude_shared_token_cache_credential=True, exclude_environment_credential=True, exclude_managed_identity_credential=True, exclude_visual_studio_code_credential=True)
KV_PARTIAL_NAME = "KeyVault_name-"

def get_secret(kv_client, secret_name):
  return kv_client.get_secret(secret_name).value 
  
# Subscription identification    
def environments():
  args = parse_arguments() 
  env = args.env_name
  subscription_name = env
  if env == "Prod":
     subscription_name = "Production"
  print(f"The script is running on {subscription_name}")
  azure_key_vault_env_specific_url = f"https://{KV_PARTIAL_NAME}-{env}.vault.azure.net/"
  _secret_client = SecretClient(vault_url=azure_key_vault_env_specific_url, credential=CREDENTIAL)
  configuration = [ ]
  for configuration in _secret_client:
    ELASTIC_SEARCH_USER = _secret_client.get_secret("Elastic--User").value
    ELASTIC_SEARCH_PASSWORD = _secret_client.get_secret("Elastic--Password").value
    ELASTIC_SEARCH_URL = _secret_client.get_secret("Elastic--URL").value
  return azure_key_vault_env_specific_url, configuration
  
# Provide argument -e for the script to choose the correct subscription
def parse_arguments():
  parser = argparse.ArgumentParser()
  parser.add_argument('--env_name', '-e', type=str, choices=['Dev', 'QA', 'Prod'],  help='The short for the environment for subscription', required=True)
  return parser.parse_args()
  
def indice_delete(configuration):
  elastic_auth_uri = f"https://{ELASTIC_SEARCH_URL}/security/_authenticate"
  response = requests.get(elastic_auth_uri, auth=(ELASTIC_SEARCH_USER, ELASTIC_SEARCH_PASSWORD))        
  search_url_index = "_cat/indices/"
  params_dict = {
  "h":"index,docs.count",
  "s":"docs.count:asc",
  "format":"json"
  }
  elastic_console = f"https://{ELASTIC_SEARCH_URL}/{search_url_index}"
  getRequestElasticSearch = requests.get(elastic_console, auth=(ELASTIC_SEARCH_USER, ELASTIC_SEARCH_PASSWORD), params=params_dict)
  content = json.loads(getRequestElasticSearch.text)
  elastic_console_delete = f"https://{ELASTIC_SEARCH_URL}/"
  for index in content:
   indiciesList = index
   collectdoccount = index['docs.count']
   search_int = int(collectdoccount)
   if search_int == 0:
      index_name = index['index']
      delete_url = f"{elastic_console_delete}{index_name}"
      response = requests.delete(delete_url, auth=(ELASTIC_SEARCH_USER, ELASTIC_SEARCH_PASSWORD))
      if  response.status_code == 200:
          print("index deleted -" , "index name:" , index['index'] , ",  doc.count:" , index['docs.count'] , ",  elasticsearch url index:" , delete_url)
      if  response.status_code != 200:
          print ("index not deleted -" , "index name:" , index['index'] , ",  Reason:" , response.content)
          
          
if __name__ == '__main__':
   pass_credentials = environments()
   indice_delete(pass_credentials)

目前，在environments函数中的for循环里的变量，无法在其他函数中使用，比如indice_delete。

Answer 1

你可以使用 global 这个关键词来在函数内部声明变量为全局变量。

def environments():
  for configuration in _secret_client:
    global ELASTIC_SEARCH_USER = _secret_client.get_secret("Elastic--User").value
    global ELASTIC_SEARCH_PASSWORD = _secret_client.get_secret("Elastic--Password").value
    global ELASTIC_SEARCH_URL = _secret_client.get_secret("Elastic--URL").value
  

就像这里说的

你也可以把这些变量的值返回，然后在下一个函数中使用这些返回的值。

return azure_key_vault_env_specific_url, configuration,[ELASTIC_SEARCH_USER,ELASTIC_SEARCH_PASSWORD,ELASTIC_SEARCH_URL]