Question

这是我基本上要做的事情：

import sh, os

with sh.cd('/tmp'):
  print os.getcwd()

print os.getcwd()

我收到以下错误

line 3, in <module>
    with sh.cd('/tmp'):
AttributeError: __exit__

我在这里缺少什么？是否有替代解决方案来更改上下文中的目录？

Answer 1

你不能只使用任何类/函数作为上下文管理器，它必须实际上以这种方式显式实现，使用函数上的contextlib.contextmanager装饰器，或者在类的情况下，通过定义__enter__ and __exit__实例方法。

您正在使用的sh.cd函数只是os.chdir的包装：

>>> import sh
>>> sh.cd
<bound method Environment.b_cd of {}>

b_cd定义为：

def b_cd(self, path):
    os.chdir(path)

正如你所看到的，它只是一个正常的功能;它不能用作上下文管理器。

提供的链接whereswalden显示了实现您想要作为一个类的行为的好方法。它可以类似地实现为这样的函数：

import contextlib
import os

@contextlib.contextmanager
def cd(path):
   old_path = os.getcwd()
   os.chdir(path)
   try:
       yield
   finally:
       os.chdir(old_path)

样本用法：

print(os.getcwd())
with cd("/"):
    print os.getcwd()
print(os.getcwd())

输出：

'/home/dan'
'/'
'/home/dan'

sh.cd使用上下文管理器

1 个答案: