Enable more linting and type checking (#62)

Author: robolyst

pyproject.toml

@@ -50,23 +50,24 @@ build-backend = "uv_build"
 [tool.ruff]
 exclude = [
     "examples",
+    ".devcontainer",
 ]
 
 [tool.ruff.lint]
 # Full list of rules here: https://docs.astral.sh/ruff/rules/
 select = [
     # Core rules
-    # "E",     # pycodestyle errors
+    "E",     # pycodestyle errors
     "F",     # Pyflakes
-    # "UP",    # pyupgrade
+    "UP",    # pyupgrade
     "I",     # isort
 
     # Quality and style
-    # "B",     # flake8-bugbear
+    "B",     # flake8-bugbear
     "Q",     # flake8-quotes
-    # "SIM",   # flake8-simplify
+    "SIM",   # flake8-simplify
     "FLY",   # flynt
-    # "PERF",  # Perflint
+    "PERF",  # Perflint
 
     # Disabled rules (uncomment to enable):
     # "AIR",   # Airflow
@@ -130,7 +131,10 @@ select = [
 # =============================================================================
 
 [tool.pyright]
-include = ["src"]
+include = [
+    "src",
+    "tests",
+]
 
 
 # =============================================================================

src/streetview/api.py

@@ -1,5 +1,4 @@
 from io import BytesIO
-from typing import Dict, Union
 
 import requests
 from PIL import Image
@@ -64,8 +63,8 @@ def get_streetview(
     """
 
     url = "https://maps.googleapis.com/maps/api/streetview"
-    params: Dict[str, Union[str, int]] = {
-        "size": "%dx%d" % (width, height),
+    params: dict[str, str | int] = {
+        "size": f"{width:.0f}x{height:.0f}",
         "fov": fov,
         "pitch": pitch,
         "heading": heading,

src/streetview/download.py

@@ -2,9 +2,9 @@
 import concurrent.futures
 import itertools
 import time
+from collections.abc import AsyncGenerator, Generator
 from dataclasses import dataclass
 from io import BytesIO
-from typing import AsyncGenerator, Generator, Tuple
 
 import httpx
 import requests
@@ -30,7 +30,7 @@ class Tile:
     image: Image.Image
 
 
-def get_width_and_height_from_zoom(zoom: int) -> Tuple[int, int]:
+def get_width_and_height_from_zoom(zoom: int) -> tuple[int, int]:
     """
     Returns the width and height of a panorama at a given zoom level, depends on the
     zoom level.
@@ -58,7 +58,7 @@ def fetch_panorama_tile(
         try:
             response = requests.get(tile_info.fileurl, stream=True)
             return Image.open(BytesIO(response.content))
-        except requests.ConnectionError:
+        except requests.ConnectionError:  # noqa: PERF203
             print("Connection error. Trying again in 2 seconds.")
             time.sleep(2)
     raise requests.ConnectionError("Max retries exceeded.")
@@ -75,7 +75,7 @@ async def fetch_panorama_tile_async(
             response = await async_client.get(tile_info.fileurl)
             return Image.open(BytesIO(response.content))
 
-        except httpx.RequestError as e:
+        except httpx.RequestError as e:  # noqa: PERF203
             print(f"Request error {e}. Trying again in 2 seconds.")
             await asyncio.sleep(2)
 
@@ -117,9 +117,8 @@ def iter_tiles(
             try:
                 image = future.result()
             except Exception as exc:
-                raise Exception(
-                    f"Failed to download tile {info.fileurl} due to Exception: {exc}"
-                )
+                msg = f"Failed to download tile {info.fileurl} due to Exception: {exc}"
+                raise Exception(msg) from exc
             else:
                 yield Tile(x=info.x, y=info.y, image=image)
 

src/streetview/search.py

@@ -1,6 +1,5 @@
 import json
 import re
-from typing import List, Optional
 
 import requests
 from pydantic import BaseModel
@@ -12,10 +11,10 @@ class Panorama(BaseModel):
     lat: float
     lon: float
     heading: float
-    pitch: Optional[float]
-    roll: Optional[float]
-    date: Optional[str]
-    elevation: Optional[float]
+    pitch: float | None
+    roll: float | None
+    date: str | None
+    elevation: float | None
 
 
 def make_search_url(lat: float, lon: float) -> str:
@@ -43,7 +42,7 @@ def search_request(lat: float, lon: float) -> Response:
     return requests.get(url)
 
 
-def extract_panoramas(text: str) -> List[Panorama]:
+def extract_panoramas(text: str) -> list[Panorama]:
     """
     Given a valid response from the panoids endpoint, return a list of all the
     panoids.
@@ -61,10 +60,7 @@ def extract_panoramas(text: str) -> List[Panorama]:
 
     raw_panos = subset[3][0]
 
-    if len(subset) < 9 or subset[8] is None:
-        raw_dates = []
-    else:
-        raw_dates = subset[8]
+    raw_dates = [] if (len(subset) < 9 or subset[8] is None) else subset[8]
 
     # For some reason, dates do not include a date for each panorama.
     # the n dates match the last n panos. Here we flip the arrays
@@ -89,7 +85,7 @@ def extract_panoramas(text: str) -> List[Panorama]:
     ]
 
 
-def search_panoramas(lat: float, lon: float) -> List[Panorama]:
+def search_panoramas(lat: float, lon: float) -> list[Panorama]:
     """
     Gets the closest panoramas (ids) to the GPS coordinates.
     """