fetchNoiseMap.py

Go to the documentation of this file.

#! /usr/bin/env python

# A script to copy a noise channel map for specific file or time

import subprocess, sys, os
from datetime import datetime,timedelta

def get_time(time_str):
    return datetime.strptime(time_str,"%Y-%m-%dT%H:%M:%S+00:00")

def get_mapname(stime):
    return stime.strftime("mask.FD.%Y%m%d_%H00*.nhm")

def get_metadata(fname):
    """ get the file metadata as a dict"""
    p = subprocess.Popen(['samweb','get-metadata',fname], 
                         stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    resp_s,err=p.communicate()
    if(len(err)): print("### Error: "+err);

    resp={}
    for line in resp_s.split('\n'):
        pos = line.find(':') #find the delimiter
        if(pos==-1): continue
        key=line[:pos].strip()
        val=line[(pos+1):].strip()
        resp[key]=val
    return resp

def fetch_map(fname,tgt):
    """ Copy a noise channel mask here"""
    p = subprocess.call(['scp','novadaq@novadaq-far-trigger-2016.fnal.gov:/ddt/NoiseMaps/backup/'+fname,tgt])
    return p



import argparse
parser = argparse.ArgumentParser(description='Fetch the noise channel mask for given file/timestamp')
parser.add_argument('FILE', help='A filename, which timestamp will be used')
parser.add_argument('--output','-o', default='.', help='Where to put the mask')
args = parser.parse_args()
print(args)

fname=args.FILE
target = args.output

mdata = get_metadata(fname)
stime = get_time(mdata['Start Time'])

print("Start time = {}".format(stime))
attempt = 0
while True:
    mapname=get_mapname(stime)
    print("Attempt #{}: fetching file {} to {}".format(attempt,mapname,target))
    if(fetch_map(mapname, target)==0): 
        print("Success")
        break;
    print("Probably map doesn't exist, trying 1 hour before...")    
    attempt+=1
    stime-=timedelta(hours=1)
    if(attempt>3): 
        print("Failed")
        sys.exit(1)