import os
from flask import Flask, redirect, url_for
from flask_discord import DiscordOAuth2Session, requires_authorization
app = Flask(__name__)
app.secret_key = b"%\xe0'\x01\xdeH\x8e\x85m|\xb3\xffCN\xc9g"
app.config["DISCORD_CLIENT_ID"] = 490732332240863233
app.config["DISCORD_CLIENT_SECRET"] = os.getenv("DISCORD_CLIENT_SECRET")
app.config["DISCORD_BOT_TOKEN"] = os.getenv("DISCORD_BOT_TOKEN")
app.config["DISCORD_REDIRECT_URI"] = "http://127.0.0.1:5000/callback"
discord = DiscordOAuth2Session(app)
HYPERLINK = '{}'
@app.route("/")
def index():
if not discord.authorized:
return HYPERLINK.format(url_for(".login"), "Login")
return f"""
{HYPERLINK.format(url_for(".me"), "@ME")}
{HYPERLINK.format(url_for(".logout"), "Logout")}
{HYPERLINK.format(url_for(".user_guilds"), "My Servers")}
{HYPERLINK.format(url_for(".add_to_guild", guild_id=475549041741135881), "Add bot to 475549041741135881.")}
"""
@app.route("/login/")
def login():
return discord.create_session()
@app.route("/callback/")
def callback():
discord.callback()
return redirect(url_for(".index"))
@app.route("/me/")
def me():
user = discord.fetch_user()
return f"""
{user.name}
Connections
"""
@app.route("/me/guilds/")
def user_guilds():
guilds = discord.fetch_guilds()
return "
".join([g.name for g in guilds])
@app.route("/add_to//")
def add_to_guild(guild_id):
user = discord.fetch_user()
return user.add_to_guild(guild_id)
@app.route("/me/connections/")
def my_connections():
user = discord.fetch_user()
connections = discord.fetch_connections()
return f"""
{user.name}
{str([f"{connection.name} - {connection.type}" for connection in connections])}
"""
@app.route("/logout/")
def logout():
discord.revoke()
return redirect(url_for(".index"))
@app.route("/secret/")
@requires_authorization
def secret():
return os.urandom(16)
if __name__ == "__main__":
app.run(debug=True)