annotate utils.py @ 15:eb3eb139be22 draft

Uploaded
author greg
date Fri, 05 May 2017 09:33:55 -0400
parents f4361d941aa3
children a9d8e2ee41f6
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
f4361d941aa3 Uploaded
greg
parents:
diff changeset
1 import os
f4361d941aa3 Uploaded
greg
parents:
diff changeset
2 import shutil
15
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
3 import subprocess
0
f4361d941aa3 Uploaded
greg
parents:
diff changeset
4 import sys
f4361d941aa3 Uploaded
greg
parents:
diff changeset
5
15
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
6 FSTDERR = 'stderr.txt'
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
7 FSTDOUT = 'stdout.txt'
0
f4361d941aa3 Uploaded
greg
parents:
diff changeset
8
15
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
9
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
10 def check_execution_errors(rc, fstderr, fstdout):
0
f4361d941aa3 Uploaded
greg
parents:
diff changeset
11 if rc != 0:
15
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
12 fh = open(fstdout, 'rb')
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
13 out_msg = fh.read()
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
14 fh.close()
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
15 fh = open(fstderr, 'rb')
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
16 err_msg = fh.read()
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
17 fh.close()
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
18 msg = '%s\n%s\n' % (str(out_msg), str(err_msg))
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
19 stop_err(msg)
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
20
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
21
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
22 def get_response_buffers():
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
23 fstderr = os.path.join(os.getcwd(), FSTDERR)
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
24 fherr = open(fstderr, 'wb')
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
25 fstdout = os.path.join(os.getcwd(), FSTDOUT)
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
26 fhout = open(fstdout, 'wb')
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
27 return fherr, fhout
0
f4361d941aa3 Uploaded
greg
parents:
diff changeset
28
f4361d941aa3 Uploaded
greg
parents:
diff changeset
29
f4361d941aa3 Uploaded
greg
parents:
diff changeset
30 def move_directory_files(source_dir, destination_dir):
f4361d941aa3 Uploaded
greg
parents:
diff changeset
31 source_directory = os.path.abspath(source_dir)
f4361d941aa3 Uploaded
greg
parents:
diff changeset
32 destination_directory = os.path.abspath(destination_dir)
f4361d941aa3 Uploaded
greg
parents:
diff changeset
33 if not os.path.isdir(destination_directory):
f4361d941aa3 Uploaded
greg
parents:
diff changeset
34 os.makedirs(destination_directory)
f4361d941aa3 Uploaded
greg
parents:
diff changeset
35 for dir_entry in os.listdir(source_directory):
f4361d941aa3 Uploaded
greg
parents:
diff changeset
36 source_entry = os.path.join(source_directory, dir_entry)
f4361d941aa3 Uploaded
greg
parents:
diff changeset
37 shutil.move(source_entry, destination_directory)
f4361d941aa3 Uploaded
greg
parents:
diff changeset
38
f4361d941aa3 Uploaded
greg
parents:
diff changeset
39
15
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
40 def run_command(cmd):
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
41 fherr, fhout = get_response_buffers()
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
42 proc = subprocess.Popen(args=cmd, stderr=fherr, stdout=fhout, shell=True)
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
43 rc = proc.wait()
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
44 # Check results.
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
45 fherr.close()
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
46 fhout.close()
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
47 check_execution_errors(rc, fstderr, fstdout)
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
48
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
49
0
f4361d941aa3 Uploaded
greg
parents:
diff changeset
50 def stop_err(msg):
15
eb3eb139be22 Uploaded
greg
parents: 0
diff changeset
51 sys.exit(msg)
0
f4361d941aa3 Uploaded
greg
parents:
diff changeset
52
f4361d941aa3 Uploaded
greg
parents:
diff changeset
53
f4361d941aa3 Uploaded
greg
parents:
diff changeset
54 def write_html_output(output, title, dir):
f4361d941aa3 Uploaded
greg
parents:
diff changeset
55 with open(output, 'w') as fh:
f4361d941aa3 Uploaded
greg
parents:
diff changeset
56 fh.write('<html><head><h3>%s</h3></head>\n' % title)
f4361d941aa3 Uploaded
greg
parents:
diff changeset
57 fh.write('<body><p/><table cellpadding="2">\n')
f4361d941aa3 Uploaded
greg
parents:
diff changeset
58 fh.write('<tr><th>Size</th><th>Name</th></tr>\n')
f4361d941aa3 Uploaded
greg
parents:
diff changeset
59 for index, fname in enumerate(sorted(os.listdir(dir))):
f4361d941aa3 Uploaded
greg
parents:
diff changeset
60 if index % 2 == 0:
f4361d941aa3 Uploaded
greg
parents:
diff changeset
61 bgcolor = '#D8D8D8'
f4361d941aa3 Uploaded
greg
parents:
diff changeset
62 else:
f4361d941aa3 Uploaded
greg
parents:
diff changeset
63 bgcolor = '#FFFFFF'
f4361d941aa3 Uploaded
greg
parents:
diff changeset
64 try:
f4361d941aa3 Uploaded
greg
parents:
diff changeset
65 size = str(os.path.getsize(os.path.join(dir, fname)))
f4361d941aa3 Uploaded
greg
parents:
diff changeset
66 except:
f4361d941aa3 Uploaded
greg
parents:
diff changeset
67 size = 'unknown'
f4361d941aa3 Uploaded
greg
parents:
diff changeset
68 link = '<a href="%s" type="text/plain">%s</a>\n' % (fname, fname)
f4361d941aa3 Uploaded
greg
parents:
diff changeset
69 fh.write('<tr bgcolor="%s"><td>%s</td><td>%s</td></tr>\n' % (bgcolor, size, link))
f4361d941aa3 Uploaded
greg
parents:
diff changeset
70 fh.write('</table></body></html>\n')