Using Python's os to walk directories again