python os.environ,os.putenv,/ usr / bin / env

时间:2013-07-17 16:34:13

标签: python

我想确保os.system('env')不包含某些特定变量myname~/.bashrc中导出export myname=csj

因此,我在下面写了python代码:

import os

def print_all():
    print "os.environ['myname']=%s" % os.environ.get('myname')
    print "os.getenv('myname')=%s" % os.getenv('myname')
    os.system('env | grep myname')
    print

def delete_myname():
    if 'myname' in os.environ: os.environ.pop('myname')
    if os.getenv('myname'): os.unsetenv('myname')

print_all()

os.putenv('myname', 'csj2')
print "---------------------"
delete_myname()
print_all()

os.putenv('myname', 'csj3')
print "---------------------"
delete_myname()
print_all()

我认为同时检查os.environ['myname']os.getenv('myname') 然后删除它们(如果存在) 可以确保os.system('env | grep myname')什么也得不到。

然而,结果是:

os.environ['myname']=csj
os.getenv('myname')=csj
myname=csj

---------------------
os.environ['myname']=None
os.getenv('myname')=None

---------------------
os.environ['myname']=None
os.getenv('myname')=None
myname=csj3

我不明白为什么我仍然csj3上有os.system('env | grep myname')

2 个答案:

答案 0 :(得分:16)

来自docs

  

注意:直接调用putenv()不会改变os.environ,所以最好修改os.environ。

对于unsetenv,也有类似的警告:

  

但是,对unsetenv()的调用不会更新os.environ,所以实际上最好删除os.environ的项目。

getenv只返回os.environ的值,因为它是implementation shows,所以通过使用它,你会进入一个状态,当你查找它时似乎没有设置值来自python,而它实际上是在真实的环境中。现在我能想到的唯一方法是使用ctypes调用c getenv函数...

如果我修改您的代码以使用os.environ,而不是调用putenv / unsetenv,一切都按预期工作:

import os

def print_all():
    print "os.environ['myname']=%s" % (os.environ['myname'] if 'myname' in os.environ else "None")
    os.system('env | grep myname')
    print

def delete_myname():
    if 'myname' in os.environ: os.environ.pop('myname')

print_all()

os.environ['myname'] = 'csj2'
print "---------------------"
print_all()
delete_myname()
print_all()

os.environ['myname'] = 'csj3'
print "---------------------"
print_all()
delete_myname()
print_all()

输出:

$ myname=somevalue python2 test.py 
os.environ['myname']=somevalue
myname=somevalue

---------------------
os.environ['myname']=csj2
myname=csj2

os.environ['myname']=None

---------------------
os.environ['myname']=csj3
myname=csj3

os.environ['myname']=None

答案 1 :(得分:0)

一个好的做法可能是:

  • 删除myname环境变量(如果存在),
  • 运行您的功能
  • 在函数完成时恢复myname环境变量。

Yu可以使用此question中描述的modified_environ上下文管理器轻松完成此操作。

with modified_environ('myname'):
    call_my_function()