#!/usr/bin/env python2
#

import os
import re
import sys

##  Prepare list of tuples: Reference file name, Reference file md5sum, Data file md5sum

inputfiles = []

inputfiles.append(("SAT_fgd_v3r14.log",     "9cdf4ab471fd63cac823709bdab8677c", "e693aa6de5bc5fc634bde0c0db78c478"))
inputfiles.append(("SAT_ingrid_v3r14.log",  "42c44468da539e805f15bd1a2f0d3d15", "ae5074b5ea467db691f4fc0580118f6d"))
inputfiles.append(("SAT_ingrid2_v3r14.log", "b02ca24b3b9217376eac4db0fd100bd7", "17eca3e57aab5f4523d2527dbbf3ec9a"))
inputfiles.append(("SAT_tript_v3r14.log",   "8f81c4b24891647d482de58925f7f7c1", "9c82f04742a5416907366389e14c284e"))
inputfiles.append(("SAT_tpc_v3r14.log",     "87494f66e12f6c53a81068d9147b1d5a", "44aea5c31b6c05487a91aadd3b56cb4d"))

## Loop over all input files using comet-get-validation-input to get the reference and data file.

print "\nFetching SAT (StandAlone Testing) input files ..."
for (ref_name,ref_md5sum,data_md5sum) in inputfiles:
    mo = re.search('(.*)_',ref_name)
    if not mo:
        print "Badly formed Reference file name: '%s'" % ref_name
        print "FAIL"
        sys.exit(1)
    data_name = mo.group(1) + ".mid.gz"
    for (file_name,file_md5sum) in ((ref_name,ref_md5sum),(data_name,data_md5sum)):
        print "  Getting %s ..." % file_name
        if os.system("comet-get-validation-input oaRawEvent %s %s" % (file_name,file_md5sum)):
            print "Failed to %s" % file_name
            print "FAIL"
            sys.exit(1)

print "All SAT input files ready"
sys.exit(0)