Changes between Version 5 and Version 6 of PythonSandbox

Timestamp:

08/19/08 14:29:08 (10 years ago)

Author:

kmilner

Comment:

--

PythonSandbox

@@ -36 +36 @@
 }}}
 
-= Delete a directory in python (helpful for cleanup) =
-
-Note: this will not delete subdirs, if you want to do that then you should recursively call the function if f is a dir (there's an os method to find that out)
+= Delete a directory recursively in python (helpful for cleanup) =
 
 {{{
-items = os.listdir(tmpdir)
-for f in items:
-        if f == '.' or f == '..': continue
-        file = tmpdir + os.sep + f
-        print "Deleting " + file
-        os.remove(file)
-os.rmdir(tmpdir)
+# Delete a directory, recursively
+def delete_dir(dir):
+        items = os.listdir(dir)
+        for item in items:
+                if item == '.' or item == '..': continue
+                file = dir + os.sep + item
+                if os.path.isdir(file):
+                        # if this file is actually a dir, delete the dir
+                        delete_dir(file)
+                else: # it's just a file
+                        print "Deleting " + file  
+                        os.remove(file)
+        os.rmdir(dir)
 }}}