Refine table printing logic

ToucanToco · Sep 3, 2024 · 96f74a1 · 96f74a1
1 parent 4cafbfb
commit 96f74a1
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/test.py b/test.py
@@ -28,11 +28,16 @@ def main():
     use_columns = args.column or None
 
     if args.print_tables:
-        print(excel_file.table_names())
+        table_names = excel_file.table_names()
+        if len(table_names) > 0:
+            print(f"Available tables are {', '.join(table_names)}")
+        else:
+            print("No tables found")
 
     for _ in range(args.iterations):
         if args.table:
-            tbl = excel_file.load_table(args.table, header_row=0)
+            tbl = excel_file.load_table(args.table)
+            print(f"Found table {args.table}:")
             print(tbl.to_polars())
         else:
             for sheet_name in excel_file.sheet_names: