如何在python代码中设置环境变量

时间:2018-08-06 19:32:53

标签: python linux

我最近开始用python编码。我正在尝试创建一个环境变量,并使用python为其分配列表。因此,当我尝试通过printenv之类的命令行读取环境变量时,它将在此处列出。

这是我在python中的代码:

from API_CALLS import Post_Request as Request
import os

class VTM_Config:

    @staticmethod
    def validate_pool_nodes(url, headers, expected_num_of_active_nodes):
        try:
            print('\nNow Executing Validate VTM Configs...\n')
            # validate that vtm api works by sending a get_session_with_ssl call to the url
            vtm_get_session_response = Request.get_session_with_ssl(url=url, headers=headers)
            data = vtm_get_session_response
            active_nodes = [
                n['node']
                for n in data['properties']['basic']['nodes_table']
                if n['state'] == 'active'

            ]
            actual_num_of_active_nodes = len(active_nodes)
            if expected_num_of_active_nodes != actual_num_of_active_nodes:
                print("Number of Active Nodes = {}".format(actual_num_of_active_nodes))
                raise Exception("ERROR: You are expecting : {} nodes, but this pool contains {} nodes".format(
                    expected_num_of_active_nodes, actual_num_of_active_nodes))
            else:
                print("Number of Active Nodes = {}\n".format(actual_num_of_active_nodes))
            print("Active servers : {}\n".format(active_nodes))
            os.environ["ENABLED_POOL_NODES"] = active_nodes
            return os.environ["ENABLED_POOL_NODES"]

        except Exception as ex:
            raise ex

我正在尝试使用os.environ["ENABLED_POOL_NODES"] = active_nodes创建一个环境变量并试图返回它。

运行此代码时,出现如下错误: 引发TypeError(“ str是预期的,不是%s%类型(值)。名称) TypeError:str期望,而不是列表。

问题: 如何将列表分配给环境变量。

2 个答案:

答案 0 :(得分:1)

正如@Jean-Francois Fabre在上面的评论中指出的那样,这可能不是解决您要解决的问题的最佳方法。但是,要在标题和帖子的最后一行回答问题:

  

问题:如何将列表分配给环境变量。

您不能直接将列表分配给环境变量。这些本质上是字符串值,因此您需要将列表转换为某种方式的字符串。如果您只需要将整个内容传回,则可以执行以下操作:

os.envrion["ENABLED_POOL_NODES"] = str(active_nodes)

这只会将列表转换为字符串,看起来像:“ ['a', 'b', 'c']”。根据对下游env变量的处理方式,您可能需要以不同的方式处理它。

答案 1 :(得分:0)

因此,感谢大家,这是一个简单的解决方案。我最终只是返回字符串值并打印到控制台,在该控制台中,我的jenkins作业中的shell脚本将获得输出:

def validate_pool_nodes(url, headers, expected_num_of_active_nodes):
        try:
            print('\nNow Executing Validate VTM Configs...\n', file=sys.stderr)
            # validate that vtm api works by sending a get_session_with_ssl call to the url
            vtm_get_session_response = Request.get_session_with_ssl(url=url, headers=headers)
            data = vtm_get_session_response
            active_nodes = {
                n['node']
                for n in data['properties']['basic']['nodes_table']
                if n['state'] == 'active'
            }
            actual_num_of_active_nodes = len(active_nodes)
            if expected_num_of_active_nodes != actual_num_of_active_nodes:
                print("Number of Active Nodes = {}".format(actual_num_of_active_nodes), file=sys.stderr)
                raise Exception("ERROR: You are expecting : {} nodes, but this pool contains {} nodes".format(
                    expected_num_of_active_nodes, actual_num_of_active_nodes))
            else:
                print("Number of Active Nodes = {}\n".format(actual_num_of_active_nodes), file=sys.stderr)

            return str(active_nodes)

        except Exception as ex:
            raise ex

这是“主要” python方法:

if __name__ == '__main__':
    arg1 = sys.argv[1]
    arg2 = int(sys.argv[2])
    run_prereq = Prerequisites()
    run_prereq.validate_login_redirect(pool_arg=arg1)
    nodes_list = run_prereq.validate_pool_nodes(pool_arg=arg1, num__of_nodes_arg=arg2)
    sys.stdout.write(nodes_list)
    sys.exit(0)