heatmap_cli 0.22.34__tar.gz → 0.22.35__tar.gz

{heatmap_cli-0.22.34/src/heatmap_cli.egg-info → heatmap_cli-0.22.35}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: heatmap_cli
-Version: 0.22.34
+Version: 0.22.35
 Summary: A console program that generates a yearly calendar heatmap.
 Author-email: Kian-Meng Ang <kianmeng@cpan.org>
 License-Expression: AGPL-3.0-or-later

{heatmap_cli-0.22.34 → heatmap_cli-0.22.35}/pyproject.toml

@@ -25,7 +25,7 @@ classifiers = [
     "Programming Language :: Python",
     "Topic :: Scientific/Engineering :: Visualization",
 ]
-version = "0.22.34"
+version = "0.22.35"
 keywords = ["heatmap", "cli", "visualization", "calendar", "data-viz"]
 dependencies = [
     "matplotlib",

{heatmap_cli-0.22.34 → heatmap_cli-0.22.35}/src/heatmap_cli/__init__.py

@@ -15,6 +15,8 @@
 
 """A console program that generates a yearly calendar heatmap."""
 
+from __future__ import annotations
+
 import argparse
 import datetime
 import logging

{heatmap_cli-0.22.34 → heatmap_cli-0.22.35}/src/heatmap_cli/__main__.py

@@ -15,6 +15,8 @@
 
 """Main entry point when running as a module."""
 
+from __future__ import annotations
+
 from heatmap_cli.cli import main
 
 if __name__ == "__main__":

{heatmap_cli-0.22.34 → heatmap_cli-0.22.35}/src/heatmap_cli/cli.py

@@ -20,12 +20,17 @@ changelog: https://github.com/kianmeng/heatmap_cli/blob/master/CHANGELOG.md
 issues: https://github.com/kianmeng/heatmap_cli/issues
 """
 
+from __future__ import annotations
+
 import argparse
 import datetime
 import multiprocessing
 import sys
-from collections.abc import Sequence
 from itertools import zip_longest
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from collections.abc import Sequence
 
 from heatmap_cli import (
     CMAPS,
@@ -57,6 +62,7 @@ IMAGE_FORMATS = [
 logger = multiprocessing.get_logger()
 
 DEFAULT_CMAP = "RdYlGn_r"
+CMAP_ITEMS_PER_ROW = 6
 
 
 def _valid_date(s: str) -> str:
@@ -361,9 +367,8 @@ def _generate_cmap_help() -> str:
     cmap_default = f" (default: '{DEFAULT_CMAP}')"
 
     # Always show all colormaps in help text
-    items_per_row = 6
     cmap_choices = "\nAvailable colormaps:\n"
-    cmap_bygroups = zip_longest(*(iter(CMAPS),) * items_per_row)
+    cmap_bygroups = zip_longest(*(iter(CMAPS),) * CMAP_ITEMS_PER_ROW)
     for cmap_bygroup in cmap_bygroups:
         cmap_choices += "  " + ", ".join(filter(None, cmap_bygroup)) + "\n"
 

{heatmap_cli-0.22.34 → heatmap_cli-0.22.35}/src/heatmap_cli/heatmap.py

@@ -15,7 +15,8 @@
 
 """Main logic to generate heatmap."""
 
-import argparse
+from __future__ import annotations
+
 import copy
 import io
 import multiprocessing
@@ -24,7 +25,10 @@ import shutil
 import webbrowser
 from datetime import datetime, timedelta, timezone
 from pathlib import Path
-from typing import Any
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    import argparse
 
 import matplotlib as mpl
 import matplotlib.pyplot as plt
@@ -544,7 +548,11 @@ def _configure_cbar(res: plt.Axes) -> None:
     Args:
         res (plt.Axes): The matplotlib axes object returned by seaborn.
     """
+    if not res.collections:
+        return
     cbar = res.collections[0].colorbar
+    if cbar is None:
+        return
     cbar.set_label(cbar.ax.get_xlabel(), rotation=0, labelpad=8, loc="left")
 
 
@@ -625,7 +633,7 @@ def _open_heatmap(filename: Path) -> None:
     Returns:
         None
     """
-    file_uri = f"file://{filename.resolve()}"
+    file_uri = filename.resolve().as_uri()
     webbrowser.open(file_uri)
     logger.info("Open heatmap: %s using default program.", filename.resolve())
 
@@ -719,27 +727,21 @@ def _refresh_output_dir(config: argparse.Namespace) -> None:
     """
     output_dir = _get_output_dir(config)
 
-    # Determine if purging is required based on --purge and --yes flags.
-    should_purge = False
     if config.purge:
-        if config.yes:
-            should_purge = True
-        else:
-            prompt = (
-                f"Are you sure to purge output folder: {output_dir}? [y/N] "
-            )
-            match input(prompt).lower():
-                case "y" | "yes":
-                    should_purge = True
-
-    # If purging is required and the directory exists, remove it.
-    if should_purge and output_dir.exists():
-        logger.info("Purging output folder: %s", output_dir.absolute())
-        try:
-            shutil.rmtree(output_dir)
-        except OSError as e:
-            logger.error("Error removing directory: %s - %s.", output_dir, e)
-            return
+        should_purge = config.yes or input(
+            f"Are you sure to purge output folder: {output_dir}? [y/N] "
+        ).lower() in {"y", "yes"}
+        if should_purge and output_dir.exists():
+            logger.info("Purging output folder: %s", output_dir.absolute())
+            try:
+                shutil.rmtree(output_dir)
+            except OSError as e:
+                logger.error(
+                    "Error removing directory: %s - %s.",
+                    output_dir,
+                    e,
+                )
+                return
 
     # Ensure the output directory exists for writing files.
     # This will create it if it doesn't exist or was just purged.

{heatmap_cli-0.22.34 → heatmap_cli-0.22.35/src/heatmap_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: heatmap_cli
-Version: 0.22.34
+Version: 0.22.35
 Summary: A console program that generates a yearly calendar heatmap.
 Author-email: Kian-Meng Ang <kianmeng@cpan.org>
 License-Expression: AGPL-3.0-or-later