|
|
|
|
|
|
|
|
|
|
|
import sys,shutil |
|
|
from . import logger |
|
|
from .newRunJob import Job |
|
|
from . import optparser as opt |
|
|
import traceback as _tr |
|
|
|
|
|
_name = "Main" |
|
|
|
|
|
|
|
|
def _cleanup(aggrescan_job_instance): |
|
|
""" |
|
|
Done to prevent a situation where Job raises an error on initialization and there is nothing to delete |
|
|
Normally deletes or keeps the simulations temporary directory depending on verbosity settings |
|
|
(Can't check for Exceptions in finally block, hence the function) |
|
|
""" |
|
|
try: |
|
|
if logger.get_log_level() < 4: |
|
|
shutil.rmtree(aggrescan_job_instance.get_tempdir(),ignore_errors=True) |
|
|
logger.log_file(module_name=_name, msg="Removing temporary files") |
|
|
else: |
|
|
logger.log_file(module_name=_name, |
|
|
msg="Verbosity higher than 3 - temporary files kept in %s" |
|
|
% aggrescan_job_instance.get_tempdir()) |
|
|
except AttributeError: |
|
|
pass |
|
|
|
|
|
|
|
|
def run_program(): |
|
|
try: |
|
|
a = "dummy" |
|
|
options = opt.parse(options=sys.argv[1:]) |
|
|
logger.setup(log_level=options['verbose'], remote=options['remote'], work_dir=options["work_dir"]) |
|
|
a = Job(config=options) |
|
|
a.run_job() |
|
|
_cleanup(a) |
|
|
logger.info(module_name="Main", msg="Simulation completed successfully.") |
|
|
|
|
|
except KeyboardInterrupt: |
|
|
_cleanup(a) |
|
|
logger.info(module_name=_name, msg="Interrupted by user") |
|
|
|
|
|
except logger.AggrescanError as custom_error: |
|
|
_cleanup(a) |
|
|
custom_error.generate_error_file() |
|
|
logger.exit_program(module_name=custom_error.module_name, |
|
|
msg=custom_error.logger_msg, |
|
|
traceback=None, |
|
|
exc=custom_error) |
|
|
|
|
|
except Exception as e: |
|
|
_cleanup(a) |
|
|
logger.record_exception(trace_stack=_tr.format_exc()) |
|
|
logger.critical(module_name=_name, |
|
|
msg="Unhandled Exception caught: %s." % e.message) |
|
|
if logger.get_log_level() > 2: |
|
|
logger.info(module_name=_name, |
|
|
msg="Verbosity higher than 2 - raising the Exception to provide traceback.") |
|
|
raise |
|
|
else: |
|
|
logger.exit_program() |
|
|
|
