2014年7月24日 星期四

Python - 移除目錄和檔案

不管用什麼程式語言,都會遇到一個狀況,就是要刪除檔案或目錄時,沒有一個簡易型函式能一起處理檔案和目錄。沒辦法,只好自己用已有的函式拼湊一個符合目標的函式。
import os, shutil

## Remove a file or directory.
def remove(path) :
    retval = False
    try:
        if os.path.exists(path):
            if os.path.isdir(path):
                shutil.rmtree(path, ignore_errors = True)
                retval = True
            elif os.path.isfile(path):
                os.remove(path)
                retval = True
        else:
            retval = True
    except OSError as e:
        print(e)
    except Exception as e:
        print(e)
    return retval

沒有留言: