Sample logic about how to set variable on Azure DevOps

set_env_variable.py

import os
import sys

branch_name: str = sys.argv[1]
print(f"Received argument: {branch_name}")
env_image_type = "ENV_IMAGE_TYPE"


def set_value(key, value):
    print(f"Set key {key} as {value}")
    print(f"##vso[task.setvariable variable={key};]{value}")


if "develop" in branch_name:
    set_value(env_image_type, "dev")
elif "release" in branch_name:
    set_value(env_image_type, "qa")
elif "master" in branch_name:
    set_value(env_image_type, "prd")