# HG changeset patch
# User greg
# Date 1495480761 14400
# Node ID f61d568666f1f7e03fa6046abe63efe8e6ceeda2
# Parent 14af615b80b4b81271e1003295b853a207e6a80e
Uploaded
diff -r 14af615b80b4 -r f61d568666f1 utils.py
--- a/utils.py Mon May 22 15:19:10 2017 -0400
+++ b/utils.py Mon May 22 15:19:21 2017 -0400
@@ -56,18 +56,21 @@
def write_html_output(output, title, dir):
with open(output, 'w') as fh:
- fh.write('%s: %d files
\n' % (title, len(os.listdir(dir))))
+ dir_items = sorted(os.listdir(dir))
+ # Directories can only contain either files or directories, but not both.
+ item_path = os.path.join(dir, dir_items[0])
+ if os.path.isdir(item_path):
+ header = 'Directories'
+ else:
+ header = 'Datasets'
+ fh.write('%s: %d items
\n' % (title, len(dir_items)))
fh.write('\n')
- fh.write('| Size | Name |
|---|
\n')
- for index, fname in enumerate(sorted(os.listdir(dir))):
+ fh.write('| %s |
|---|
\n' % header)
+ for index, fname in enumerate(dir_items):
if index % 2 == 0:
bgcolor = '#D8D8D8'
else:
bgcolor = '#FFFFFF'
- try:
- size = str(os.path.getsize(os.path.join(dir, fname)))
- except:
- size = 'unknown'
link = '%s\n' % (fname, fname)
- fh.write('| %s | %s |
\n' % (bgcolor, size, link))
+ fh.write('| %s |
\n' % (bgcolor, link))
fh.write('
\n')