sheppy/atlantis-event-dispatcher
1
0
Fork 0
mirror of https://github.com/FAUSheppy/atlantis-event-dispatcher synced 2026-01-21 22:57:39 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: add topic query from api

Browse Source
This commit is contained in:
Johnathen Sheppard
2024-02-16 19:15:04 +01:00
parent 46260edfaa
commit 7f468ba860
1 changed files with 27 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
client/dispatch-query.py
View File

@@ -30,9 +30,29 @@ def email_send(dispatch_uuid, email_address, message, smtp_target, smtp_user, sm
smtphelper.smtp_send(smtp_target, smtp_user, smtp_pass, email_address, subject, message) smtphelper.smtp_send(smtp_target, smtp_user, smtp_pass, email_address, subject, message)
report_failed_dispatch(uuid, "Email dispatch not yet implemented") report_failed_dispatch(uuid, "Email dispatch not yet implemented")
def ntfy_api_get_topic(ntfy_api_server, ntfy_api_token, username):
'''Get the topic of the user'''
params = {
"user" : username,
"token" : ntfy_api_token,
}
r = requests.get(ntfy_api_server + "/topic", params=params)
if r.status_code != 200:
print(r.text)
return None
else:
print(r.text)
return r.json().get("topic")
def ntfy_send(dispatch_uuid, user_topic, message, ntfy_push_target, ntfy_user, ntfy_pass): def ntfy_send(dispatch_uuid, user_topic, message, ntfy_push_target, ntfy_user, ntfy_pass):
'''Send message via NTFY topic''' '''Send message via NTFY topic'''
if not user_topic:
report_failed_dispatch(dispatch_uuid, "No user topic")
return
try: try:
# build message # # build message #
@@ -90,6 +110,9 @@ if __name__ == "__main__":
parser.add_argument('--dispatch-user') parser.add_argument('--dispatch-user')
parser.add_argument('--dispatch-password') parser.add_argument('--dispatch-password')
parser.add_argument('--ntfy-api-server')
parser.add_argument('--ntfy-api-token')
parser.add_argument('--ntfy-push-target') parser.add_argument('--ntfy-push-target')
parser.add_argument('--ntfy-user') parser.add_argument('--ntfy-user')
parser.add_argument('--ntfy-pass') parser.add_argument('--ntfy-pass')
@@ -108,6 +131,9 @@ if __name__ == "__main__":
dispatch_user = args.dispatch_user or os.environ.get("DISPATCH_USER") dispatch_user = args.dispatch_user or os.environ.get("DISPATCH_USER")
dispatch_password = args.dispatch_password or os.environ.get("DISPATCH_PASSWORD") dispatch_password = args.dispatch_password or os.environ.get("DISPATCH_PASSWORD")
ntfy_api_server = args.ntfy_api_server or os.environ.get("NTFY_API_SERVER")
ntfy_api_token = args.ntfy_api_token or os.environ.get("NTFY_API_TOKEN")
ntfy_push_target = args.ntfy_push_target or os.environ.get("NTFY_PUSH_TARGET") ntfy_push_target = args.ntfy_push_target or os.environ.get("NTFY_PUSH_TARGET")
ntfy_user = args.ntfy_user or os.environ.get("NTFY_USER") ntfy_user = args.ntfy_user or os.environ.get("NTFY_USER")
ntfy_pass = args.ntfy_pass or os.environ.get("NTFY_PASS") ntfy_pass = args.ntfy_pass or os.environ.get("NTFY_PASS")
@@ -141,7 +167,6 @@ if __name__ == "__main__":
message = entry["message"] message = entry["message"]
# method dependent fields # # method dependent fields #
user_topic = entry.get("topic")
phone = entry.get("phone") phone = entry.get("phone")
email_address = entry.get("email") email_address = entry.get("email")
@@ -149,6 +174,7 @@ if __name__ == "__main__":
if method == "signal": if method == "signal":
pass pass
elif method == "ntfy": elif method == "ntfy":
user_topic = ntfy_api_get_topic(ntfy_api_server, ntfy_api_token, user)
ntfy_send(dispatch_uuid, user_topic, message, ntfy_push_target, ntfy_user, ntfy_pass) ntfy_send(dispatch_uuid, user_topic, message, ntfy_push_target, ntfy_user, ntfy_pass)
elif method == "email": elif method == "email":
email_send(dispatch_uuid, email_address, message, smtp_target, smtp_user, smtp_pass) email_send(dispatch_uuid, email_address, message, smtp_target, smtp_user, smtp_pass)