newbie : prune os.walk

[Diez B. Roggisch]

Untestet:

def foo(base, depth=2):
    for root, dirs, files in os.walk(base, True):
         if len(root.split(os.sep)) < depth:
              yield  root, dirs, files

for  root, dirs, files in foo("/tmp"):
     print root, dirs, files

-- 
Regards,

Diez B. Roggisch