move to match/case and path

Author: soodoku

layoutlens/analyzer.py

@@ -197,7 +197,8 @@ def analyze_screenshot(
 
     def _encode_image(self, image_path: str) -> str:
         """Encode image to base64 string."""
-        with open(image_path, "rb") as image_file:
+        image_file_path = Path(image_path)
+        with image_file_path.open("rb") as image_file:
             return base64.b64encode(image_file.read()).decode("utf-8")
 
     def _get_prompt_template(self, instructions: Instructions | None = None) -> PromptTemplate:

layoutlens/api/core.py

@@ -833,16 +833,18 @@ def __init__(self, *args, **kwargs):
                 super().__init__(*args, directory=str(html_file_path.parent), **kwargs)
 
             def do_GET(self):
-                # If requesting root, serve our HTML file
-                if self.path == "/" or self.path == "":
-                    self.send_response(200)
-                    self.send_header("Content-type", "text/html")
-                    self.end_headers()
-                    with open(html_file_path, "rb") as f:
-                        self.wfile.write(f.read())
-                else:
-                    # Serve other files normally (CSS, JS, images)
-                    super().do_GET()
+                # Route based on path
+                match self.path:
+                    case "/" | "":
+                        # Serve our HTML file
+                        self.send_response(200)
+                        self.send_header("Content-type", "text/html")
+                        self.end_headers()
+                        with open(html_file_path, "rb") as f:
+                            self.wfile.write(f.read())
+                    case _:
+                        # Serve other files normally (CSS, JS, images)
+                        super().do_GET()
 
             def log_message(self, format, *args):
                 # Suppress server logs
@@ -1301,12 +1303,13 @@ async def capture_batch(
 
         # Validate existing files
         for file_path in existing_files:
-            if Path(file_path).exists():
-                results[str(file_path)] = str(file_path)
+            file_path_obj = Path(file_path)
+            if file_path_obj.exists():
+                results[file_path] = file_path
                 self.logger.debug(f"Using existing file: {file_path}")
             else:
                 failed_count += 1
-                results[str(file_path)] = f"Error: File not found"
+                results[file_path] = f"Error: File not found"
                 self.logger.warning(f"File not found: {file_path}")
 
         # Capture URLs using BatchCapture for efficiency
@@ -1388,11 +1391,12 @@ async def capture_batch_async(
 
         # Validate existing files
         for file_path in existing_files:
-            if Path(file_path).exists():
-                results[str(file_path)] = str(file_path)
+            file_path_obj = Path(file_path)
+            if file_path_obj.exists():
+                results[file_path] = file_path
                 self.logger.debug(f"Using existing file: {file_path}")
             else:
-                results[str(file_path)] = f"Error: File not found"
+                results[file_path] = f"Error: File not found"
                 self.logger.warning(f"File not found: {file_path}")
 
         # Capture sources concurrently

layoutlens/logger.py

@@ -84,12 +84,13 @@ def setup_logging(
         root_logger.removeHandler(handler)
 
     # Choose formatter
-    if format_type == "debug":
-        formatter = logging.Formatter(DEBUG_FORMAT)
-    elif format_type == "console":
-        formatter = logging.Formatter(CONSOLE_FORMAT)
-    else:
-        formatter = logging.Formatter(DEFAULT_FORMAT)
+    match format_type:
+        case "debug":
+            formatter = logging.Formatter(DEBUG_FORMAT)
+        case "console":
+            formatter = logging.Formatter(CONSOLE_FORMAT)
+        case _:
+            formatter = logging.Formatter(DEFAULT_FORMAT)
 
     # Console handler
     if console:

pyproject.toml

@@ -18,6 +18,7 @@ keywords = ["ui testing", "visual regression", "ai", "computer vision", "web tes
 classifiers = [
     "Development Status :: 5 - Production/Stable",
     "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
     "Programming Language :: Python :: 3",
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",