From 2a1efff78b2da1f0afeec27ece84095e58177f39 Mon Sep 17 00:00:00 2001 From: Aevann1 Date: Sun, 10 Jul 2022 16:25:42 +0200 Subject: [PATCH] repost new posts in WPD in discord --- files/helpers/discord.py | 13 +++++++------ files/routes/posts.py | 8 +++++++- 2 files changed, 14 insertions(+), 7 deletions(-) diff --git a/files/helpers/discord.py b/files/helpers/discord.py index 3088444be..3ccbae795 100644 --- a/files/helpers/discord.py +++ b/files/helpers/discord.py @@ -2,6 +2,8 @@ import requests import threading from .const import * +headers = {"Authorization": f"Bot {DISCORD_BOT_TOKEN}"} + def discord_wrap(f): def wrapper(*args, **kwargs): @@ -23,31 +25,30 @@ def discord_wrap(f): def add_role(user, role_name): role_id = ROLES[role_name] url = f"https://discordapp.com/api/guilds/{DISCORD_SERVER_ID}/members/{user.discord_id}/roles/{role_id}" - headers = {"Authorization": f"Bot {DISCORD_BOT_TOKEN}"} requests.put(url, headers=headers, timeout=5) @discord_wrap def remove_role(user, role_name): role_id = ROLES[role_name] url = f"https://discordapp.com/api/guilds/{DISCORD_SERVER_ID}/members/{user.discord_id}/roles/{role_id}" - headers = {"Authorization": f"Bot {DISCORD_BOT_TOKEN}"} requests.delete(url, headers=headers, timeout=5) @discord_wrap def remove_user(user): url=f"https://discordapp.com/api/guilds/{DISCORD_SERVER_ID}/members/{user.discord_id}" - headers = {"Authorization": f"Bot {DISCORD_BOT_TOKEN}"} requests.delete(url, headers=headers, timeout=5) @discord_wrap def set_nick(user, nick): url=f"https://discordapp.com/api/guilds/{DISCORD_SERVER_ID}/members/{user.discord_id}" - headers = {"Authorization": f"Bot {DISCORD_BOT_TOKEN}"} data={"nick": nick} requests.patch(url, headers=headers, json=data, timeout=5) def send_discord_message(message): - headers = {"Authorization": f"Bot {DISCORD_BOT_TOKEN}"} data={"content": message} requests.post("https://discordapp.com/api/channels/924485611715452940/messages", headers=headers, data=data, timeout=5) - requests.post("https://discordapp.com/api/channels/924486091795484732/messages", headers=headers, data=data, timeout=5) \ No newline at end of file + requests.post("https://discordapp.com/api/channels/924486091795484732/messages", headers=headers, data=data, timeout=5) + +def send_wpd_message(message): + data={"content": message} + requests.post("https://discordapp.com/api/channels/995684665283313674/messages", headers=headers, data=data, timeout=5) \ No newline at end of file diff --git a/files/routes/posts.py b/files/routes/posts.py index 1ea268553..fda6ddfe5 100644 --- a/files/routes/posts.py +++ b/files/routes/posts.py @@ -4,7 +4,7 @@ import requests from files.helpers.wrappers import * from files.helpers.sanitize import * from files.helpers.alerts import * -from files.helpers.discord import send_discord_message +from files.helpers.discord import * from files.helpers.const import * from files.helpers.regex import * from files.helpers.slots import * @@ -72,6 +72,9 @@ def publish(pid, v): send_discord_message(post.permalink) cache.delete_memoized(changeloglist) + if SITE == 'watchpeopledie.co': + send_wpd_message(post.permalink) + execute_snappy(post, v) if is_site_url(request.referrer): @@ -1079,6 +1082,9 @@ def submit_post(v, sub=None): send_discord_message(post.permalink) cache.delete_memoized(changeloglist) + if SITE == 'watchpeopledie.co': + send_wpd_message(post.permalink) + if request.headers.get("Authorization"): return post.json else: post.voted = 1