annotate utils.py @ 131:656614635ebf draft

Uploaded
author greg
date Thu, 06 Apr 2017 14:21:39 -0400
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
131
656614635ebf Uploaded
greg
parents:
diff changeset
1 import os
656614635ebf Uploaded
greg
parents:
diff changeset
2 import shutil
656614635ebf Uploaded
greg
parents:
diff changeset
3 import sys
656614635ebf Uploaded
greg
parents:
diff changeset
4
656614635ebf Uploaded
greg
parents:
diff changeset
5
656614635ebf Uploaded
greg
parents:
diff changeset
6 def check_execution_errors(rc, stderr):
656614635ebf Uploaded
greg
parents:
diff changeset
7 if rc != 0:
656614635ebf Uploaded
greg
parents:
diff changeset
8 stop_err(stderr.read())
656614635ebf Uploaded
greg
parents:
diff changeset
9
656614635ebf Uploaded
greg
parents:
diff changeset
10
656614635ebf Uploaded
greg
parents:
diff changeset
11 def move_directory_files(source_dir, destination_dir):
656614635ebf Uploaded
greg
parents:
diff changeset
12 source_directory = os.path.abspath(source_dir)
656614635ebf Uploaded
greg
parents:
diff changeset
13 destination_directory = os.path.abspath(destination_dir)
656614635ebf Uploaded
greg
parents:
diff changeset
14 if not os.path.isdir(destination_directory):
656614635ebf Uploaded
greg
parents:
diff changeset
15 os.makedirs(destination_directory)
656614635ebf Uploaded
greg
parents:
diff changeset
16 for dir_entry in os.listdir(source_directory):
656614635ebf Uploaded
greg
parents:
diff changeset
17 source_entry = os.path.join(source_directory, dir_entry)
656614635ebf Uploaded
greg
parents:
diff changeset
18 shutil.move(source_entry, destination_directory)
656614635ebf Uploaded
greg
parents:
diff changeset
19
656614635ebf Uploaded
greg
parents:
diff changeset
20
656614635ebf Uploaded
greg
parents:
diff changeset
21 def stop_err(msg):
656614635ebf Uploaded
greg
parents:
diff changeset
22 sys.stderr.write(msg)
656614635ebf Uploaded
greg
parents:
diff changeset
23 sys.exit(1)
656614635ebf Uploaded
greg
parents:
diff changeset
24
656614635ebf Uploaded
greg
parents:
diff changeset
25
656614635ebf Uploaded
greg
parents:
diff changeset
26 def write_html_output(output, title, dir):
656614635ebf Uploaded
greg
parents:
diff changeset
27 with open(output, 'w') as fh:
656614635ebf Uploaded
greg
parents:
diff changeset
28 fh.write('<html><head><h3>%s</h3></head>\n' % title)
656614635ebf Uploaded
greg
parents:
diff changeset
29 fh.write('<body><p/><table cellpadding="2">\n')
656614635ebf Uploaded
greg
parents:
diff changeset
30 fh.write('<tr><th>Size</th><th>Name</th></tr>\n')
656614635ebf Uploaded
greg
parents:
diff changeset
31 for index, fname in enumerate(sorted(os.listdir(dir))):
656614635ebf Uploaded
greg
parents:
diff changeset
32 if index % 2 == 0:
656614635ebf Uploaded
greg
parents:
diff changeset
33 bgcolor = '#D8D8D8'
656614635ebf Uploaded
greg
parents:
diff changeset
34 else:
656614635ebf Uploaded
greg
parents:
diff changeset
35 bgcolor = '#FFFFFF'
656614635ebf Uploaded
greg
parents:
diff changeset
36 try:
656614635ebf Uploaded
greg
parents:
diff changeset
37 size = str(os.path.getsize(os.path.join(dir, fname)))
656614635ebf Uploaded
greg
parents:
diff changeset
38 except:
656614635ebf Uploaded
greg
parents:
diff changeset
39 size = 'unknown'
656614635ebf Uploaded
greg
parents:
diff changeset
40 link = '<a href="%s" type="text/plain">%s</a>\n' % (fname, fname)
656614635ebf Uploaded
greg
parents:
diff changeset
41 fh.write('<tr bgcolor="%s"><td>%s</td><td>%s</td></tr>\n' % (bgcolor, size, link))
656614635ebf Uploaded
greg
parents:
diff changeset
42 fh.write('</table></body></html>\n')