+++ /dev/null
-#!/usr/bin/env python\r
-# -*- coding: utf-8 -*-\r
-\r
-"""\r
-A process wrapper class that maintains the text output and execution status of\r
-a process or a list of other process wrappers which carry such data.\r
-"""\r
-\r
-from __future__ import division\r
-\r
-import os\r
-import sys\r
-import traceback\r
-\r
-__author__ = 'ACES Developers'\r
-__copyright__ = 'Copyright (C) 2014 - 2015 - ACES Developers'\r
-__license__ = ''\r
-__maintainer__ = 'ACES Developers'\r
-__email__ = 'aces@oscars.org'\r
-__status__ = 'Production'\r
-\r
-__all__ = ['read_text',\r
- 'write_text',\r
- 'Process',\r
- 'ProcessList',\r
- 'main']\r
-\r
-\r
-def read_text(text_file):\r
- """\r
- Object description.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- # TODO: Investigate if check is needed.\r
- if not text_file:\r
- return\r
-\r
- with open(text_file, 'rb') as fp:\r
- text = (fp.read())\r
-\r
- return text\r
-\r
-\r
-def write_text(text, text_file):\r
- """\r
- Object description.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- # TODO: Investigate if check is needed.\r
- if not text_file:\r
- return\r
-\r
- with open(text_file, 'wb') as fp:\r
- fp.write(text)\r
-\r
- return text\r
-\r
-\r
-class Process:\r
- """\r
- A process with logged output.\r
- """\r
-\r
- def __init__(self,\r
- description=None,\r
- cmd=None,\r
- args=None,\r
- cwd=None,\r
- env=None,\r
- batch_wrapper=False):\r
- """\r
- Initialize the standard class variables.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- if args is None:\r
- args = []\r
-\r
- self.cmd = cmd\r
- if not description:\r
- self.description = cmd\r
- else:\r
- self.description = description\r
- self.status = None\r
- self.args = args\r
- self.start = None\r
- self.end = None\r
- self.log = []\r
- self.echo = True\r
- self.cwd = cwd\r
- self.env = env\r
- self.batch_wrapper = batch_wrapper\r
- self.process_keys = []\r
-\r
- def get_elapsed_seconds(self):\r
- """\r
- Object description.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- import math\r
-\r
- if self.end and self.start:\r
- delta = (self.end - self.start)\r
- formatted = '%s.%s' % (delta.days * 86400 + delta.seconds,\r
- int(math.floor(delta.microseconds / 1e3)))\r
- else:\r
- formatted = None\r
- return formatted\r
-\r
- def write_key(self, write_dict, key=None, value=None, start_stop=None):\r
- """\r
- Writes a key / value pair in a supported format.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- if key is not None and (value is not None or start_stop is not None):\r
- indent = '\t' * write_dict['indentationLevel']\r
- if write_dict['format'] == 'xml':\r
- if start_stop == 'start':\r
- write_dict['logHandle'].write('%s<%s>\n' % (indent, key))\r
- elif start_stop == 'stop':\r
- write_dict['logHandle'].write('%s</%s>\n' % (indent, key))\r
- else:\r
- write_dict['logHandle'].write(\r
- '%s<%s>%s</%s>\n' % (indent, key, value, key))\r
- else:\r
- write_dict['logHandle'].write(\r
- '%s%40s : %s\n' % (indent, key, value))\r
-\r
- def write_log_header(self, write_dict):\r
- """\r
- Object description.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- import platform\r
-\r
- try:\r
- user = os.getlogin()\r
- except:\r
- try:\r
- user = os.getenv('USERNAME')\r
- if user is None:\r
- user = os.getenv('USER')\r
- except:\r
- user = 'unknown_user'\r
- try:\r
- (sysname, nodename, release, version, machine,\r
- processor) = platform.uname()\r
- except:\r
- (sysname, nodename, release, version, machine, processor) = (\r
- 'unknown_sysname', 'unknown_nodename', 'unknown_release',\r
- 'unknown_version', 'unknown_machine', 'unknown_processor')\r
-\r
- self.write_key(write_dict, 'process', None, 'start')\r
- write_dict['indentationLevel'] += 1\r
-\r
- self.write_key(write_dict, 'description', self.description)\r
- self.write_key(write_dict, 'cmd', self.cmd)\r
- if self.args:\r
- self.write_key(write_dict, 'args', ' '.join(self.args))\r
- self.write_key(write_dict, 'start', self.start)\r
- self.write_key(write_dict, 'end', self.end)\r
- self.write_key(write_dict, 'elapsed', self.get_elapsed_seconds())\r
-\r
- self.write_key(write_dict, 'user', user)\r
- self.write_key(write_dict, 'sysname', sysname)\r
- self.write_key(write_dict, 'nodename', nodename)\r
- self.write_key(write_dict, 'release', release)\r
- self.write_key(write_dict, 'version', version)\r
- self.write_key(write_dict, 'machine', machine)\r
- self.write_key(write_dict, 'processor', processor)\r
-\r
- if len(self.process_keys) > 0:\r
- self.write_key(write_dict, 'processKeys', None, 'start')\r
- for pair in self.process_keys:\r
- (key, value) = pair\r
- write_dict['indentationLevel'] += 1\r
- self.write_key(write_dict, key, value)\r
- write_dict['indentationLevel'] -= 1\r
- self.write_key(write_dict, 'processKeys', None, 'stop')\r
-\r
- self.write_key(write_dict, 'status', self.status)\r
-\r
- def write_log_footer(self, write_dict):\r
- """\r
- Object description.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- write_dict['indentationLevel'] -= 1\r
- self.write_key(write_dict, 'process', None, 'stop')\r
-\r
- def write_log(self,\r
- log_handle=sys.stdout,\r
- indentation_level=0,\r
- format='xml'):\r
- """\r
- Writes logging information to the specified handle.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- write_dict = {}\r
- write_dict['logHandle'] = log_handle\r
- write_dict['indentationLevel'] = indentation_level\r
- write_dict['format'] = format\r
-\r
- if log_handle:\r
- self.write_log_header(write_dict)\r
-\r
- if self.log:\r
- self.write_key(write_dict, 'output', None, 'start')\r
- if format == 'xml':\r
- log_handle.write('<![CDATA[\n')\r
- for line in self.log:\r
- log_handle.write('%s%s\n' % ('', line))\r
- if format == 'xml':\r
- log_handle.write(']]>\n')\r
- self.write_key(write_dict, 'output', None, 'stop')\r
-\r
- self.write_log_footer(write_dict)\r
-\r
- def write_log_to_disk(self, log_filename=None, format='xml', header=None):\r
- """\r
- Object description.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- if log_filename:\r
- try:\r
- # TODO: Review statements.\r
- # 3.1\r
- try:\r
- log_handle = (\r
- open(log_filename, mode='wt', encoding='utf-8'))\r
- # 2.6\r
- except:\r
- log_handle = open(log_filename, mode='wt')\r
- except:\r
- print('Couldn\'t open log : %s' % log_filename)\r
- log_handle = None\r
-\r
- if log_handle:\r
- if header:\r
- if format == 'xml':\r
- log_handle.write('<![CDATA[\n')\r
- log_handle.write(header)\r
- if format == 'xml':\r
- log_handle.write(']]>\n')\r
- self.write_log(log_handle, format=format)\r
- log_handle.close()\r
-\r
- def log_line(self, line):\r
- """\r
- Adds a line of text to the log.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- self.log.append(line.rstrip())\r
- if self.echo:\r
- print('%s' % line.rstrip())\r
-\r
- def execute(self):\r
- """\r
- Executes the current process.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- import datetime\r
- import traceback\r
-\r
- try:\r
- import subprocess as sp\r
- except:\r
- sp = None\r
-\r
- self.start = datetime.datetime.now()\r
-\r
- cmdargs = [self.cmd]\r
- cmdargs.extend(self.args)\r
-\r
- if self.echo:\r
- if sp:\r
- print(\r
- '\n%s : %s\n' % (self.__class__, sp.list2cmdline(cmdargs)))\r
- else:\r
- print('\n%s : %s\n' % (self.__class__, ' '.join(cmdargs)))\r
-\r
- process = None\r
- tmp_wrapper = None\r
- stdout = None\r
- stdin = None\r
- parentenv = os.environ\r
- parentcwd = os.getcwd()\r
-\r
- try:\r
- # Using *subprocess*.\r
- if sp:\r
- if self.batch_wrapper:\r
- cmd = ' '.join(cmdargs)\r
- tmp_wrapper = os.path.join(self.cwd, 'process.bat')\r
- write_text(cmd, tmp_wrapper)\r
- print('%s : Running process through wrapper %s\n' % (\r
- self.__class__, tmp_wrapper))\r
- process = sp.Popen([tmp_wrapper], stdout=sp.PIPE,\r
- stderr=sp.STDOUT,\r
- cwd=self.cwd, env=self.env)\r
- else:\r
- process = sp.Popen(cmdargs, stdout=sp.PIPE,\r
- stderr=sp.STDOUT,\r
- cwd=self.cwd, env=self.env)\r
-\r
- # using *os.popen4*.\r
- else:\r
- if self.env:\r
- os.environ = self.env\r
- if self.cwd:\r
- os.chdir(self.cwd)\r
-\r
- stdin, stdout = os.popen4(cmdargs, 'r')\r
- except:\r
- print('Couldn\'t execute command : %s' % cmdargs[0])\r
- traceback.print_exc()\r
-\r
- # Using *subprocess*\r
- if sp:\r
- if process is not None:\r
- # pid = process.pid\r
- # log.logLine('process id %s\n' % pid)\r
-\r
- try:\r
- # This is more proper python, and resolves some issues with\r
- # a process ending before all of its output has been\r
- # processed, but it also seems to stall when the read\r
- # buffer is near or over its limit. This happens\r
- # relatively frequently with processes that generate lots\r
- # of print statements.\r
- for line in process.stdout:\r
- self.log_line(line)\r
-\r
- # So we go with the, um, uglier option below.\r
-\r
- # This is now used to ensure that the process has finished.\r
- line = ''\r
- while line is not None and process.poll() is None:\r
- try:\r
- line = process.stdout.readline()\r
- except:\r
- break\r
- # 3.1\r
- try:\r
- # TODO: Investigate previous eroneous statement.\r
- # self.log_line(str(line, encoding='utf-8'))\r
- self.log_line(str(line))\r
- # 2.6\r
- except:\r
- self.log_line(line)\r
- except:\r
- self.log_line('Logging error : %s' % sys.exc_info()[0])\r
-\r
- self.status = process.returncode\r
-\r
- if self.batch_wrapper and tmp_wrapper:\r
- try:\r
- os.remove(tmp_wrapper)\r
- except:\r
- print(\r
- 'Couldn\'t remove temp wrapper : %s' % tmp_wrapper)\r
- traceback.print_exc()\r
-\r
- # Using *os.popen4*.\r
- else:\r
- exit_code = -1\r
- try:\r
- stdout_lines = stdout.readlines()\r
- # TODO: Investigate if this is the good behavior, close() does\r
- # not return anything / None.\r
- exit_code = stdout.close()\r
-\r
- stdout.close()\r
- stdin.close()\r
-\r
- if self.env:\r
- os.environ = parentenv\r
- if self.cwd:\r
- os.chdir(parentcwd)\r
-\r
- if len(stdout_lines) > 0:\r
- for line in stdout_lines:\r
- self.log_line(line)\r
-\r
- if not exit_code:\r
- exit_code = 0\r
- except:\r
- self.log_line('Logging error : %s' % sys.exc_info()[0])\r
-\r
- self.status = exit_code\r
-\r
- self.end = datetime.datetime.now()\r
-\r
-\r
-class ProcessList(Process):\r
- """\r
- A list of processes with logged output.\r
- """\r
-\r
- def __init__(self, description, blocking=True, cwd=None, env=None):\r
- """\r
- Object description.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- Process.__init__(self, description, None, None, cwd, env)\r
- 'Initialize the standard class variables'\r
- self.processes = []\r
- self.blocking = blocking\r
-\r
- def generate_report(self, write_dict):\r
- """\r
- Generates a log based on the success of the child processes.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- if self.processes:\r
- _status = True\r
- indent = '\t' * (write_dict['indentationLevel'] + 1)\r
-\r
- self.log = []\r
-\r
- for child in self.processes:\r
- if isinstance(child, ProcessList):\r
- child.generate_report(write_dict)\r
-\r
- key = child.description\r
- value = child.status\r
- if write_dict['format'] == 'xml':\r
- child_result = (\r
- '%s<result description=\'%s\'>%s</result>' % (\r
- indent, key, value))\r
- else:\r
- child_result = ('%s%40s : %s' % (indent, key, value))\r
- self.log.append(child_result)\r
-\r
- if child.status != 0:\r
- _status = False\r
- if not _status:\r
- self.status = -1\r
- else:\r
- self.status = 0\r
- else:\r
- self.log = ['No child processes available to generate a report']\r
- self.status = -1\r
-\r
- def write_log_header(self, write_dict):\r
- """\r
- Object description.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- self.write_key(write_dict, 'processList', None, 'start')\r
- write_dict['indentationLevel'] += 1\r
-\r
- self.write_key(write_dict, 'description', self.description)\r
- self.write_key(write_dict, 'start', self.start)\r
- self.write_key(write_dict, 'end', self.end)\r
- self.write_key(write_dict, 'elapsed', self.get_elapsed_seconds())\r
-\r
- self.generate_report(write_dict)\r
-\r
- self.write_key(write_dict, 'status', self.status)\r
-\r
- def write_log_footer(self, write_dict):\r
- """\r
- Object description.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- write_dict['indentationLevel'] -= 1\r
- self.write_key(write_dict, 'processList', None, 'stop')\r
-\r
- def write_log(self,\r
- log_handle=sys.stdout,\r
- indentation_level=0,\r
- format='xml'):\r
- """\r
- Writes logging information to the specified handle.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- write_dict = {}\r
- write_dict['logHandle'] = log_handle\r
- write_dict['indentationLevel'] = indentation_level\r
- write_dict['format'] = format\r
-\r
- if log_handle:\r
- self.write_log_header(write_dict)\r
-\r
- if self.log:\r
- self.write_key(write_dict, 'output', None, 'start')\r
- for line in self.log:\r
- log_handle.write('%s%s\n' % ('', line))\r
- self.write_key(write_dict, 'output', None, 'stop')\r
-\r
- if self.processes:\r
- self.write_key(write_dict, 'processes', None, 'start')\r
- for child in self.processes:\r
- child.write_log(log_handle, indentation_level + 1, format)\r
- self.write_key(write_dict, 'processes', None, 'stop')\r
-\r
- self.write_log_footer(write_dict)\r
-\r
- def execute(self):\r
- """\r
- Executes the list of processes.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- import datetime\r
-\r
- self.start = datetime.datetime.now()\r
-\r
- self.status = 0\r
- if self.processes:\r
- for child in self.processes:\r
- if child:\r
- try:\r
- child.execute()\r
- except:\r
- print('%s : caught exception in child class %s' % (\r
- self.__class__, child.__class__))\r
- traceback.print_exc()\r
- child.status = -1\r
-\r
- if self.blocking and child.status != 0:\r
- print('%s : child class %s finished with an error' % (\r
- self.__class__, child.__class__))\r
- self.status = -1\r
- break\r
-\r
- self.end = datetime.datetime.now()\r
-\r
-\r
-def main():\r
- """\r
- Object description.\r
-\r
- Parameters\r
- ----------\r
- parameter : type\r
- Parameter description.\r
-\r
- Returns\r
- -------\r
- type\r
- Return value description.\r
- """\r
-\r
- import optparse\r
-\r
- p = optparse.OptionParser(description='A process logging script',\r
- prog='process',\r
- version='process 0.1',\r
- usage=('%prog [options] '\r
- '[options for the logged process]'))\r
- p.add_option('--cmd', '-c', default=None)\r
- p.add_option('--log', '-l', default=None)\r
-\r
- options, arguments = p.parse_args()\r
-\r
- cmd = options.cmd\r
- log_filename = options.log\r
-\r
- try:\r
- args_start = sys.argv.index('--') + 1\r
- args = sys.argv[args_start:]\r
- except:\r
- args = []\r
-\r
- if cmd is None:\r
- print('process: No command specified')\r
-\r
- # Testing regular logging.\r
- process = Process(description='a process', cmd=cmd, args=args)\r
-\r
- # Testing report generation and writing a log.\r
- process_list = ProcessList('a process list')\r
- process_list.processes.append(process)\r
- process_list.echo = True\r
- process_list.execute()\r
-\r
- process_list.write_log_to_disk(log_filename)\r
-\r
-\r
-if __name__ == '__main__':\r
- main()\r