getreco.py

Go to the documentation of this file.

#!/usr/bin/env python

from __future__ import print_function
import sys
import samweb_cli
from subprocess import Popen, PIPE
import os

# usage: getreco.py <run_number> <event_number>

samweb = samweb_cli.SAMWebClient(experiment='dune')

files = samweb.listFiles("run_number %s " % (sys.argv[1]) + "and run_type protodune-sp and data_tier raw")

#print (files)
for file in files:
    #print (file)
    events = samweb.getURL('/files/metadata/event_numbers', {'file_name': file}).json()
    if int(sys.argv[2]) in events :
        print ("Raw: ", file, '--nskip',events.index(int(sys.argv[2])))
        print ("Locations:")
        cmd = ["samweb", "locate-file", file]
        p = Popen(cmd, stdout=PIPE, stderr=PIPE)
        stdout, stderr = p.communicate()
        if not stderr:
          print (stdout.decode('ascii'))
        print ("Access URLs:")        
        cmd = ["samweb", "get-file-access-url", file, "--schema=xroot"]
        p = Popen(cmd, stdout=PIPE, stderr=PIPE)
        stdout, stderr = p.communicate()
        if not stderr:
          print (stdout.decode('ascii'))
        filename =  os.path.splitext(file)[0]
        recofiles = samweb.listFiles("run_number %s " % (sys.argv[1]) + "and run_type protodune-sp and (data_tier full-reconstructed or data_tier reco-recalibrated)")
        for recofile in recofiles:
          if filename+'_' in recofile:
            print ("Reco: ", recofile, '--nskip',events.index(int(sys.argv[2])))
            print ("Locations:")
            cmd = ["samweb", "locate-file", recofile]
            p = Popen(cmd, stdout=PIPE, stderr=PIPE)
            stdout, stderr = p.communicate()
            if not stderr:
              print (stdout.decode('ascii'))
            print ("Access URLs:")        
            cmd = ["samweb", "get-file-access-url", recofile, "--schema=xroot"]
            p = Popen(cmd, stdout=PIPE, stderr=PIPE)
            stdout, stderr = p.communicate()
            if not stderr:
              print (stdout.decode('ascii'))


#from subprocess import Popen, PIPE
#import sys
#
#command = ["getfile.py", str(sys.argv[1]), str(sys.argv[2])]
#
#p = Popen(command, stdout=PIPE, stderr=PIPE)
#stdout, stderr = p.communicate()
#
#if not stderr:
#  #should just be one line
#  line = stdout.split("\n")[0]
#  print ("Raw: ", line) 
#  
#  nskip = line.split(" ")[1:] 
#  raw = line.split(" ")[0]
#  sub = raw.split("_")[3] + "_" + raw.split("_")[4].strip(".root")
#
#  print (nskip, raw, sub)
#
#  #find the runset for the run
#  cmd = ["samweb", "list-definitions"]
#  p = Popen(cmd, stdout=PIPE, stderr=PIPE)
#  stdout, stderr = p.communicate()
#  if not stderr:
#    for d in stdout.split("\n"):
#      if "reco-unified" in d and str(sys.argv[1]) in d: 
#
#        cmd = ["samweb", "list-files", "defname:", d]
#      
#        p = Popen(cmd, stdout=PIPE, stderr=PIPE)
#        stdout, stderr = p.communicate()
#        if not stderr:
#          for f in stdout.split("\n"):
#            if raw.strip(".root") in f: 
#              print ("Reco:", f, nskip[0], nskip[1] )
#              print(" ")
#              print ("Locations:")
#      
#              cmd = ["samweb", "locate-file", f]
#              p = Popen(cmd, stdout=PIPE, stderr=PIPE)
#              stdout, stderr = p.communicate()
#              if not stderr:
#                print (stdout)
#      
#              print(" ")
#              print ("Access URLs:")        
#              cmd = ["samweb", "get-file-access-url", f, "--schema=xroot"]
#              p = Popen(cmd, stdout=PIPE, stderr=PIPE)
#              stdout, stderr = p.communicate()
#              if not stderr:
#                print (stdout)
#  else: print stderr