2

将 os.environ 重置为在命令 shell 中可以找到的默认值的 pythonic 方式是什么？我可以通过首先将 os.environ 推送到默认字典中来处理这个问题，但是如果 os.environ 在导入我的模块之前被另一个模块更改，该方法将失败。

在 Windows 中，我目前可以像这样重置值：

import os, subprocess, tempfile

def is_locked(filepath):
    ''' Needed to determine when the set command below completes
    '''
    locked = None
    file_object = None
    if os.path.exists(filepath):
        try:
            buffer_size = 8
            file_object = open(filepath, 'a', buffer_size)
            if file_object:
                locked = False
        except IOError, message:
            locked = True
        finally:
            if file_object:
                file_object.close()
    else:
        locked = True
    return locked

# Define a Windows command which would dump default env variables into a tempfile
#
# - start /i will give default cmd.exe environment variable to what follows.
#   It's the only way i found to get cmd.exe's default env variables and because
#   start terminates immediately i need to use is_locked defined above to wait for
#   Window's set command to finish dumping it's variables into the temp file
# - /b is will prevent a command shell to pop up
temp = tempfile.mktemp()
cmd = 'start /i /b cmd /c set>%s'%temp
p = subprocess.Popen(cmd, shell=True)
p.wait()

# Wait for the command to complete and parse the tempfile
data = []
while(is_locked(temp)):
    pass

with open(temp,'r') as file:
    data = file.readlines()
os.remove(temp)

# defaults will contain default env variables
defaults = dict()
for env in data:
    env = env.strip().split('=')
    defaults[env[0]]=env[1]

    print '%s %s'%(env[0],env[1])


os.environ = dict(defaults)

我的方式目前在 python 2.7.3 64 位中有效，但我只是注意到当我在 32 位中运行它时，PROGRAMFILES 和 PROGRAMFILES(x86) env 变量都指向“程序文件 (x86)”，这是这里讨论的一个问题.

在此先感谢您的帮助！

4

2 回答 2

0

我想知道您的 hack 似乎有效，因为孩子继承父母的当前环境是设计使然。所以我认为除了在应用程序启动时保存环境之外别无选择。但是，这不一定是从开始菜单启动的 shell 的环境，因为您的父母也可以操纵它。

做以下实验：启动一个资源管理器，从中运行一个 shell 并调查环境。然后直接从开始菜单启动一个shell并再次检查。现在转到控制面板-> 系统并添加一个变量。重新启动两个 shell（但不是资源管理器）并注意区别。

要获得“root”环境（在登录时创建的环境），您可以在 HKCU/Environment 读取密钥。但这也可能是hacky。

于 2013-10-25T05:09:39.240 回答
0

您可能需要类似于此问题modified_environ中描述的上下文管理器来设置/恢复环境变量。

我建议你用这个上下文管理器包装你的主要功能：

with modified_environ():
    main()

这样，一切都将在功能完成时恢复。

您还可以包装导入：

with modified_environ():
    import pkg.which.can.change.env
    main()
于 2017-05-10T20:44:26.060 回答