Abort in case input file does not contain events ttree (#402)

HEP-FCC · Sep 16, 2024 · b0d4bd8 · b0d4bd8
1 parent ecf6135
commit b0d4bd8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/python/process.py b/python/process.py
@@ -22,8 +22,11 @@ def get_entries(inpath: str) -> int:
     '''
     nevents = None
     with ROOT.TFile(inpath, 'READ') as infile:
-        tt = infile.Get("events")
-        nevents = tt.GetEntries()
+        try:
+            nevents = infile.Get("events").GetEntries()
+        except AttributeError:
+            LOGGER.error('Input file is missing "events" TTree!\nAborting...')
+            sys.exit(3)
     return nevents