Python os.environ 处理环境变量

最近在写自动化代码的过程中，需要往环境变量写入一些变量，然后其他方法取出来，所以OS这个库很简单的解决了这个问题：

import os


def set_env():
    # 设置环境变量
    os.environ.setdefault('test', 'case')


def get_env():
    # 获取环境变量
    if os.environ.get('test'):
        print(os.environ['test'])
    else:
        print(None)


def del_env():
    # 删除环境变量
    del os.environ['test']

set_env()
get_env()
del_env()
get_env()