gpt4 book ai didi

python - 如何编写恢复 cwd 的装饰器?

转载 作者:IT老高 更新时间:2023-10-28 22:12:59 28 4
gpt4 key购买 nike

如何编写一个装饰器,将当前工作目录恢复到调用装饰函数之前的状态?换句话说,如果我在执行 os.chdir() 的函数上使用装饰器,则调用该函数后 cwd 不会更改。

最佳答案

已经给出了装饰者的答案;它按要求在函数定义阶段工作。

使用 Python 2.5+,您还可以选择在函数 call 阶段使用上下文管理器执行此操作:

from __future__ import with_statement # needed for 2.5 ≤ Python < 2.6
import contextlib, os

@contextlib.contextmanager
def remember_cwd():
curdir= os.getcwd()
try: yield
finally: os.chdir(curdir)

如果需要,可以在函数调用时使用:

print "getcwd before:", os.getcwd()
with remember_cwd():
walk_around_the_filesystem()
print "getcwd after:", os.getcwd()

这是一个不错的选择。

编辑:我按照 codeape 的建议添加了错误处理。由于我的答案已被投票,因此提供一个完整的答案是公平的,所有其他问题除外。

关于python - 如何编写恢复 cwd 的装饰器?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/169070/

28 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com