annotate utils.py @ 44:58382c8ed01c draft

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