From 943d0fde0b46b8da93e10d6c65e03b3052eb3130 Mon Sep 17 00:00:00 2001
From: lonkaars <loek@pipeframe.xyz>
Date: Sun, 5 Feb 2023 18:21:06 +0100
Subject: only write debug api output if `debug` file is present

---
 .gitignore     | 1 +
 autoplanner.py | 4 +---
 shared.py      | 7 +++++++
 storingen.py   | 4 +---
 4 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/.gitignore b/.gitignore
index 0a17bdf..32e1c0f 100644
--- a/.gitignore
+++ b/.gitignore
@@ -3,3 +3,4 @@ autoplanner.json
 venv/
 __pycache__/
 *-api-response.json
+debug
diff --git a/autoplanner.py b/autoplanner.py
index fe3e728..6af5c6d 100755
--- a/autoplanner.py
+++ b/autoplanner.py
@@ -52,9 +52,7 @@ def get_trip(date):
   response = conn.getresponse()
   data = response.read()
   conn.close()
-  f = open("./autoplanner-api-response.json", "w+")
-  f.write(str(data, "utf-8"))
-  f.close()
+  debug_output("autoplanner", data)
   return data
 
 def leg2desc(leg):
diff --git a/shared.py b/shared.py
index 72db189..0eeee53 100644
--- a/shared.py
+++ b/shared.py
@@ -1,6 +1,7 @@
 #!/bin/python3
 import urllib.request
 import re
+import os
 from uuid import uuid4
 from bs4 import BeautifulSoup
 
@@ -15,3 +16,9 @@ def read_file(filename):
   r = str(f.read())
   f.close()
   return r
+
+def debug_output(name, data):
+  if not os.path.exists("./debug"): return
+  f = open(f"./{name}-api-response.json", "w+")
+  f.write(str(data, "utf-8"))
+  f.close()
diff --git a/storingen.py b/storingen.py
index 5eb9af7..2a93936 100755
--- a/storingen.py
+++ b/storingen.py
@@ -19,9 +19,7 @@ def get_disruptions():
   response = conn.getresponse()
   data = response.read()
   conn.close()
-  f = open("./storingen-api-response.json", "w+")
-  f.write(str(data, "utf-8"))
-  f.close()
+  debug_output("storingen", data)
   return data
 
 def disruption2ical(disruption):
-- 
cgit v1.2.3