最近在写自动化代码的过程中,需要往环境变量写入一些变量,然后其他方法取出来,所以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()