How do I write a decorator that restores the cwd?

by

The answer for a decorator has been given; it works at the function definition stage as requested.

With Python 2.5+, you also have an option to do that at the function call stage using a context manager:

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)

which can be used if needed at the function call time as:

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

It’s a nice option to have.

EDIT: I added error handling as suggested by codeape. Since my answer has been voted up, it’s fair to offer a complete answer, all other issues aside.

More Related Contents:

  1. How to make function decorators and chain them together?
  2. Creating a singleton in Python
  3. Decorators with parameters?
  4. What does functools.wraps do?
  5. Preserving signatures of decorated functions
  6. How to decorate a class?
  7. How to decorate all functions of a class without typing it over and over for each method? [duplicate]
  8. How can I separate the functions of a class into multiple files?
  9. Attaching a decorator to all functions within a class
  10. Decorator execution order
  11. Can a decorator of an instance method access the class?
  12. How to create a decorator that can be used either with or without parameters?
  13. Python decorator handling docstrings
  14. How to inject variable into scope with a decorator?
  15. How to decorate a method inside a class?
  16. Python Class Based Decorator with parameters that can decorate a method or a function
  17. What is the best way to do automatic attribute assignment in Python, and is it a good idea?
  18. How do I make function decorators and chain them together?
  19. How to make function decorators and chain them together in Python?
  20. How to do a conditional decorator in python?
  21. How can I pass a variable in a decorator to function’s argument in a decorated function?
  22. Using python decorator with or without parentheses
  23. How to create a custom decorator in Django?
  24. Currying decorator in python
  25. Python functools.wraps equivalent for classes
  26. How to auto register a class when it’s defined
  27. Auto-register class methods using decorator
  28. Applying a decorator to every method in a class?
  29. Decorator error: NoneType object is not callable
  30. How to skip a pytest using an external fixture?

Leave a Comment