mirror of
https://github.com/yawaflua/Flask-Discord.git
synced 2025-12-10 12:29:30 +02:00
106 lines
2.3 KiB
Python
106 lines
2.3 KiB
Python
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 = '<a href="{}">{}</a>'
|
|
|
|
|
|
@app.route("/")
|
|
def index():
|
|
if not discord.authorized:
|
|
return HYPERLINK.format(url_for(".login"), "Login")
|
|
return f"""
|
|
{HYPERLINK.format(url_for(".me"), "@ME")}<br />
|
|
{HYPERLINK.format(url_for(".logout"), "Logout")}<br />
|
|
{HYPERLINK.format(url_for(".user_guilds"), "My Servers")}<br />
|
|
{HYPERLINK.format(url_for(".add_to_guild", guild_id=475549041741135881), "Add me 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"""
|
|
<html>
|
|
<head>
|
|
<title>{user.name}</title>
|
|
</head>
|
|
<body><img src='{user.avatar_url or user.default_avatar_url}' />
|
|
<p>Is avatar animated: {str(user.is_avatar_animated)}</p>
|
|
<a href={url_for("my_connections")}>Connections</a>
|
|
<br />
|
|
</body>
|
|
</html>
|
|
|
|
"""
|
|
|
|
|
|
@app.route("/me/guilds/")
|
|
def user_guilds():
|
|
guilds = discord.fetch_guilds()
|
|
return "<br />".join([g.name for g in guilds])
|
|
|
|
|
|
@app.route("/add_to/<int:guild_id>/")
|
|
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"""
|
|
<html>
|
|
<head>
|
|
<title>{user.name}</title>
|
|
</head>
|
|
<body>
|
|
{str([f"{connection.name} - {connection.type}" for connection in connections])}
|
|
</body>
|
|
</html>
|
|
|
|
"""
|
|
|
|
|
|
@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)
|