fotolab 0.28.6__py3-none-any.whl → 0.29.1__py3-none-any.whl

fotolab/__init__.py

@@ -24,7 +24,7 @@ from pathlib import Path
 
 from PIL import Image
 
-__version__ = "0.28.6"
+__version__ = "0.29.1"
 
 log = logging.getLogger(__name__)
 

fotolab/cli.py

@@ -115,6 +115,7 @@ def build_parser() -> argparse.ArgumentParser:
     subparsers = parser.add_subparsers(
         help="sub-command help",
         dest="command",
+        required=True,
     )
     fotolab.subcommands.build_subparser(subparsers)
 
@@ -128,16 +129,24 @@ def main(args: Optional[Sequence[str]] = None) -> None:
 
     try:
         parser = build_parser()
-        if len(args) == 0:
-            parser.print_help(sys.stderr)
-            return
-
         parsed_args = parser.parse_args(args)
         setup_logging(parsed_args)
 
-        if hasattr(parsed_args, "func"):
+        if parsed_args.command is not None:
             log.debug(parsed_args)
-            parsed_args.func(parsed_args)
+            # Ensure the function attribute exists (set by set_defaults in
+            # subcommands)
+            if hasattr(parsed_args, "func"):
+                parsed_args.func(parsed_args)
+            else:
+                # This case should ideally not happen if subcommands are set up
+                # correctly
+                log.error(
+                    "subcommand '%s' is missing its execution function.",
+                    parsed_args.command
+                )
+                parser.print_help(sys.stderr)
+                raise SystemExit(1)
         else:
             parser.print_help(sys.stderr)
 

fotolab/subcommands/halftone.py

@@ -18,6 +18,7 @@
 import argparse
 import logging
 import math
+from typing import NamedTuple
 
 from PIL import Image, ImageDraw
 
@@ -26,6 +27,14 @@ from fotolab import save_gif_image, save_image
 log = logging.getLogger(__name__)
 
 
+class HalftoneCell(NamedTuple):
+    """Represents a cell in the halftone grid."""
+
+    col: int
+    row: int
+    cellsize: float
+
+
 def build_subparser(subparsers) -> None:
     """Build the subparser."""
     halftone_parser = subparsers.add_parser(
@@ -69,6 +78,24 @@ def build_subparser(subparsers) -> None:
         help="set default output folder (default: '%(default)s')",
     )
 
+    halftone_parser.add_argument(
+        "-c",
+        "--cells",
+        dest="cells",
+        type=int,
+        default=50,
+        help="set number of cells across the image width (default: %(default)s)",
+    )
+
+    halftone_parser.add_argument(
+        "-g",
+        "--grayscale",
+        default=False,
+        action="store_true",
+        dest="grayscale",
+        help="convert image to grayscale before applying halftone",
+    )
+
 
 def run(args: argparse.Namespace) -> None:
     """Run halftone subcommand.
@@ -83,7 +110,9 @@ def run(args: argparse.Namespace) -> None:
 
     for image_filename in args.image_filenames:
         original_image = Image.open(image_filename)
-        halftone_image = create_halftone_image(original_image)
+        halftone_image = create_halftone_image(
+            original_image, args.cells, args.grayscale
+        )
 
         if args.before_after:
             save_gif_image(
@@ -97,8 +126,58 @@ def run(args: argparse.Namespace) -> None:
             save_image(args, halftone_image, image_filename, "halftone")
 
 
+def _draw_halftone_dot(
+    draw: ImageDraw.ImageDraw,
+    source_image: Image.Image,
+    cell: HalftoneCell,
+    grayscale: bool,
+    fill_color_dot,
+) -> None:
+    """Calculate properties and draw a single halftone dot."""
+    # Calculate center point of current cell
+    img_width, img_height = source_image.size
+
+    # Calculate center point of current cell and clamp to image bounds
+    x = min(int(cell.col * cell.cellsize + cell.cellsize / 2), img_width - 1)
+    y = min(int(cell.row * cell.cellsize + cell.cellsize / 2), img_height - 1)
+
+    # Ensure coordinates are non-negative (shouldn't happen with current logic, but safe)
+    x = max(0, x)
+    y = max(0, y)
+
+    # Get pixel value (brightness or color) from the source image using clamped coordinates
+    pixel_value = source_image.getpixel((x, y))
+
+    if grayscale:
+        brightness = pixel_value
+        dot_fill = fill_color_dot  # Use white for grayscale dots
+    else:
+        # Calculate brightness (luminance) from the RGB color
+        brightness = int(
+            0.299 * pixel_value[0]
+            + 0.587 * pixel_value[1]
+            + 0.114 * pixel_value[2]
+        )
+        dot_fill = pixel_value  # Use original color for color dots
+
+    # Calculate dot radius relative to cell size based on brightness
+    # Max radius is half the cell size. Scale by brightness (0-255).
+    dot_radius = (brightness / 255.0) * (cell.cellsize / 2)
+
+    # Draw the dot
+    draw.ellipse(
+        [
+            x - dot_radius,
+            y - dot_radius,
+            x + dot_radius,
+            y + dot_radius,
+        ],
+        fill=dot_fill,
+    )
+
+
 def create_halftone_image(
-    original_image: Image.Image, cell_count: int = 50
+    original_image: Image.Image, cell_count: int, grayscale: bool = False
 ) -> Image.Image:
     """Create a halftone version of the input image.
 
