sheppy/atlantis-event-dispatcher
1
0
Fork 0
mirror of https://github.com/FAUSheppy/atlantis-event-dispatcher synced 2025-12-06 06:21:36 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: support authentication

Browse Source
This commit is contained in:
Johnathen Sheppard
2023-07-21 16:51:10 +02:00
parent 6292e745a8
commit 1015991e71
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
signal-query-dispatch.py
View File

@@ -18,7 +18,8 @@ def signal_send(user, message):
def confirm_dispatch(target, uid): def confirm_dispatch(target, uid):
'''Confirm to server that message has been dispatched and can be removed''' '''Confirm to server that message has been dispatched and can be removed'''
response = requests.post(target + "/confirm-dispatch", json=[{ "uid" : uid }]) response = requests.post(target + "/confirm-dispatch", json=[{ "uid" : uid }],
auth=(args.user, args.password))
if response.status_code not in [200, 204]: if response.status_code not in [200, 204]:
print("Failed to confirm disptach with server for {} ({})".format(uid, response.text), file=sys.stderr) print("Failed to confirm disptach with server for {} ({})".format(uid, response.text), file=sys.stderr)
@@ -28,17 +29,22 @@ if __name__ == "__main__":
parser = argparse.ArgumentParser(description='Query Atlantis Dispatch for Signal', parser = argparse.ArgumentParser(description='Query Atlantis Dispatch for Signal',
formatter_class=argparse.ArgumentDefaultsHelpFormatter) formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--target', required=True) parser.add_argument('--target', required=True)
parser.add_argument('--method', default="signal") parser.add_argument('--method', default="signal")
parser.add_argument('--no-confirm', action="store_true") parser.add_argument('--no-confirm', action="store_true")
parser.add_argument('--signal-cli-bin') parser.add_argument('--signal-cli-bin')
parser.add_argument('--user')
parser.add_argument('--password')
args = parser.parse_args() args = parser.parse_args()
if args.signal_cli_bin: if args.signal_cli_bin:
signal_cli_bin = args.signal_cli_bin signal_cli_bin = args.signal_cli_bin
response = requests.get(args.target + "/get-dispatch?method={}".format(args.method)) response = requests.get(args.target + "/get-dispatch?method={}".format(args.method),
auth=(args.user, args.password))
# check status # # check status #
if response.status_code == HTTP_NOT_FOUND: if response.status_code == HTTP_NOT_FOUND: