python切换test、uat、prod环境

def environment_setup():

    # 获取环境变量，确定当前运行的环境

    env= 'test'

    # 根据环境选择不同的URL和登录凭证

    if env== 'test':

        url= 'http://test.example.com'

        username= 'test_user'

        password= 'test_password'

    elif env== 'uat':

        url= 'http://uat.example.com'

        username= 'uat_user'

        password= 'uat_password'

    elif env== 'prod':

        url= 'http://prod.example.com'

        username= 'prod_user'

        password= 'prod_password'

    else:

        raise ValueError('Invalid environment specified')