log-input-files

#!/usr/bin/env python

# Process job reports and create summary file with performance information.
from argparse import ArgumentParser
from xml.sax import parseString, ContentHandler
from hashlib import sha1
from datetime import datetime
import urllib2, base64, json
from os import getenv
from cmsutils import getRelValsFiles, readRelValFile
import re

# Pipes results to elasticsearch.
class ElasticSearchPrinter(object):
  def __init__(self):
    self.hostname = getenv("ES_HOSTNAME")
    self.auth = getenv("ES_AUTH")

  # We use the REST API because the python one does not seem to work
  # with our ssl setup.
  def __call__(self, release, architecture, workflow, step, content):
    sha1_id = sha1(release + architecture +  workflow + step + content).hexdigest()
    datepart = "201" + release.split("201")[1]
    d = datetime.strptime(datepart, "%Y-%m-%d-%H00")
    payload = {"release": release,
               "architecture": architecture,
               "workflow": workflow,
               "step": step,
               "file": content
               }
    url = "https://%s/ib-files-%s/ib-file-data/%s" % (self.hostname, 
                                                      d.strftime("%Y.%m"), 
                                                      sha1_id)
    request = urllib2.Request(url)
    if self.auth:
      base64string = base64.encodestring(self.auth).replace('\n', '')
      request.add_header("Authorization", "Basic %s" % base64string)
    request.get_method = lambda: 'PUT'
    result = urllib2.urlopen(request, data=json.dumps(payload))

class JobReportHandler(ContentHandler):
  def __init__(self, release, architecture, workflow, step, files):
    self.release = release
    self.architecture = architecture
    self.workflow = workflow
    self.step = step
    self.printer = None
    if getenv("ES_HOSTNAME"):
      self.printer = ElasticSearchPrinter()
    self.inLFN = False
    self.files = files
    self.content = ''

  def startElement(self, name, attrs):
    if name != "LFN":
      return
    self.content = ''
    self.inLFN = True 

  def characters(self, content):
    if self.inLFN:
      self.content += content

  def endElement(self, name):
    if name != "LFN":
      return
    if self.printer:
      self.printer(self.release, self.architecture, self.workflow, self.step, self.content)
    self.files[self.content] = 1
    self.inLFN = False

if __name__ == "__main__":
  parser = ArgumentParser()
  parser.add_argument("input", nargs=1, help="input file")
  args = parser.parse_args()
  # We start with empty workflows
  results = {}

  printHeader = True
  files = {}
  for inFile in args.input:
    m = re.match(".*/([^/]+)/www/.*/([^/]+)/.*", inFile)
    architecture, release = m.groups()
    inFiles, zipFile = getRelValsFiles(inFile, "*/JobReport*.xml", ".*/JobReport([0-9]).xml")
    table = {}
    for l in inFiles:
      m = re.match("(.+/|)([^/]+)/JobReport([0-9]).xml", l)
      if not m: continue
      handler = JobReportHandler(release, architecture, m.group(2), m.group(3), files)
      try:
        parseString(readRelValFile(l,zipFile), handler)
      except Exception, e:
        continue
  print "\n".join(files.keys())