Mercurial > repos > greg > plant_tribes_gene_family_classifier
diff utils.py @ 3:3e10a2683769 draft
Uploaded
| author | greg |
|---|---|
| date | Mon, 30 Oct 2017 09:52:34 -0400 |
| parents | f8603464bea7 |
| children |
line wrap: on
line diff
--- a/utils.py Thu Aug 24 13:26:35 2017 -0400 +++ b/utils.py Mon Oct 30 09:52:34 2017 -0400 @@ -27,7 +27,7 @@ return fstderr, fherr, fstdout, fhout -def move_directory_files(source_dir, destination_dir, copy=False): +def move_directory_files(source_dir, destination_dir, copy=False, remove_source_dir=False): source_directory = os.path.abspath(source_dir) destination_directory = os.path.abspath(destination_dir) if not os.path.isdir(destination_directory): @@ -38,6 +38,8 @@ shutil.copy(source_entry, destination_directory) else: shutil.move(source_entry, destination_directory) + if remove_source_dir: + os.rmdir(source_directory) def run_command(cmd): @@ -52,29 +54,3 @@ def stop_err(msg): sys.exit(msg) - - -def write_html_output(output, title, dir): - with open(output, 'w') as fh: - dir_items = sorted(os.listdir(dir)) - # Directories can only contain either files or directories, - # but not both. - if len(dir_items) > 0: - item_path = os.path.join(dir, dir_items[0]) - if os.path.isdir(item_path): - header = 'Directories' - else: - header = 'Datasets' - else: - header = '' - fh.write('<html><head><h3>%s: %d items</h3></head>\n' % (title, len(dir_items))) - fh.write('<body><p/><table cellpadding="2">\n') - fh.write('<tr><b>%s</th></b>\n' % header) - for index, fname in enumerate(dir_items): - if index % 2 == 0: - bgcolor = '#D8D8D8' - else: - bgcolor = '#FFFFFF' - link = '<a href="%s" type="text/plain">%s</a>\n' % (fname, fname) - fh.write('<tr bgcolor="%s"><td>%s</td></tr>\n' % (bgcolor, link)) - fh.write('</table></body></html>\n')