@@ -107,39 +186,44 @@ def create_halftone_image(
 
     Args:
         original_image: The source image to convert
-        cell_count: Number of cells across the width (default: 50)
+        cell_count: Number of cells across the width
+        grayscale: Whether to convert to grayscale first (default: False)
 
     Returns:
         Image.Image: The halftone converted image
     """
-    grayscale_image = original_image.convert("L")
+    if grayscale:
+        source_image = original_image.convert("L")
+        output_mode = "L"
+        fill_color_black = 0
+        fill_color_dot = 255
+    else:
+        source_image = original_image.convert("RGB")
+        output_mode = "RGB"
+        fill_color_black = (0, 0, 0)
+        fill_color_dot = None  # Will be set inside the loop for color images
+
     width, height = original_image.size
 
-    halftone_image = Image.new("L", (width, height), "black")
+    # Create a new image for the output, initialized to black
+    halftone_image = Image.new(output_mode, (width, height), fill_color_black)
     draw = ImageDraw.Draw(halftone_image)
 
     cellsize = width / cell_count
     rowtotal = math.ceil(height / cellsize)
 
+    # Determine the fill color for dots once if grayscale
+    grayscale_fill_color_dot = fill_color_dot if grayscale else None
+
     for row in range(rowtotal):
         for col in range(cell_count):
-            # Calculate center point of current cell
-            x = int(col * cellsize + cellsize / 2)
-            y = int(row * cellsize + cellsize / 2)
-
-            # Get brightness and calculate dot size
-            brightness = grayscale_image.getpixel((x, y))
-            dot_size = 10 * brightness / 200
-
-            # Draw the dot
-            draw.ellipse(
-                [
-                    x - dot_size / 2,
-                    y - dot_size / 2,
-                    x + dot_size / 2,
-                    y + dot_size / 2,
-                ],
-                fill=255,
+            cell = HalftoneCell(col=col, row=row, cellsize=cellsize)
+            _draw_halftone_dot(
+                draw,
+                source_image,
+                cell,
+                grayscale,
+                grayscale_fill_color_dot,
             )
 
     return halftone_image

fotolab/subcommands/info.py

@@ -74,25 +74,41 @@ def run(args: argparse.Namespace) -> None:
     """
     log.debug(args)
 
-    info = []
+    # TODO: Add error handling for file open
+    # TODO: Use context manager `with Image.open(...)`
     image = Image.open(args.image_filename)
 
+    exif_tags = extract_exif_tags(image, args.sort)
+
+    if not exif_tags:
+        print("No metadata found!")
+        # Close the image if opened outside a 'with' block
+        image.close()
+        return
+
+    output_info = []
+    specific_info_requested = False
+
     if args.camera:
-        info.append(camera_metadata(image))
+        specific_info_requested = True
+        # TODO: Add error handling for missing keys
+        output_info.append(camera_metadata(exif_tags))
 
     if args.datetime:
-        info.append(datetime(image))
+        specific_info_requested = True
+        # TODO: Add error handling for missing keys
+        output_info.append(datetime(exif_tags))
 
-    if info:
-        print("\n".join(info))
+    if specific_info_requested:
+        print("\n".join(output_info))
     else:
-        exif_tags = extract_exif_tags(image)
-        if exif_tags:
-            tag_name_width = max(map(len, exif_tags))
-            for tag_name, tag_value in exif_tags.items():
-                print(f"{tag_name:<{tag_name_width}}: {tag_value}")
-        else:
-            print("No metadata found!")
+        # Print all tags if no specific info was requested
+        tag_name_width = max(map(len, exif_tags))
+        for tag_name, tag_value in exif_tags.items():
+            print(f"{tag_name:<{tag_name_width}}: {tag_value}")
+
+    # Close the image if opened outside a 'with' block
+    image.close()
 
 
 def extract_exif_tags(image: Image.Image, sort: bool = False) -> dict:
@@ -117,14 +133,16 @@ def extract_exif_tags(image: Image.Image, sort: bool = False) -> dict:
     return filtered_info
 
 
-def datetime(image: Image.Image):
+def datetime(exif_tags: dict):
     """Extract datetime metadata."""
-    exif_tags = extract_exif_tags(image)
+    # TODO: Add error handling for missing key
     return exif_tags["DateTime"]
 
 
-def camera_metadata(image: Image.Image):
+def camera_metadata(exif_tags: dict):
     """Extract camera and model metadata."""
-    exif_tags = extract_exif_tags(image)
-    metadata = f'{exif_tags["Make"]} {exif_tags["Model"]}'
+    # TODO: Add error handling for missing keys
+    make = exif_tags.get("Make", "")
+    model = exif_tags.get("Model", "")
+    metadata = f"{make} {model}"
     return metadata.strip()

fotolab/subcommands/watermark.py

@@ -281,9 +281,17 @@ def watermark_image(
 
 def prepare_text(args: argparse.Namespace, image: Image.Image) -> str:
     """Prepare the watermark text."""
-    text: str = args.text
-    if args.camera and camera_metadata(image):
-        text = camera_metadata(image)
+    text: str = args.text  # Default text
+    if args.camera:
+        metadata_text: str | None = camera_metadata(image)
+        if metadata_text:  # Use metadata only if it's not None or empty
+            text = metadata_text
+        else:
+            log.warning(
+                "Camera metadata requested but not found or empty; "
+                "falling back to default text: '%s'",
+                args.text,
+            )
 
     if args.lowercase:
         text = text.lower()

{fotolab-0.28.6.dist-info → fotolab-0.29.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fotolab
-Version: 0.28.6
+Version: 0.29.1
 Summary: A console program that manipulate images.
 Keywords: photography,photo
 Author-email: Kian-Meng Ang <kianmeng@cpan.org>
@@ -75,7 +75,7 @@ positional arguments:
     animate         animate an image
     auto            auto adjust (resize, contrast, and watermark) a photo
     border          add border to image
-    contrast        contrast an image
+    contrast        contrast an image.
     env             print environment information for bug reporting
     halftone        halftone an image
     info            info an image
@@ -200,8 +200,8 @@ positional arguments:
 
 options:
   -h, --help            show this help message and exit
-  -c, --cutoff CUTOFF   set the percentage of lightest or darkest pixels to
-                        discard from histogram (default: '1')
+  -c, --cutoff CUTOFF   set the percentage (0-50) of lightest or darkest
+                        pixels to discard from histogram (default: '1.0')
   -op, --open           open the image using default program (default:
                         'False')
   -od, --output-dir OUTPUT_DIR

{fotolab-0.28.6.dist-info → fotolab-0.29.1.dist-info}/RECORD

@@ -1,21 +1,21 @@
-fotolab/__init__.py,sha256=yiRE-ZW8LgYnZAJ7wKNY_XdaaPC9IbHRMnlRKLYjsdA,3262
+fotolab/__init__.py,sha256=KwMFpmP-DDvkiBIx6zr9XGWSONGzZo5Yr0aHKjYUmlg,3262
 fotolab/__main__.py,sha256=aboOURPs_snOXTEWYR0q8oq1UTY9e-NxCd1j33V0wHI,833
-fotolab/cli.py,sha256=ffT3CbMN37Qehw4l7YvwUbPFY-SezLLn4Xthi0ejOmk,3956
+fotolab/cli.py,sha256=NH_u73SJhzwDJSRqpi1I4dc8wBfcqbDxNne3cio163A,4411
 fotolab/subcommands/__init__.py,sha256=l3DlIaJ3u3jGjnC1H1yV8LZ_nPqOLJ6gikD4BCaMAQ0,1129
 fotolab/subcommands/animate.py,sha256=OcS6Q6JM1TW6HF0dSgBsV9mpEGwDMcJlQssupkf0_ZA,3393
 fotolab/subcommands/auto.py,sha256=ia-xegV1Z4HvYsbKgmTzf1NfNFdTDPWfZe7vQ1_90Ik,2425
 fotolab/subcommands/border.py,sha256=BS3BHytdWiNumxdKulKYK-WigbsKtPxECdvInUhUjSQ,4608
 fotolab/subcommands/contrast.py,sha256=ZHTvAJhRYZjNTrkHRZq3O6wba3LS7QjH1BfiGf4ZvGY,3005
 fotolab/subcommands/env.py,sha256=QoxRvzZKgmoHTUxDV4QYhdChCpMWs5TbXFY_qIpIQpE,1469
-fotolab/subcommands/halftone.py,sha256=bRGG_Bg1T5ZV3UWsdMszOdPmc6g9_R0yNp5mb0g8hcE,4152
-fotolab/subcommands/info.py,sha256=Qp-Zu7Xp1ptLK211hOkFM5oxZWGTrlNBqCpbx2IjL9Y,3371
+fotolab/subcommands/halftone.py,sha256=YOGtdl6N3aC4E47ac1rGzHU0HczVsujKNIBMyiRf2-c,6681
+fotolab/subcommands/info.py,sha256=EYuU06-rrnZZMY0XKwaIiqLypbd6x9KmCydiB1UhtTU,4010
 fotolab/subcommands/montage.py,sha256=d_3EcyRSFS8fKkczlHO8IRP-mrKhQUtkQndjfd0MKsc,2566
 fotolab/subcommands/resize.py,sha256=UOb2rg_5ArRj0gxPTDOww_ix3tqJRC0W5b_S-Lt1G4w,5444
 fotolab/subcommands/rotate.py,sha256=uBFjHyjiBSQLtrtH1p9myODIHUDr1gkL4PpU-6Y1Ofo,2575
 fotolab/subcommands/sharpen.py,sha256=YNho2IPbc-lPvSy3Bsjehc2JOEy27LPqFSGRULs9MyY,3492
-fotolab/subcommands/watermark.py,sha256=iEuek0BM2PJlGMLq8VZhHxaQ8BInuN2_mvssA7Dhgy8,10032
-fotolab-0.28.6.dist-info/entry_points.txt,sha256=mvw7AY_yZkIyjAxPtHNed9X99NZeLnMxEeAfEJUbrCM,44
-fotolab-0.28.6.dist-info/LICENSE.md,sha256=tGtFDwxWTjuR9syrJoSv1Hiffd2u8Tu8cYClfrXS_YU,31956
-fotolab-0.28.6.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
-fotolab-0.28.6.dist-info/METADATA,sha256=3E5duvvv00UeRQoiWsueL2cdsEj1l17jdkwxY7Nal68,12970
-fotolab-0.28.6.dist-info/RECORD,,
+fotolab/subcommands/watermark.py,sha256=qRGUp1Lc22fZSJDFRqQGPiz8RSB293ebvOVTdsDLUE4,10351
+fotolab-0.29.1.dist-info/entry_points.txt,sha256=mvw7AY_yZkIyjAxPtHNed9X99NZeLnMxEeAfEJUbrCM,44
+fotolab-0.29.1.dist-info/LICENSE.md,sha256=tGtFDwxWTjuR9syrJoSv1Hiffd2u8Tu8cYClfrXS_YU,31956
+fotolab-0.29.1.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
+fotolab-0.29.1.dist-info/METADATA,sha256=0jUgjFRshumTtyFXLKe6_sQVnpyGLNez-Fl-KB5vHMk,12980
+fotolab-0.29.1.dist-info/RECORD,,