update xml tag for table row count to resolve #131 (#132)

USEPA · Jan 20, 2023 · 7821acb · 7821acb
1 parent 5b69640
commit 7821acb
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/stewi/GHGRP.py b/stewi/GHGRP.py
@@ -126,10 +126,11 @@ def get_row_count(table, report_year):
     try:
         count_request = requests.get(count_url)
         count_xml = minidom.parseString(count_request.text)
-        table_count = count_xml.getElementsByTagName('RequestRecordCount')
+        table_count = count_xml.getElementsByTagName('TOTALQUERYRESULTS')
         table_count = int(table_count[0].firstChild.nodeValue)
-    except:
-        log.exception('error in url request')
+    except IndexError:
+        log.exception('error accessing table count')
+        raise
     return table_count