[Python] Pitfalls: Be Careful with os.chdir

One thing you need to keep in mind is that when using os.chdir to change the working directory of current programme (script), make sure if we need to come back to the current working directory again.

Using os.chdir to change the current working directory anywhere in the programme will effect permanently, regardless if you are using it in a function or not, see the following example:

 1 import os
 2 
 3 def chdir_in_func():
 4     print "Before chaning, in func"
 5     print os.getcwd()
 6     os.chdir("/home/")
 7     print "After chaning, in func"
 8     print os.getcwd()
 9 
10 
11 if __name__ == "__main__":
12     print "Before changing, outside func"
13     print os.getcwd()
14     chdir_in_func()
15     print "After changing, outside func"
16     print os.getcwd()

The output of this script will be:

Before changing, outside func
/home/chdir_in_func
Before chaning, in func
/home/chdir_in_func
After chaning, in func
/home
After changing, outside func
/home

We can see that in the function, if we changes the directory, the current working directory is changed too.

A suggested good practice to avoid such unconsciously mistake is that alway change back to your original working directory after the actions done in the new directory, unless you know you need to stay there for next operations. I would strongly recommend that in functions, if you have to change directory, always changes it back before returning:

def func():
    cwd_save = os.getcwd()
    os.chdir(new_dir)
    # Actions in new_dir
    os.chdir(cwd_save)
    return True

 

郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。