|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
import xml.etree.ElementTree as ET
import sys
## declare all constants here
TARGET_LIST_OF_TAGS=['BeamSpotObject_ByLumi', # beamspot
'BeamSpotObject_ByRun',
'BeamSpotObjectHP_ByLumi',
'BeamSpotObjectHP_ByRun',
'SiPixelLA_pcl', # SiPixel
'SiPixelLAMCS_pcl',
'SiPixelQualityFromDbRcd_other',
'SiPixelQualityFromDbRcd_prompt',
'SiPixelQualityFromDbRcd_stuckTBM',
'SiStripApvGain_pcl', # SiStrip
'SiStripApvGainAAG_pcl',
'SiStripBadStrip_pcl',
'SiStripBadStripRcdHitEff_pcl',
'SiStripLA_pcl',
'SiPixelAli_pcl', # Alignment
'SiPixelAliHG_pcl',
'SiPixelAliHGCombined_pcl',
'SiPixelAliHLTHGCombined_pcl'
]
TARGET_DQM_FILES=1
TARGET_DQM_FILENAME='./DQM_V0001_R000325022__Express__PCLTest__ALCAPROMPT.root'
TARGET_DB_FILES=len(TARGET_LIST_OF_TAGS)
TARGET_DB_FILENAME='sqlite_file:testPCLAlCaHarvesting.db'
TARGET_XML_FILENAME='testPCLAlCaHarvesting.xml'
TOTAL_TARGET_FILES=TARGET_DQM_FILES+TARGET_DB_FILES
#_____________________________________________________
def parseXML(xmlfile):
# create element tree object
tree = ET.parse(xmlfile)
# get root element
root = tree.getroot()
totAnaEntries=len(root.findall('AnalysisFile'))
if(totAnaEntries!=TOTAL_TARGET_FILES):
print("ERROR: found a not expected number (",totAnaEntries,") of AnalysisFile entries in the FrameworkJobReport.xml, expecting",TOTAL_TARGET_FILES)
return -1
listOfInputTags=[]
countDBfiles=0
countDQMfiles=0
# iterate news items
for item in root.findall('AnalysisFile'):
# iterate child elements of item
for child in item:
if(child.tag == 'FileName'):
if(child.text==TARGET_DB_FILENAME):
countDBfiles+=1
elif(child.text==TARGET_DQM_FILENAME):
countDQMfiles+=1
else:
pass
if(child.tag == 'inputtag'):
listOfInputTags.append(child.attrib['Value'])
if(countDBfiles!=TARGET_DB_FILES):
print("ERROR! Found an unexpected number of DB files (",countDBfiles,")")
return -1
if(countDQMfiles!=TARGET_DQM_FILES):
print("ERROR! Found an uexpected number of DQM files (",countDQMfiles,")")
return -1
## first sort to avoid spurious false positives
listOfInputTags.sort()
TARGET_LIST_OF_TAGS.sort()
## That's strict!
if(listOfInputTags!=TARGET_LIST_OF_TAGS):
print(" list of input tags found in file:",listOfInputTags)
print(" target list of tags:",TARGET_LIST_OF_TAGS)
if (listOfInputTags>TARGET_LIST_OF_TAGS):
print("ERROR!\n This ",[x for x in TARGET_LIST_OF_TAGS if x not in listOfInputTags]," is the set of expected tags not found in the FwdJobReport!")
else:
print("ERROR!\n This ",[x for x in listOfInputTags if x not in TARGET_LIST_OF_TAGS]," is the set of tags found in the FwkJobReport, but not expected!")
return -1
return 0
#_____________________________________________________
def main():
try:
f = open(TARGET_XML_FILENAME)
except IOError:
print("%s is not accessible" % TARGET_XML_FILENAME)
sys.exit(1)
# parse xml file
result = parseXML(TARGET_XML_FILENAME)
if(result==0):
print("All is fine with the world!")
sys.exit(0)
else:
print("Parsing the FwkJobReport results in failure!")
sys.exit(1)
#_____________________________________________________
if __name__ == "__main__":
# calling main function
main()
|