changeset 31:5691bc0a32d2 draft

Uploaded
author greg
date Mon, 15 May 2017 08:26:23 -0400
parents 3b80ba2c0fd1
children 14af615b80b4
files utils.py
diffstat 1 files changed, 6 insertions(+), 3 deletions(-) [+]
line wrap: on
line diff
--- a/utils.py	Tue May 09 13:02:31 2017 -0400
+++ b/utils.py	Mon May 15 08:26:23 2017 -0400
@@ -27,14 +27,17 @@
     return fstderr, fherr, fstdout, fhout
 
 
-def move_directory_files(source_dir, destination_dir):
+def move_directory_files(source_dir, destination_dir, copy=False):
     source_directory = os.path.abspath(source_dir)
     destination_directory = os.path.abspath(destination_dir)
     if not os.path.isdir(destination_directory):
         os.makedirs(destination_directory)
     for dir_entry in os.listdir(source_directory):
         source_entry = os.path.join(source_directory, dir_entry)
-        shutil.move(source_entry, destination_directory)
+        if copy:
+            shutil.copy(source_entry, destination_directory)
+        else:
+            shutil.move(source_entry, destination_directory)
 
 
 def run_command(cmd):
@@ -53,7 +56,7 @@
 
 def write_html_output(output, title, dir):
     with open(output, 'w') as fh:
-        fh.write('<html><head><h3>%s</h3></head>\n' % title)
+        fh.write('<html><head><h3>%s: %d files</h3></head>\n' % (title, len(os.listdir(dir))))
         fh.write('<body><p/><table cellpadding="2">\n')
         fh.write('<tr><th>Size</th><th>Name</th></tr>\n')
         for index, fname in enumerate(sorted(os.listdir(dir))):