feat: add loop support for container

This commit is contained in:
2024-02-17 17:03:14 +01:00
parent 7d3c449c16
commit d6a2f8ec15

View File

@@ -1,6 +1,7 @@
#!/usr/bin/python3
import sys
import time
import argparse
import subprocess
import os
@@ -121,6 +122,8 @@ if __name__ == "__main__":
parser.add_argument('--smtp-user')
parser.add_argument('--smtp-pass')
parser.add_argument('--loop', default=True, action=argparse.BooleanOptionalAction)
args = parser.parse_args()
# set dispatch server & authentication #
@@ -142,6 +145,9 @@ if __name__ == "__main__":
smtp_user = args.smtp_user or os.environ.get("SMTP_USER")
smtp_pass = args.smtp_pass or os.environ.get("SMTP_PASS")
first_run = True
while args.loop or first_run:
# request dispatches #
response = requests.get(args.dispatch_server + "/get-dispatch?method=all&timeout=0", auth=AUTH)
@@ -188,4 +194,9 @@ if __name__ == "__main__":
print("Unsupported dispatch method {}".format(entry["method"]), sys=sys.stderr)
continue
sys.exit(0)
# wait a moment #
if args.loop:
time.sleep(5)
# handle non-loop runs #
first_run = False