annotate utils.py @ 7:529f4635ddb4 draft

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