Advertisement
Guest User

Untitled

a guest
May 26th, 2015
232
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.00 KB | None | 0 0
  1. import os
  2. import sys
  3. import argparse
  4.  
  5. __author__ = 'aolo2'
  6.  
  7. parser = argparse.ArgumentParser()
  8. parser.add_argument('-d', help='List directories only.', action="store_true")
  9. parser.add_argument('-o', help='Send output to FILENAME', type=str)
  10. args = parser.parse_args()
  11.  
  12.  
  13. def visit(path, reclvl):
  14.     recbac = reclvl
  15.     dirlist = ""
  16.  
  17.     try:
  18.         dirlist = os.listdir(path)
  19.     except OSError:
  20.         print('Folder is not a folder')
  21.  
  22.     for name in dirlist:
  23.         if name[0] != '.':
  24.             if os.path.isdir(name):
  25.                 reclvl += 1
  26.  
  27.                 for i in range(reclvl - 1):
  28.                     sys.stdout.write('\t')
  29.  
  30.                 print('|' + name + '|')
  31.                 visit(path + '/' + name, reclvl)
  32.  
  33.             else:
  34.                 reclvl = recbac
  35.                 if not args.d:
  36.                     for i in range(reclvl):
  37.                         sys.stdout.write('\t')
  38.                     print('|-' + name)
  39.  
  40.  
  41. start = os.getcwd()
  42. visit(start, 0)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement