Created
August 22, 2019 07:37
-
-
Save thepurpleowl/cf2b4b0caf59c2f0ab36785f40f334ca to your computer and use it in GitHub Desktop.
File structure tree
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# Written by Doug Dahms | |
# | |
# Prints the tree structure for the path specified on the command line | |
from os import listdir, sep | |
from os.path import abspath, basename, isdir | |
from sys import argv | |
def tree(dir, padding, print_files=False): | |
print (padding[:-1] + '+-' + basename(abspath(dir)) + '/') | |
padding = padding + ' ' | |
files = [] | |
if print_files: | |
files = listdir(dir) | |
else: | |
files = [x for x in listdir(dir) if isdir(dir + sep + x)] | |
count = 0 | |
for file in files: | |
count += 1 | |
print (padding + '|') | |
path = dir + sep + file | |
if isdir(path): | |
if count == len(files): | |
tree(path, padding + ' ', print_files) | |
else: | |
tree(path, padding + '|', print_files) | |
else: | |
print (padding + '+-' + file) | |
def usage(): | |
return '''Usage: %s [-f] <PATH> | |
Print tree structure of path specified. | |
Options: | |
-f Print files as well as directories | |
PATH Path to process''' % basename(argv[0]) | |
def main(): | |
if len(argv) == 1: | |
print (usage()) | |
elif len(argv) == 2: | |
# print just directories | |
path = argv[1] | |
if isdir(path): | |
tree(path, ' ') | |
else: | |
print ('ERROR: \'' + path + '\' is not a directory') | |
elif len(argv) == 3 and argv[1] == '-f': | |
# print directories and files | |
path = argv[2] | |
if isdir(path): | |
tree(path, ' ', True) | |
else: | |
print ('ERROR: \'' + path + '\' is not a directory') | |
else: | |
print (usage()) | |
if __name__ == '__main__': | |
main() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment