IMAGEDIFF: Implement screenshots comparison

ImageDiffwithPython/config.py

@@ -0,0 +1,3 @@
+import os
+
+SCREENSHOTS_DIR = os.environ.get("SCREENSHOTS_DIR", "./screenshots/")

ImageDiffwithPython/imagediff.py

@@ -0,0 +1,101 @@
+from PIL import Image, ImageChops
+import os
+import base64
+from io import BytesIO
+from config import SCREENSHOTS_DIR
+
+def encode_image(image):
+    """Encode an image to a base64 string."""
+    buffered = BytesIO()
+    image.save(buffered, format="PNG")
+    return base64.b64encode(buffered.getvalue()).decode('utf-8')
+
+def image_diff(src_img_path, cmp_img_path):
+    """
+    Compute the difference between two images and return the results.
+    """
+    try:
+        src_img = Image.open(src_img_path)
+        cmp_img = Image.open(cmp_img_path)
+        diff_img = ImageChops.difference(src_img, cmp_img).convert('RGB')
+
+        has_diff = diff_img.getbbox() is not None
+
+        return {
+            'src_img_data': encode_image(src_img),
+            'cmp_img_data': encode_image(cmp_img),
+            'diff_img_data': encode_image(diff_img) if has_diff else None,
+            'has_diff': has_diff
+        }
+    except IOError:
+        return {}
+
+def movie_diff(src_build, cmp_build, target, movie):
+    """
+    Compare all frames of a movie between two builds and determine if there are any differences.
+
+    Args:
+        src_build (str): The source build name
+        cmp_build (str): The comparison build name
+        target (str): The target name
+        movie (str): The movie name
+
+    Returns:
+        bool: True if any frame has differences, False otherwise
+    """
+    src_build_path = os.path.join(SCREENSHOTS_DIR, target, src_build)
+    cmp_build_path = os.path.join(SCREENSHOTS_DIR, target, cmp_build)
+
+    # Ensure both build paths exist
+    if not os.path.exists(src_build_path) or not os.path.exists(cmp_build_path):
+        return False
+
+    # Get all frames for this movie in both builds
+    src_frames = [f for f in os.listdir(src_build_path)
+                  if os.path.isfile(os.path.join(src_build_path, f)) and f.startswith(f"{movie}-")]
+
+    cmp_frames = [f for f in os.listdir(cmp_build_path)
+                  if os.path.isfile(os.path.join(cmp_build_path, f)) and f.startswith(f"{movie}-")]
+
+    # If frame counts differ, there's definitely a difference
+    if len(src_frames) != len(cmp_frames):
+        return True
+
+    # Helper function to extract frame number from filename
+    def get_frame_number(filename):
+        parts = filename.split('-')
+        if len(parts) > 1:
+            try:
+                return int(parts[1].split('.')[0])
+            except ValueError:
+                return 0
+        return 0
+
+    src_frame_map = {get_frame_number(f): f for f in src_frames}
+    cmp_frame_map = {get_frame_number(f): f for f in cmp_frames}
+
+    all_frame_numbers = sorted(set(list(src_frame_map.keys()) + list(cmp_frame_map.keys())))
+
+    if set(src_frame_map.keys()) != set(cmp_frame_map.keys()):
+        return True
+
+    for frame_num in all_frame_numbers:
+        src_frame = src_frame_map.get(frame_num)
+        cmp_frame = cmp_frame_map.get(frame_num)
+
+        if src_frame and cmp_frame:
+            src_img_path = os.path.join(src_build_path, src_frame)
+            cmp_img_path = os.path.join(cmp_build_path, cmp_frame)
+
+            try:
+                diff_result = image_diff(src_img_path, cmp_img_path)
+
+                if diff_result.get('has_diff', False):
+                    return True
+            except Exception as e:
+                print(f"Error comparing frames {src_frame} and {cmp_frame}: {e}")
+                return True
+        else:
+            return True
+
+    return False