Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import os
- import sys
- import argparse
- __author__ = 'aolo2'
- parser = argparse.ArgumentParser()
- parser.add_argument('-d', help='List directories only.', action="store_true")
- parser.add_argument('-o', help='Send output to FILENAME', type=str)
- args = parser.parse_args()
- def visit(path, reclvl):
- recbac = reclvl
- dirlist = ""
- try:
- dirlist = os.listdir(path)
- except OSError:
- print('Folder is not a folder')
- for name in dirlist:
- if name[0] != '.':
- if os.path.isdir(name):
- reclvl += 1
- for i in range(reclvl - 1):
- sys.stdout.write('\t')
- print('|' + name + '|')
- visit(path + '/' + name, reclvl)
- else:
- reclvl = recbac
- if not args.d:
- for i in range(reclvl):
- sys.stdout.write('\t')
- print('|-' + name)
- start = os.getcwd()
- visit(start, 0)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement