edg-l/ddnet
1
0
Fork 0
mirror of https://github.com/ddnet/ddnet.git synced 2024-09-20 09:34:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update modhelp.py

Browse source 
Make a couple of things customizable using environment variables.
This commit is contained in:
heinrich5991 2018-07-10 18:59:40 +02:00
parent 94a8fe79bc
commit fe1c54c11e
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/modhelp/server.py
View file

@ -1,16 +1,19 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
from flask import Flask, make_response, request from flask import Flask, make_response, request
import http import http
import os
import re import re
import requests import requests
SERVER=None SERVER=os.getenv('MODHELP_SERVER')
# Generate one by right-clicking on the server icon in the sidebar, clicking on # Generate one by right-clicking on the server icon in the sidebar, clicking on
# "Server Settings" → "Webhooks" → "Create Webhook". You can then select the # "Server Settings" → "Webhooks" → "Create Webhook". You can then select the
# channel in which the messages should appear. Copy the "Webhook URL" to the # channel in which the messages should appear. Copy the "Webhook URL" to the
# following config variable: # following config variable:
# DISCORD_WEBHOOK="https://discordapp.com/api/webhooks/.../..." # DISCORD_WEBHOOK="https://discordapp.com/api/webhooks/.../..."
DISCORD_WEBHOOK=None DISCORD_WEBHOOK=os.getenv('MODHELP_DISCORD_WEBHOOK')
DISCORD_MESSAGE=os.getenv('MODHELP_DISCORD_FORMAT', "<{player_id}:**{player_name}**> {message}")
DISCORD_MESSAGE_PREFIX=os.getenv('MODHELP_DISCORD_PREFIX')
app = Flask(__name__) app = Flask(__name__)
@ -20,6 +23,9 @@ def sanitize(s):
def no_content(): def no_content():
return make_response("", http.HTTPStatus.NO_CONTENT) return make_response("", http.HTTPStatus.NO_CONTENT)
def sanitize_string_values(dictionary):
return {k: v if not isinstance(v, str) else sanitize(v) for k, v in dictionary.items()}
@app.route("/modhelp", methods=['POST']) @app.route("/modhelp", methods=['POST'])
def modhelp(): def modhelp():
json = request.get_json() json = request.get_json()
@ -32,11 +38,13 @@ def modhelp():
user = "{port}".format(**json) user = "{port}".format(**json)
else: else:
user = "{server}:{port}".format(**json) user = "{server}:{port}".format(**json)
message = "<{player_id}:{player_name}> {message}".format(**json)
discord_prefix = DISCORD_MESSAGE_PREFIX + " " if DISCORD_MESSAGE_PREFIX is not None else ""
discord_message = discord_prefix + DISCORD_MESSAGE.format(**sanitize_string_values(json))
if DISCORD_WEBHOOK: if DISCORD_WEBHOOK:
try: try:
requests.post(DISCORD_WEBHOOK, data={"username": user, "content": sanitize(message)}).raise_for_status() requests.post(DISCORD_WEBHOOK, data={"username": user, "content": discord_message}).raise_for_status()
except requests.HTTPError as e: except requests.HTTPError as e:
print(repr(e)) print(repr(e))
raise raise