Export fixes (#1060)

core/schemas/task.py

@@ -59,7 +59,7 @@ class Task(YetiModel, database_arango.ArangoYetiConnector):
     # only used for cron tasks
     frequency: datetime.timedelta | None = None
 
-    def run(self, params: "TaskParams"):
+    def run(self, params: dict):
         """Runs the task"""
         raise NotImplementedError("run() must be implemented in subclass")
 

core/schemas/template.py

@@ -34,5 +34,6 @@ def render(self, data: list["Observable"], output_file: str | None) -> None | st
             os.makedirs(os.path.dirname(output_file), exist_ok=True)
             with open(output_file, "w+") as fd:
                 fd.write(result)
+            return None
         else:
             return result

core/web/apiv2/tasks.py

@@ -109,15 +109,13 @@ async def patch_export(request: PatchExportRequest) -> ExportTask:
     return new
 
 
-@router.get("/export/{export_name}/content")
-async def export_content(export_name: str):
+@router.get("/export/{export_id}/content")
+async def export_content(export_id: str):
     """Downloads the latest contents of a given ExportTask."""
-    export = ExportTask.find(name=export_name)
+    export = ExportTask.get(export_id)
     if not export:
-        raise HTTPException(
-            status_code=404, detail=f"ExportTask {export_name} not found"
-        )
-    return FileResponse(export.output_file)
+        raise HTTPException(status_code=404, detail=f"ExportTask {export_id} not found")
+    return FileResponse(export.output_file, headers={"Cache-Control": "no-cache"})
 
 
 @router.delete("/export/{export_name}")