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