dashboard.py

import config
from quart import Quart, render_template, request, session, redirect, url_for
from quart_discord import DiscordOAuth2Session
from discord.ext import ipc
from utils import database
from globals import modules

app = Quart(__name__, static_folder="web/public", template_folder="web/templates")
ipc_client = ipc.Client(secret_key=config.ipc_key)

app.config["SECRET_KEY"] = config.secret_key
app.config["DISCORD_CLIENT_ID"] = config.discord_client_id
app.config["DISCORD_CLIENT_SECRET"] =  config.discord_client_secret
app.config["DISCORD_REDIRECT_URI"] = config.discord_redirect_uri

discord = DiscordOAuth2Session(app)

##################  DASHBOARD ##################

@app.route("/")
async def home():
    authorized = await discord.authorized
    if authorized == True:
        return redirect("/dashboard")
    else:
        return await render_template("index.html")

@app.route("/login/")
async def login():
    return await discord.create_session()

@app.route("/callback/")
async def callback():
    try:
        await discord.callback()
    except:
        return redirect(url_for("login"))
    
    return redirect("/dashboard")

@app.route("/dashboard")
async def dashboard():
    authorized = await discord.authorized
    if authorized != True:
        return redirect("/")
    
    guild_ids = await ipc_client.request("get_mutual_guilds")
    user_guilds = await discord.fetch_guilds()
    user = await discord.fetch_user()
    admin_guilds = []
    mutual_guilds = []

    for guild in user_guilds:
        if guild.permissions.administrator == True and guild.id not in guild_ids:
            admin_guilds.append(guild)
        elif guild.permissions.administrator == True and guild.id in guild_ids:
            mutual_guilds.append(guild)

    return await render_template("dashboard.html", admin_guilds=admin_guilds, mutual_guilds=mutual_guilds, user=user, config=config)

@app.route("/dashboard/<guild_id>")
async def guild_dashboard(guild_id):
    authorized = await discord.authorized
    if authorized != True:
        return redirect("/")

    user = await discord.fetch_user()
    guilds = await discord.fetch_guilds()

    for guild in guilds:
        if guild.id == int(guild_id):
            return await render_template("guild-dashboard.html", guild=guild, user=user, modules=modules)

@app.route("/dashboard/<guild_id>/<module>")
async def guild_dashboard_leveling(guild_id, module):
    authorized = await discord.authorized
    if authorized != True:
        return redirect("/")
    
    user = await discord.fetch_user()
    guilds = await discord.fetch_guilds()

    for guild in guilds:
        if guild.id == int(guild_id):
            return await render_template(f"{module}.html", guild=guild, user=user, config=database.get_config((guild.id)))
    
    return redirect("/")

@app.route("/logout")
async def logout():
    discord.revoke()
    return redirect("/")


##########################  API ##########################

@app.route("/api/leaderboard/<guild_id>")
async def api_get_leaderboard(guild_id):
    data = {}
    i = 0
    users = database.get_leaderboard(guild_id, request.args["offset"], request.args["limit"])
    for user in users:
        display_name = await ipc_client.request("get_display_name_by_id", user_id=user["user_id"])
        user["display_name"] = display_name
        data[i] = user
        i += 1
    return data

@app.route("/api/settings/get/<guild_id>")
async def api_get_setting(guild_id):
    authorized = await discord.authorized
    if authorized != True:
        return "unauthorized"

    config = database.get_config(int(guild_id))
    return config

@app.route("/api/settings/change/<guild_id>", methods=["POST"])
async def api_change_setting(guild_id):
    authorized = await discord.authorized
    if authorized != True:
        return "unauthorized"
    
    args = await request.form
    callback = database.change_config(guild_id, args["item"], args["value"])
    if callback == True:
        return "done"
    else:
        return "error"

app.run("127.0.0.1", 5000, debug=True)