MinosRSMJob.py

Go to the documentation of this file.

import os
import re

from GBSJob       import GBSJob
from MinosRSMTask import IsValidJobName

class MinosRSMJob(GBSJob) :
    """Object to submit, and if necessary resubmit a job until it
    succeeds or needs user intervention.

    This class inherits from GBSJob and exploits the constraint,
    imposed by MinosRSMTask, that its name must be of the form:-

      job_rrrrrrrr_ssss

      where

        rrrrrrrr  is an 8 digit zero padded run number
        ssss      is a 4 digit zero padded subrun number

    to add to its local environment:-

      run=<run-number>
      subrun=<subrun-number>

    It introduces the new methods:-

      GetRun()
      GetSubrun()
      Rename(new_name)

    """

    ######  GBSObject inherited responsibilities   ###### 
      

    def __init__(self,name,parent,model,model_args):
        GBSJob.__init__(self,name,parent,model,model_args)
        self.SetLocalEnvironment(self.GetLocalEnvironment())

    def _DoMemberIO(self,ioh):
        GBSJob._DoMemberIO(self,ioh)

    def GetType(self): return "MinosRSMJob"

    def __repr__(self) : return self.AsString()
    
    ######  User Callable Methods (Getters then Setters)  ###### 

    def GetRun(self):
        """Return run number"""
        return int(re.search(r"job_(\d+)",self.GetName()).group(1))

    def GetSubrun(self):
        """Return subrun number"""
        return int(re.search(r"job_\d+_(\d+)",self.GetName()).group(1))

    def SetLocalEnvironment(self,env_str):

        """Set, as a comma separated list string, the environment that local to this job.

        Extends the GBSJob method by adding run=<run-number>,subrun=<subrun-number>."""

        GBSJob.SetLocalEnvironment(self,env_str)
        GBSJob.SetLocalEnvironment(self,"+run=" + str(self.GetRun()) + ",subrun=" + str(self.GetSubrun()))

        
    ######  Private Methods (not user callable)  ######

    def _Rename(self,new_name):
        """Rename job to new_name.  Should only be called by parent Task."""

        #  Check that rename is legal.
        if not IsValidJobName(new_name): return
        if self.GetParent().GetJob(new_name,False):
            print "Sorry, there already is a job named " + new_name
            return
        
        # Perform rename including environmental variables, state file and child directory (if present).
        old_state_file = self.GetStoreLocation("self")
        old_child_dir  = self.GetStoreLocation("child_dir")
        GBSJob.Rename(self,new_name)
        new_state_file = self.GetStoreLocation("self")
        new_child_dir  = self.GetStoreLocation("child_dir")
        os.rename(old_state_file,new_state_file)
        if os.path.isdir(old_child_dir): os.rename(old_child_dir,new_child_dir)
        self.SetLocalEnvironment(self.GetLocalEnvironment())
        self.Write()        

---