feat: optional logging

1 files changed, 6 insertions, 3 deletions

diff --git a/ami.py b/ami.py
index 21062b5..3d510dd 100644
--- a/ami.py
+++ b/ami.py
@@ -23,9 +23,12 @@ target_cities = environ.get("TARGET_CITIES") or ""
 
 
 def log_response(data):
-    with open("jobs.json", "a") as log_file:
-        json.dump({"timestamp": datetime.now().isoformat(), "data": data}, log_file)
-        log_file.write("\n")
+    log = environ.get("LOG")
+
+    if log is not None:
+        with open(log, "a") as log_file:
+            json.dump({"timestamp": datetime.now().isoformat(), "data": data}, log_file)
+            log_file.write("\n")
 
 
 def notify(title, body, job_id=None):