Add type annotations

Author: t-persson

python/src/etos_api/library/opentelemetry.py

@@ -39,7 +39,7 @@
 from pydantic import BaseModel
 
 
-def setup_opentelemetry(app, version: str):
+def setup_opentelemetry(app, version: str) -> Resource:
     """Set up OpenTelemetry for ETOS API."""
     otel_resource = Resource.create(
         {

python/src/etos_api/routers/v0/router.py

@@ -55,7 +55,7 @@
 async def start_etos(
     etos: StartEtosRequest,
     ctx: Annotated[otel_context.Context, Depends(context)],
-):
+) -> dict:
     """Start ETOS execution on post.
 
     :param etos: ETOS pydantic model.
@@ -70,7 +70,7 @@ async def start_etos(
 
 
 @ETOSV0.delete("/etos/{suite_id}", tags=["etos"], response_model=AbortEtosResponse)
-async def abort_etos(suite_id: str, ctx: Annotated[otel_context.Context, Depends(context)]):
+async def abort_etos(suite_id: str, ctx: Annotated[otel_context.Context, Depends(context)]) -> dict:
     """Abort ETOS execution on delete.
 
     :param suite_id: ETOS suite id

python/src/etos_api/routers/v1alpha/router.py

@@ -60,7 +60,7 @@
 @ETOSV1ALPHA.post("/testrun", tags=["etos"], response_model=StartTestrunResponse)
 async def start_testrun(
     etos: StartTestrunRequest, ctx: Annotated[otel_context.Context, Depends(context)]
-):
+) -> dict:
     """Start ETOS testrun on post.
 
     :param etos: ETOS pydantic model.
@@ -75,7 +75,9 @@ async def start_testrun(
 
 
 @ETOSV1ALPHA.delete("/testrun/{suite_id}", tags=["etos"], response_model=AbortTestrunResponse)
-async def abort_testrun(suite_id: str, ctx: Annotated[otel_context.Context, Depends(context)]):
+async def abort_testrun(
+    suite_id: str, ctx: Annotated[otel_context.Context, Depends(context)]
+) -> dict:
     """Abort ETOS testrun on delete.
 
     :param suite_id: ETOS suite id