feat(send): SEND_URL and SIGN_KEY now settable with build-vars

gadicc · gadicc · commit 01cf35461c58 · 2022-11-26T11:30:59.000Z
diff --git a/Dockerfile b/Dockerfile
@@ -163,5 +163,10 @@ ADD train_dreambooth.py .
 ADD send.py .
 ADD app.py .
 
+ARG SEND_URL
+ENV SEND_URL=${SEND_URL}
+ARG SIGN_KEY
+ENV SIGN_KEY=${SIGN_KEY}
+
 CMD python3 -u server.py
 
diff --git a/send.py b/send.py
@@ -16,7 +16,13 @@ def get_now():
 
 
 send_url = os.getenv("SEND_URL")
+if send_url == "":
+    send_url = None
+
 sign_key = os.getenv("SIGN_KEY")
+if sign_key == "":
+    sign_key = None
+
 init_time = get_now()
 session = FuturesSession()
 last_time = init_time
