forked from rDrama/rDrama
1
0
Fork 0

repost new posts in WPD in discord

master
Aevann1 2022-07-10 16:25:42 +02:00
parent bb59cd0476
commit 2a1efff78b
2 changed files with 14 additions and 7 deletions

View File

@ -2,6 +2,8 @@ import requests
import threading import threading
from .const import * from .const import *
headers = {"Authorization": f"Bot {DISCORD_BOT_TOKEN}"}
def discord_wrap(f): def discord_wrap(f):
def wrapper(*args, **kwargs): def wrapper(*args, **kwargs):
@ -23,31 +25,30 @@ def discord_wrap(f):
def add_role(user, role_name): def add_role(user, role_name):
role_id = ROLES[role_name] role_id = ROLES[role_name]
url = f"https://discordapp.com/api/guilds/{DISCORD_SERVER_ID}/members/{user.discord_id}/roles/{role_id}" 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) requests.put(url, headers=headers, timeout=5)
@discord_wrap @discord_wrap
def remove_role(user, role_name): def remove_role(user, role_name):
role_id = ROLES[role_name] role_id = ROLES[role_name]
url = f"https://discordapp.com/api/guilds/{DISCORD_SERVER_ID}/members/{user.discord_id}/roles/{role_id}" 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) requests.delete(url, headers=headers, timeout=5)
@discord_wrap @discord_wrap
def remove_user(user): def remove_user(user):
url=f"https://discordapp.com/api/guilds/{DISCORD_SERVER_ID}/members/{user.discord_id}" 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) requests.delete(url, headers=headers, timeout=5)
@discord_wrap @discord_wrap
def set_nick(user, nick): def set_nick(user, nick):
url=f"https://discordapp.com/api/guilds/{DISCORD_SERVER_ID}/members/{user.discord_id}" url=f"https://discordapp.com/api/guilds/{DISCORD_SERVER_ID}/members/{user.discord_id}"
headers = {"Authorization": f"Bot {DISCORD_BOT_TOKEN}"}
data={"nick": nick} data={"nick": nick}
requests.patch(url, headers=headers, json=data, timeout=5) requests.patch(url, headers=headers, json=data, timeout=5)
def send_discord_message(message): def send_discord_message(message):
headers = {"Authorization": f"Bot {DISCORD_BOT_TOKEN}"}
data={"content": message} 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/924485611715452940/messages", headers=headers, data=data, timeout=5)
requests.post("https://discordapp.com/api/channels/924486091795484732/messages", headers=headers, data=data, timeout=5) 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)

View File

@ -4,7 +4,7 @@ import requests
from files.helpers.wrappers import * from files.helpers.wrappers import *
from files.helpers.sanitize import * from files.helpers.sanitize import *
from files.helpers.alerts 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.const import *
from files.helpers.regex import * from files.helpers.regex import *
from files.helpers.slots import * from files.helpers.slots import *
@ -72,6 +72,9 @@ def publish(pid, v):
send_discord_message(post.permalink) send_discord_message(post.permalink)
cache.delete_memoized(changeloglist) cache.delete_memoized(changeloglist)
if SITE == 'watchpeopledie.co':
send_wpd_message(post.permalink)
execute_snappy(post, v) execute_snappy(post, v)
if is_site_url(request.referrer): if is_site_url(request.referrer):
@ -1079,6 +1082,9 @@ def submit_post(v, sub=None):
send_discord_message(post.permalink) send_discord_message(post.permalink)
cache.delete_memoized(changeloglist) cache.delete_memoized(changeloglist)
if SITE == 'watchpeopledie.co':
send_wpd_message(post.permalink)
if request.headers.get("Authorization"): return post.json if request.headers.get("Authorization"): return post.json
else: else:
post.voted = 1 post.voted = 1