Feature/pre commit

.github/workflows/test.yml → .github/workflows/ci.yml

@@ -27,9 +27,13 @@ jobs:
       run: |
         python -m pip install --upgrade pip
         pip install -r space2stats_api/requirements.txt
+        pip install pre-commit
 
     - name: Set PYTHONPATH
       run: echo "PYTHONPATH=$(pwd)/space2stats_api" >> $GITHUB_ENV
 
+    - name: Run pre-commit
+      run: pre-commit run --all-files
+
     - name: Run tests
       run: pytest space2stats_api/tests

.pre-commit-config.yaml

@@ -0,0 +1,15 @@
+repos:
+  - repo: https://github.com/charliermarsh/ruff-pre-commit
+    rev: v0.5.5  
+    hooks:
+      - id: ruff
+        args: [--fix]
+        files: ^space2stats_api/
+
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: v1.11.1   
+    hooks:
+      - id: mypy
+        args: [--ignore-missing-imports]
+        files: ^space2stats_api/
+        exclude: ^space2stats/env/

space2stats_api/app/routers/api.py

@@ -1,7 +1,7 @@
-from typing import List, Dict, Any, Literal, Optional
+from typing import List, Dict, Any, Literal, Optional, TypeAlias
 from fastapi import APIRouter
 
-from pydantic import BaseModel, create_model
+from pydantic import BaseModel
 from geojson_pydantic import Feature, Polygon
 
 from app.utils.h3_utils import generate_h3_ids, generate_h3_geometries
@@ -10,22 +10,14 @@
 
 router = APIRouter()
 
-AOIModel = Feature[Polygon, Dict]
+AOIModel: TypeAlias = Feature[Polygon, Dict]
 
 
 class SummaryRequest(BaseModel):
     aoi: AOIModel
     spatial_join_method: Literal["touches", "centroid", "within"]
     fields: List[str]
-    geometry: Optional[Literal["polygon", "point"]] = False
-
-
-def create_response_model(fields: List[str]):
-    field_definitions = {field: (Any, ...) for field in fields}
-    field_definitions["hex_id"] = (str, ...)
-    if "geometry" in fields:
-        field_definitions["geometry"] = (Dict[str, Any], ...)
-    return create_model("DynamicSummaryResponse", **field_definitions)
+    geometry: Optional[Literal["polygon", "point"]] = None
 
 
 @router.post("/summary", response_model=List[Dict[str, Any]])

space2stats_api/app/settings.py

@@ -8,8 +8,8 @@ class Settings(BaseSettings):
     DB_USER: str
     DB_PASSWORD: str
     DB_TABLE_NAME: str
-
-    model_config = SettingsConfigDict(env_file="../local_db.env")
+        
+    model_config = SettingsConfigDict(env_file='../db.env')
 
 
 settings = Settings()

space2stats_api/tests/test_api.py

@@ -3,7 +3,6 @@
 from unittest.mock import patch
 
 from app.main import app
-from app.utils.h3_utils import generate_h3_geometries
 
 client = TestClient(app)
 

space2stats_api/tests/test_h3_utils.py

@@ -1,5 +1,5 @@
 import pytest
-from shapely.geometry import Polygon, mapping, Point
+from shapely.geometry import Polygon, mapping
 from app.utils.h3_utils import generate_h3_ids, generate_h3_geometries
 
 polygon_coords = [