Execution.py

Go to the documentation of this file.

for script in opts.scripts:
   try:
      include( script )
   except Exception as e:
      if isinstance(e,SystemExit):
         raise
 
    # "sanitize" the traceback under the assumption that it is the user's fault
      import traceback
 
      if not _msg.isEnabledFor( logging.DEBUG ):
    # "sanitize" the traceback under the assumption that it is the user's fault
         exc_info = sys.exc_info()
         tb = traceback.extract_tb( exc_info[2] )
 
         short_tb = []
         for frame_info in tb:
            if not 'AthenaCommon' in frame_info[0]:
               short_tb.append( frame_info )
 
         print ('Shortened traceback (most recent user call last):')
         print (''.join( traceback.format_list( short_tb ) ),)
         print (''.join( traceback.format_exception_only( exc_info[0], exc_info[1] ) ),)
      else:
         traceback.print_exc()
 
      if not opts.interactive:
 
         # --- Start of CA-in-Legacy warning check ---
         if opts.scripts:
             try:
                 with open(opts.scripts[0], 'r') as f:
                     script_content = f.read(5000) #limit on large files
                     if "ComponentAccumulator" in script_content or "ConfigFlags" in script_content:
                         log.warning("The script '%s' appears to use ComponentAccumulator syntax,", opts.scripts[0])
                         log.warning("but athena.py is executing it in Legacy mode.")
                         log.warning("FIX: Add '#!/usr/bin/env athena.py' as the first line of your script.")
             except Exception:
                 # We wrap this in a try/except to ensure a file-read error 
                 # doesn't prevent the actual job from attempting to run.
                 pass
         # --- End of warning check ---
 
       # additional processing to get right error codes
         import AthenaCommon.ExitCodes as ExitCodes
         if isinstance( e, IncludeError ):
            theApp._exitstate = ExitCodes.INCLUDE_ERROR
            print('If the script is a ComponentAccumulator configuration, add a shebang (#!) '
                  'to its first line or run with the --CA option.\n')
         elif isinstance( e, ImportError ):
            theApp._exitstate = ExitCodes.IMPORT_ERROR
         else:
            theApp._exitstate = ExitCodes.UNKNOWN_EXCEPTION
 
         del ExitCodes
         theApp.exit()
 
    # for interactive: drop into prompt
      break
 
else:
 
   if "-h" in opts.user_opts or "--help" in opts.user_opts:
      # The user script was displaying help messages stop here
      # In most cases, the user script should kill the process already (argparse
      # does this for you) but just in case print our help message and stop
      from AthenaCommon.AthOptionsParser import _error_msg
      print(_error_msg)
      theApp.exit()
 
 
   if opts.fromdb:
      _msg.info( 'now loading %s  ... ', opts.fromdb )
      import AthenaCommon.ConfigurationShelve
      AthenaCommon.ConfigurationShelve.loadJobOptionsCatalogue( opts.fromdb )
      del AthenaCommon.ConfigurationShelve
      if opts.fromdb.startswith( 'TempAthenaConfig.' ):
         try:
            os.remove( opts.fromdb )
         except OSError:
            pass
      opts.fromdb = None
 
 
   if opts.perfmon:
      import PerfMonComps.PerfMonFlags as _pmf
      _pmf._decode_pmon_opts(opts.perfmon.split(','))
 
 
   if DbgStage.value == "init":
      hookDebugger()
 
 
   if not opts.fromdb and opts.nprocs and (opts.nprocs >= 1 or opts.nprocs==-1):
      from AthenaCommon.AppMgr import ServiceMgr as svcMgr
      import AthenaMP.PyComps as _amppy
      svcMgr += _amppy.MpEvtLoopMgr(NWorkers=opts.nprocs)
 
   #if EvtMax and SkipEvents are set, use them
   from AthenaCommon.AthenaCommonFlags import jobproperties as jps
   if jps.AthenaCommonFlags.EvtMax.statusOn:
      theApp.EvtMax = jps.AthenaCommonFlags.EvtMax()
   if jps.AthenaCommonFlags.SkipEvents.statusOn:
      if hasattr(svcMgr,"EventSelector"):
         if not hasattr(svcMgr, "SecondaryEventSelector"):
            svcMgr.EventSelector.SkipEvents = jps.AthenaCommonFlags.SkipEvents()
      else:
         _msg.warning('No EventSelector in svcMgr, not skipping events')
 
 
 
   if not opts.interactive:
      
      include( "AthenaCommon/runbatch.py" )
   else:
      #pass
      # done, back to the user
 
 
      if opts.nprocs and (opts.nprocs >= 1 or opts.nprocs==-1):
         _msg.warning('interactive mode not validated with AthenaMP')
 
      _msg.info( 'entered interactive mode - have fun!' )
      _msg.info( 'you can call theApp.run() to execute and use ctrl-d to exit' )
      os.environ['PYTHONINSPECT'] = '1'