fix stats crash

boldandbrad · Oct 27, 2023 · 9469094 · 9469094
1 parent c23241a
commit 9469094
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/meeple/command/stats.py b/src/meeple/command/stats.py
@@ -9,7 +9,7 @@
     invalid_collection_error,
     warn_msg,
 )
-from meeple.util.output_util import print_table
+from meeple.util.output_util import ItemHeader, print_table
 
 
 @click.command()
@@ -103,7 +103,12 @@ def stats(collection_name: str, item_type: str) -> None:
     info_msg(
         f"Showing average stats for {len(result_items)} of {len(collection.data.items)} items from collection {collection.fmt_name()}."
     )
-    headers = ["Rank", "Rating", "Weight", "Max Players"]
+    headers = [
+        ItemHeader.RANK,
+        ItemHeader.RATING,
+        ItemHeader.WEIGHT,
+        ItemHeader.MAX_PLAYERS,
+    ]
     rows = [
         [
             f"{fmt_avg_rank(avg_rank)}",

diff --git a/src/meeple/util/output_util.py b/src/meeple/util/output_util.py
@@ -24,6 +24,7 @@ class ItemHeader(Enum):
     RATING = ("Rating", "rating", "right")
     WEIGHT = ("Weight", "weight", "right")
     PLAYERS = ("Players", "players", "right")
+    MAX_PLAYERS = ("Max Players", "maxplayers", "right")
     TIME = ("Play Time", "time", "right")