Mercurial > repos > galaxyp > utilities

comparison pepxml_viewer_wrapper.py @ 0:9156a440afed draft default tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Improved some datatype handling
author galaxyp
date Thu, 20 Jun 2013 11:09:24 -0400
parents
children
comparison
equal deleted inserted replaced
-1:000000000000 0:9156a440afed
1 #!/usr/bin/env python
2 import optparse
3 import os
4 import sys
5 import tempfile
6 import shutil
7 import subprocess
8 import re
9 from os.path import basename
10 import logging
11
12 assert sys.version_info[:2] >= ( 2, 6 )
13
14 log = logging.getLogger(__name__)
15 working_directory = os.getcwd()
16 tmp_stderr_name = tempfile.NamedTemporaryFile(dir = working_directory, suffix = '.stderr').name
17 tmp_stdout_name = tempfile.NamedTemporaryFile(dir = working_directory, suffix = '.stdout').name
18
19 def stop_err( msg ):
20 sys.stderr.write( "%s\n" % msg )
21 sys.exit()
22
23 def read_stderr():
24 stderr = ''
25 if(os.path.exists(tmp_stderr_name)):
26 with open(tmp_stderr_name, 'rb') as tmp_stderr:
27 buffsize = 1048576
28 try:
29 while True:
30 stderr += tmp_stderr.read(buffsize)
31 if not stderr or len(stderr) % buffsize != 0:
32 break
33 except OverflowError:
34 pass
35 return stderr
36
37 def execute(command, stdin=None):
38 with open(tmp_stderr_name, 'wb') as tmp_stderr:
39 with open(tmp_stdout_name, 'wb') as tmp_stdout:
40 proc = subprocess.Popen(args=command, shell=True, stderr=tmp_stderr.fileno(), stdout=tmp_stdout.fileno(), stdin=stdin, env=os.environ)
41 returncode = proc.wait()
42 if returncode != 0:
43 raise Exception, "Program returned with non-zero exit code %d. stderr: %s" % (returncode, read_stderr())
44
45 def delete_file(path):
46 if os.path.exists(path):
47 try:
48 os.remove(path)
49 except:
50 pass
51
52 def delete_directory(directory):
53 if os.path.exists(directory):
54 try:
55 shutil.rmtree(directory)
56 except:
57 pass
58
59 def symlink(source, link_name):
60 import platform
61 if platform.system() == 'Windows':
62 import win32file
63 win32file.CreateSymbolicLink(source, link_name, 1)
64 else:
65 os.symlink(source, link_name)
66
67
68 def copy_to_working_directory(data_file, relative_path):
69 if os.path.abspath(data_file) != os.path.abspath(relative_path):
70 shutil.copy(data_file, relative_path)
71 return relative_path
72
73 def __main__():
74 run_script()
75
76 #ENDTEMPLATE
77
78
79 def run_script():
80 parser = optparse.OptionParser()
81 parser.add_option("--input")
82 parser.add_option("--export_spreadsheet", action="store_true", dest="export_spreadsheet")
83 parser.set_defaults(export_spreadsheet=False)
84 (options, args) = parser.parse_args()
85
86 copy_to_working_directory(options.input, "input.pep.xml")
87 cmd = "PepXMLViewer.cgi -I input.pep.xml"
88 cmd = "%s %s" % (cmd, "-B exportSpreadsheet")
89 if options.export_spreadsheet:
90 cmd = "%s %s" % (cmd, "1")
91 else:
92 cmd = "%s %s" % (cmd, "0")
93 execute(cmd)
94
95 if __name__ == '__main__': __main__()