fix(splat): wipe tmp directory after run

uptick · Nov 13, 2024 · 1af3da9 · 1af3da9
1 parent 265303a
commit 1af3da9
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/lambda_function.py b/lambda_function.py
@@ -3,6 +3,7 @@
 import json
 import logging
 import os
+import pathlib
 import subprocess
 import sys
 import tempfile
@@ -407,6 +408,12 @@ def lambda_handler(event: dict, context: dict) -> dict:  # noqa
     except Exception as e:
         logger.error(f"splat|unknown_error|{str(e)}|stacktrace:", exc_info=True)
         resp = SplatPDFGenerationFailure(status_code=500, message=str(e)).as_response().as_dict()
+    finally:
+        try:
+            execute(["rm", "-rf", "/tmp/*"])
+        except Exception as e:
+            logger.error(f"splat|cleanup_error|{str(e)}|stacktrace:", exc_info=True)
+
     return resp