MooRTT_dynamic_pagemaker.py

Go to the documentation of this file.

#! /usr/bin/env python
 
# Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
 
import os
import sys
 
# module: MooRTT_dynamic_pagemaker.py
        
class MooRTT_dynamic_pagemaker:
   def __init__(self,argDict={}):
 
       # access my args from the dictionary passed in, and convert any basic types
       # self.myArg = argDict.get('MyArg') # = the Python string type '3'
       # self.myArg = int(self.myArg) # = the Python integer type 3  
 
       # access the RTT descriptor instance
       rttDescriptor = argDict.get('JobDescriptor')
     
       # grab some stuff from it
       self.release = str(rttDescriptor.paths.release)
       self.build   = str(rttDescriptor.paths.build)
       self.jobGroup = str(rttDescriptor.jobGroup)
       self.platform = str(rttDescriptor.paths.targetCMTCONFIG)
       self.branch = str(rttDescriptor.paths.branch)
       self.runPath = str(rttDescriptor.runPath)
 
   def run(self):
       # do the stuff I wanted to do...
       logfile = open(self.runPath + "/MooRTT_dynamic_pagemaker.log",'a')
       logfile.write("Starting MooRTT_dynamic_pagemaker.py run method\n")
 
       inputfile = self.runPath + "/ALL_MooRTT_Plots.html"
       outputfile = inputfile.replace(".html","_Dynamic.html")
 
       datasetlist = {"AthenaMooPerformance-MoMu-mupt100-geo-02-01-00"     :  " SingleMu_100GeV_15616 " ,
                      "AthenaMooPerformance-MoMu-mupt10-geo-02-01-00"      :  " SingleMu_10GeV_15616 " ,
                      "AthenaMooPerformance-MoMu-Zprime-geo-02-01-00"      :  " SingleMu_5GeV_15616 " ,
                      "AthenaMooPerformance-MoMu-h4lep-geo-02-01-00"       :  " SingleMu_3GeV_15616 " ,
                      "AthenaMooPerformance-MoMu-mupt5-hyb-geo-08-00-02mupt10-geo-02-01-00"   :  " SingleMu_5GeV_15616 " ,
                      "AthenaMooPerformance-MoMu-mupt5-TGmat-geo-08-00-02mupt10-geo-02-01-00" :  " SingleMu_5GeV_15616 " ,
                      "AthenaMooPerformance-MoMu-Jpsi-geo-08-00-02"        :  " Jpsi_15616 " ,
                      "AthenaMooPerformance-MoMu-zmumu-geo-08-00-02"       :  " Zmumu_15616 " ,
                      "AthenaMooPerformance-MoMu-ttbar-geo-08-00-02"       :  " ttbar_15616 " ,
                      "AthenaMooPerformance-MoMu-zmumu-csc-01-02-00-sf05"  :  " pile1sf05_misal1_csc11.005145.PythiaZmumu.digit.RDO.v12000605 " 
                      }
       
       # helper function
       def szsplit(stringlist,splitchar=" "):
         if str(splitchar) == splitchar:
           try:
             return stringlist.split(splitchar)
           except:
             newlist = []
             for item in stringlist:
               newlist += item.split(splitchar)
             return newlist
         else:
           for onechar in splitchar:
             stringlist = szsplit(stringlist,onechar)
           return stringlist
 
       logfile.write("Input file: " + inputfile + "\n")
       logfile.write("Output file: " + outputfile + "\n")
       logfile.write("datasetlist: \n")
       for key in datasetlist.keys():
         logfile.write("  " + key + ": " + datasetlist[key] +"\n")
 
       logfile.write("Getting dataset... ")
       try:
          dataset = datasetlist[self.jobGroup]
          logfile.write("OK!\n")
       except:
          dataset = "Unknown (jobGroup = " + self.jobGroup + ")"
          logfile.write("Unknown!\n")
 
       logfile.write("  Dataset name is " + dataset + "\n")
 
       jobinfolist = { "Release" : self.release , "Branch" : self.branch , "Build" : self.build , "Platform" : self.platform  , "Dataset" : dataset }
 
       # For testing purposes
       if self.runPath == ".":
          self.runPath = "/afs/cern.ch/atlas/project/RTT/Results/rel_3/devval/build/i686-slc4-gcc34-dbg/offline/MooPerformance/AthenaMooPerformance-MoMu-mupt10-csc-01-02-00/MoMuStandalone_topOptions/100"
          
       logfile.write("jobinfolist: \n")
       for key in jobinfolist.keys():
          logfile.write("  " + key + ": " + jobinfolist[key] +"\n")
 
       logfile.write("Opening input file: " + inputfile + "\n")
       input = open(inputfile,'r')
       
       logfile.write("Opening output file: " + outputfile + "\n")
       output = open(outputfile,'w')
 
       logfile.write("Starting copy...\n")
 
       for line in input:
          logfile.write("Line: " + line)
          if line.count("<h2>Contents</h2>"):
             logfile.write("  Contains <h2>Contents</h2>\n")
             output.write("<TABLE BORDER=6 CELLSPACING=4 CELLPADDING=4>\n")
             for key in jobinfolist.keys():
                logfile.write("  Going through jobinfo list and writing value of " + key + "... ")
                output.write("  <TR><TD><b>" + key + "</b></TD><TD>" + jobinfolist[key] + "</TD></TR>\n")
                logfile.write("Done.\n")
             logfile.write("Writing date and finishing table... ")
             output.write("  <TR><TD><b>" + "Date" + "</b></TD><TD>" + os.popen("date").readline().strip() + "</TD></TR>\n")
             output.write("</TABLE>\n\n")
             logfile.write("Done.\n")
          if line.count("RTT default page"):
            logfile.write("  Contains RTT default page")
            htmlline = "<li><a href=\"http://atlas-project-rtt-results.web.cern.ch/atlas-project-rtt-results/page2.php?xml="
            htmlline += szsplit(self.runPath,["Results/","/MooPerformance"])[1]
            htmlline += "/RTTSummary.xml&package=MooPerformance&job=MoMuStandalone_topOptions&id=" + self.runPath.split("/")[-1]
            htmlline += "\">RTT default page</a></li>\n"
            output.write(htmlline)
            logfile.write("    Writing: " + htmlline)
          else:
            if line.count("Log files"):
              logfile.write("  Contains log files")
              htmlline = "<li>Log files: "
              for filename in os.listdir(self.runPath):
                if filename.count("log"):
                  htmlline += "<a href=\"" + filename + "\">" + filename + "</a> "
              htmlline += "</li>\n"
              output.write(htmlline)
              logfile.write("    Writing: " + htmlline)
            else:
              output.write(line)
              logfile.write("  Output original line.\n")
 
       logfile.write("Now closing files.\n")
       output.close()
       input.close()
       logfile.close()
 
class fakePaths:
   def __init__(self,release="release",build="build",platform="platform",branch="branch"):
      self.release = release
      self.build = build
      self.targetCMTCONFIG = platform
      self.branch = branch
      self.runPath = os.popen("pwd").readline().strip()
 
class fakeRttDescriptor:
   def __init__(self):
      self.paths = fakePaths()
      self.jobGroup = "AthenaMooPerformance-MoMu-mupt10-csc-01-02-00"
      self.runPath = "."
 
# Test run
if sys.argv[0] == "./MooRTT_dynamic_pagemaker.py":
 
   fakedecr = fakeRttDescriptor()
   fakeArgDict = { 'JobDescriptor' : fakedecr }
   pagemaker = MooRTT_dynamic_pagemaker(fakeArgDict)
   pagemaker.run()