varia/distribusi
14
0
Fork 4
Code Issues 5 Pull Requests Releases Activity

added possibility to exclude directories with --exclude-directory option

Browse Source
This commit is contained in:
RRA 2019-09-15 13:58:00 +02:00
parent 3247286a7e
commit d5335f3212
2 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
distribusi/cli.py
View File

@ -59,6 +59,12 @@ def build_argparser():
help="Removes previously made index.html. Warning: this will remove ALL index.html files in the directories listed", help="Removes previously made index.html. Warning: this will remove ALL index.html files in the directories listed",
action="store_true") action="store_true")
parser.add_argument(
'-e',
'--exclude-directory',
help="Exclude one or multiple directories from indexing",
nargs="*")
return parser return parser

10
distribusi/distribusi.py
View File

@ -70,8 +70,17 @@ def div(args, type_, subtype, tag, name):
def distribusify(args, directory): # noqa def distribusify(args, directory): # noqa
for root, dirs, files in os.walk(directory): for root, dirs, files in os.walk(directory):
if args.exclude_directory:
if args.verbose:
print('Excluding directory:', ", ".join(args.exclude_directory))
dirs[:] = [d for d in dirs if d not in args.exclude_directory]
if not args.remove_index: if not args.remove_index:
html = [] html = []
if args.verbose: if args.verbose:
print('Generating directory listing for', root) print('Generating directory listing for', root)
@ -147,6 +156,7 @@ def distribusify(args, directory): # noqa
if not args.no_template: if not args.no_template:
f.write(html_footer) f.write(html_footer)
if args.remove_index: if args.remove_index:
index = os.path.join(root, 'index.html') index = os.path.join(root, 'index.html')
if 'index.html' in files: if 'index.html' in files: