Fixed #526

 * The code already have and supported the choice of worksheet
 * Enhance code for better error handling and show which worksheet is currently using/working

Signed-off-by: Chin Yeung Li <[email protected]>

Author: chinyeungli

CHANGELOG.rst

@@ -1,5 +1,12 @@
 ==============================
 Changelog
+2023-xx-xx
+    Release x.x.x
+
+    * The tool will now show which worksheet (if .xlsx input) is the tool working on
+    * Error handling if defined worksheet does not exist
+
+
 2023-03-09
     Release 8.0.0
 

src/attributecode/util.py

@@ -680,10 +680,16 @@ def load_excel(location, worksheet=None):
     # This is to prevent showing the: warn("Workbook contains no default style, apply openpyxl's default")
     with warnings.catch_warnings(record=True):
         input_bom = openpyxl.load_workbook(location)
+        sheetnames = input_bom.sheetnames
         if worksheet:
+            if worksheet not in sheetnames:
+                import sys
+                print("The input worksheet name does not exist. Exiting.")
+                sys.exit(1)
             sheet_obj = input_bom[worksheet]
         else:
             sheet_obj = input_bom.active
+    print("Working on the " + sheet_obj.title + " worksheet.")
     max_col = sheet_obj.max_column
 
     index = 1