annotate utils.py @ 8:a4009bf30e70 draft default tip

Uploaded
author greg
date Thu, 06 Sep 2018 09:38:43 -0400
parents 276f0f31ddb0
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
1 import os
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
2 import shutil
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
3 import subprocess
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
4 import sys
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
5
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
6 FSTDERR = 'stderr.txt'
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
7 FSTDOUT = 'stdout.txt'
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
8
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
9
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
10 def check_execution_errors(rc, fstderr, fstdout):
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
11 if rc != 0:
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
12 fh = open(fstdout, 'rb')
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
13 out_msg = fh.read()
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
14 fh.close()
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
15 fh = open(fstderr, 'rb')
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
16 err_msg = fh.read()
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
17 fh.close()
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
18 msg = '%s\n%s\n' % (str(out_msg), str(err_msg))
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
19 stop_err(msg)
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
20
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
21
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
22 def get_response_buffers():
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
23 fstderr = os.path.join(os.getcwd(), FSTDERR)
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
24 fherr = open(fstderr, 'wb')
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
25 fstdout = os.path.join(os.getcwd(), FSTDOUT)
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
26 fhout = open(fstdout, 'wb')
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
27 return fstderr, fherr, fstdout, fhout
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
28
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
29
3
276f0f31ddb0 Uploaded
greg
parents: 0
diff changeset
30 def move_directory_files(source_dir, destination_dir, copy=False, remove_source_dir=False):
0
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
31 source_directory = os.path.abspath(source_dir)
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
32 destination_directory = os.path.abspath(destination_dir)
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
33 if not os.path.isdir(destination_directory):
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
34 os.makedirs(destination_directory)
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
35 for dir_entry in os.listdir(source_directory):
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
36 source_entry = os.path.join(source_directory, dir_entry)
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
37 if copy:
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
38 shutil.copy(source_entry, destination_directory)
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
39 else:
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
40 shutil.move(source_entry, destination_directory)
3
276f0f31ddb0 Uploaded
greg
parents: 0
diff changeset
41 if remove_source_dir:
276f0f31ddb0 Uploaded
greg
parents: 0
diff changeset
42 os.rmdir(source_directory)
0
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
43
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
44
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
45 def run_command(cmd):
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
46 fstderr, fherr, fstdout, fhout = get_response_buffers()
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
47 proc = subprocess.Popen(args=cmd, stderr=fherr, stdout=fhout, shell=True)
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
48 rc = proc.wait()
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
49 # Check results.
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
50 fherr.close()
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
51 fhout.close()
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
52 check_execution_errors(rc, fstderr, fstdout)
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
53
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
54
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
55 def stop_err(msg):
fa38de0b1f1a Uploaded
greg
parents:
diff changeset
56 sys.exit(msg)