पायथन - मैं एक सजावट कैसे लिखूं जो सीडब्ल्यूडी को पुनर्स्थापित करता है?

Question

मैं एक सजावट कैसे लिखूं जो वर्तमान कार्यशील निर्देशिका को पुनर्स्थापित करता है जो सजाए गए समारोह से पहले था? दूसरे शब्दों में, अगर मैं एक समारोह पर सजावट का उपयोग करता हूं जो os.chdir() करता है, तो फ़ंक्शन कहने के बाद cwd बदला नहीं जाएगा।

Answer 1

path.py मॉड्यूल (जो तुम सच में अगर अजगर स्क्रिप्ट में रास्तों के साथ काम का उपयोग करना चाहिए) एक संदर्भ प्रबंधक है:

subdir = d/'subdir' #subdir is a path object, in the path.py module 
with subdir: 
    # here current dir is subdir 

#not anymore 

(क्रेडिट this blog post पर रोब से जाता है erto Alsina)

Answer 2

def preserve_cwd(function): 
    def decorator(*args, **kwargs): 
     cwd = os.getcwd() 
     result = function(*args, **kwargs) 
     os.chdir(cwd) 
     return result 
    return decorator 

यहाँ उसके उपयोग के तरीके:

@preserve_cwd 
def test(): 
    print 'was:',os.getcwd() 
    os.chdir('/') 
    print 'now:',os.getcwd() 

>>> print os.getcwd() 
/Users/dspitzer 
>>> test() 
was: /Users/dspitzer 
now:/
>>> print os.getcwd() 
/Users/dspitzer 

Answer 3

एक डेकोरेटर के लिए इस सवाल का जवाब दिया गया है; यह अनुरोध के अनुसार फ़ंक्शन परिभाषा चरण पर काम करता है।

अजगर 2.5+ के साथ

, आप भी ऐसा करने के लिए समारोह कॉल चरण में एक संदर्भ प्रबंधक का प्रयोग पर एक विकल्प है:

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() 

यह एक अच्छा विकल्प है।

संपादित करें: मैंने कोडेप द्वारा सुझाए गए त्रुटि प्रबंधन को जोड़ा। चूंकि मेरा जवाब चुना गया है, इसलिए एक पूर्ण उत्तर, अन्य सभी मुद्दों को अलग करना उचित है।

Answer 4

दिए गए उत्तरों को ध्यान में रखना विफल रहता है कि लपेटा हुआ कार्य अपवाद बढ़ा सकता है। उस स्थिति में, निर्देशिका को कभी भी बहाल नहीं किया जाएगा। नीचे दिया गया कोड पिछले उत्तरों में अपवाद हैंडलिंग जोड़ता है।

एक डेकोरेटर के रूप में:

def preserve_cwd(function): 
    @functools.wraps(function) 
    def decorator(*args, **kwargs): 
     cwd = os.getcwd() 
     try: 
      return function(*args, **kwargs) 
     finally: 
      os.chdir(cwd) 
    return decorator 

और एक संदर्भ प्रबंधक के रूप में:

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