heatmap_cli 0.22.33__tar.gz → 0.22.35__tar.gz

{heatmap_cli-0.22.33/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.33
+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.33 → heatmap_cli-0.22.35}/pyproject.toml

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

{heatmap_cli-0.22.33 → 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
@@ -56,7 +58,7 @@ def setup_logging(config: argparse.Namespace) -> None:
 
     match config:
         case argparse.Namespace(quiet=True):
-            logging.disable(logging.NOTSET)
+            logging.disable()
             return
         case argparse.Namespace(debug=True):
             level = logging.DEBUG

{heatmap_cli-0.22.33 → 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.33 → 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"
 
@@ -438,10 +443,6 @@ def main(args: Sequence[str] | None = None) -> None:
     """
     argv = args if args is not None else sys.argv[1:]
     parser = build_parser()
-    # Use parse_known_args to potentially capture debug flag even if other args
-    # fail. However, we'll rely on the full parse_args within the try block for
-    # simplicity here.
-    parsed_args = None
     debug_active = False  # Initialize to False, will be set after parsing
 
     try:

{heatmap_cli-0.22.33 → 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
@@ -73,7 +77,6 @@ COLORBAR_ANNOTATION_SUFFIX = ", count: nearest hundred"
 
 # Define constants for count truncation
 TRUNCATE_DIVISOR = 100
-TRUNCATE_ROUND_DECIMALS = -2
 
 # Define constants for week ranges
 ALL_WEEKS_RANGE = range(1, 54)
@@ -375,7 +378,7 @@ def _filter_and_pivot_data(
 def _truncate_rounded_count(count: float) -> int:
     """Truncate and round count values to fit them in heatmap box.
 
-    Rounds the count to the nearest hundred and then divides by 100.
+    Divides the count by 100 and rounds to the nearest integer.
     This transformation is applied when annotations are enabled to simplify
     large numbers and make them fit visually within the heatmap cells.
     For example, 12345 becomes 123, 5678 becomes 57.
@@ -386,7 +389,7 @@ def _truncate_rounded_count(count: float) -> int:
     Returns:
         int: Truncated count value (divided by 100).
     """
-    return int(round(count, TRUNCATE_ROUND_DECIMALS) / TRUNCATE_DIVISOR)
+    return round(count / TRUNCATE_DIVISOR)
 
 
 def _setup_plot_artists(
@@ -401,7 +404,7 @@ def _setup_plot_artists(
     Args:
         cmap (str): Colormap name used for the heatmap.
         config (argparse.Namespace): Config from command line arguments.
-        dataframe (pd.core.frame.DataFrame): DataFrame with data.
+        dataframe (pd.DataFrame): DataFrame with data.
 
     Returns:
         plt.Axes: The Axes object with the heatmap plotted.
@@ -444,7 +447,7 @@ def _generate_heatmap(
         seq (int): Sequence number for generated heatmap image file.
         cmap (str): Colormap name used for the heatmap.
         config (argparse.Namespace): Config from command line arguments.
-        dataframe (pd.core.frame.DataFrame): DataFrame with data loaded from
+        dataframe (pd.DataFrame): DataFrame with data loaded from
         CSV file.
         return_image (bool): If True, returns a PIL Image object instead of
         saving to disk.
@@ -506,9 +509,9 @@ def _configure_heatmap_options(
         "cbar_kws": cbar_options,
     }
 
-    if config.cmap_min:
+    if config.cmap_min is not None:
         options["vmin"] = config.cmap_min
-    if config.cmap_max:
+    if config.cmap_max is not None:
         options["vmax"] = config.cmap_max
 
     if config.annotate:
@@ -545,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")
 
 
@@ -626,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())
 
@@ -720,29 +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)
-            # Depending on desired error handling, you might want to raise an
-            # exception here.
-            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.
@@ -751,7 +750,7 @@ def _refresh_output_dir(config: argparse.Namespace) -> None:
 
 
 def _get_output_dir(config: argparse.Namespace) -> Path:
-    """Get the current working directory.
+    """Get the output directory path, resolved relative to CWD.
 
     Args:
         config (argparse.Namespace): Config from command line arguments.

{heatmap_cli-0.22.33 → 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.33
+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