feat(series): Enhance tree representation with season breakdown

This commit is contained in:
Andy
2025-08-04 05:55:32 +00:00
parent 3e0835d9fb
commit 63c697f082

View File

@@ -201,9 +201,10 @@ class Series(SortedKeyList, ABC):
def tree(self, verbose: bool = False) -> Tree:
seasons = Counter(x.season for x in self)
num_seasons = len(seasons)
num_episodes = sum(seasons.values())
sum(seasons.values())
season_breakdown = ", ".join(f"S{season}({count})" for season, count in sorted(seasons.items()))
tree = Tree(
f"{num_seasons} Season{['s', ''][num_seasons == 1]}, {num_episodes} Episode{['s', ''][num_episodes == 1]}",
f"{num_seasons} seasons, {season_breakdown}",
guide_style="bright_black",
)
if verbose: