diff options
| author |  Den4200 <[email protected]> | 2021-03-27 13:17:12 -0400 |
|---|---|---|
| committer | Den4200 <[email protected]> | 2021-03-27 13:26:02 -0400 |
| commit | f9f9ced5320a593bb59836086d0d5983b4df58df (patch) | |
| tree | 2705653f906b9bc534704dd0f05a1b0dc0c7e7f2 | |
| parent | Merge pull request #1466 from vcokltfre/discord-tags (diff) | |
Restrict redirects for paste uploads with an extension that is not `.py`.
| -rw-r--r-- | bot/utils/services.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/bot/utils/services.py b/bot/utils/services.py index 5949c9e48..68cbd896e 100644 --- a/bot/utils/services.py +++ b/bot/utils/services.py @@ -47,7 +47,13 @@ async def send_to_paste_service(contents: str, *, extension: str = "") -> Option continue elif "key" in response_json: log.info(f"Successfully uploaded contents to paste service behind key {response_json['key']}.") - return URLs.paste_service.format(key=response_json['key']) + extension + + paste_link = URLs.paste_service.format(key=response_json['key']) + extension + + if extension == '.py': + return paste_link + return paste_link + "?noredirect" + log.warning( f"Got unexpected JSON response from paste service: {response_json}\n" f"trying again ({attempt}/{FAILED_REQUEST_ATTEMPTS})." |