Skip to content

Commit

Permalink
read-excel error handling when header flag is not found
Browse files Browse the repository at this point in the history
  • Loading branch information
Shettland committed May 28, 2024
1 parent 2320724 commit 8a78ef7
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions relecov_tools/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,9 +64,12 @@ def read_excel_file(f_name, sheet_name, header_flag, leave_empty=True):
"""
wb_file = openpyxl.load_workbook(f_name, data_only=True)
ws_metadata_lab = wb_file[sheet_name]
heading_row = [
idx + 1 for idx, x in enumerate(ws_metadata_lab.values) if header_flag in x
][0]
try:
heading_row = [
idx + 1 for idx, x in enumerate(ws_metadata_lab.values) if header_flag in x
][0]
except IndexError:
raise KeyError(f"Header flag '{header_flag}' could not be found in {f_name}")
heading = [str(i.value).strip() for i in ws_metadata_lab[heading_row] if i.value]
ws_data = []
for row in islice(ws_metadata_lab.values, heading_row, ws_metadata_lab.max_row):
Expand Down

0 comments on commit 8a78ef7

Please sign in to comment.