Как подружить Flask и Discord.py?

Question

[Tokishu]

Я хочу сделать бота который будет вытягивать нужные данные о пользователе по запросу.
Но я не могу понять как подружить Flask и Discord чтобы они работали в гармонии и всё из-за того что Flask особо не поддерживает ассинхрон.

Вот что я написал:

import asyncio
from concurrent.futures import ThreadPoolExecutor
from flask import Flask, jsonify
import discord
from discord.ext import commands
from settings import settings

app = Flask(__name__)
intents = discord.Intents.default()
intents.message_content = True
intents.members = True
bot = commands.Bot(command_prefix='>', intents=intents)

def start_flask():
    app.run(host='localhost', port=8000, debug=True)

async def start_bot():
    await bot.start(settings['discordToken'])

def run_flask():
    loop = asyncio.new_event_loop()
    asyncio.set_event_loop(loop)
    loop.run_until_complete(start_flask())

def run_bot():
    loop = asyncio.new_event_loop()
    asyncio.set_event_loop(loop)
    loop.run_until_complete(start_bot())

@app.route('/api/getUser/<user_id>')
def get_user_route(user_id):
    try:
        loop = asyncio.new_event_loop()
        asyncio.set_event_loop(loop)
        return loop.run_until_complete(get_user(user_id))
    except discord.NotFound:
        return jsonify({"error": "User not found"}), 404

async def get_user(user_id):
    user = await bot.fetch_user(user_id)
    guild_id = settings['hakuServerId']
    guild = await bot.fetch_guild(guild_id)
    if guild:
        member = await guild.fetch_member(user.id)
        if member:
            avatar = user.avatar
            nickname = member.display_name
            banner = user.banner.url if user.banner else user.accent_color
            roles = [role.name for role in member.roles]
            banner = banner if banner is not None else user.accent_color

            return jsonify({
                "avatar": str(avatar.url) if avatar else None,
                "nickname": str(nickname),
                "roles": roles,
                "discord_id": user_id,
                "banner": str(banner) if banner else None
            })

@bot.event
async def on_ready():
    print(f'[{bot.user}] Успешная авторизация')

if __name__ == "__main__":
    with ThreadPoolExecutor(max_workers=2) as executor:
        executor.submit(run_flask)
        executor.submit(run_bot)

Как итог код запускается:

WARNING: This is a development server. Do not use it in a production deployment. Use a production WSGI server instead.
 * Running on http://localhost:8000
Press CTRL+C to quit
[Бот] Успешная авторизация

Но при запросе я не могу обратится к API


P.S: К сожалению я обычно не работаю с питоном а с C подобными языками, по этому могу тупить

Comments

[xibolba]

Попробуйте запускать корутины через asyncio.run().
Например переписать:

def get_user_route(user_id):
    try:
        return asyncio.run(get_user(user_id))
    except discord.NotFound:
        return jsonify({"error": "User not found"}), 404

Зачем тут run_flask?
Если нужен ассинхрон используйте aiohttp.
Да и разделите процессы, что б понять в чем проблема.

[Даниил Рыжов]

Как написали выше проще использовать aiohttp, ну или FastAPI, вот вам и "дружелюбная" асинхронность.

Можете использовать не объект commands.Bot, а client = discord.Client, который в контексте запроса можно будет использовать используя синтаксисasync with client.
Ну или также запускайте в отдельном потоке, еще можете рассмотреть discord.ext.tasks

[Загир Меджидов]

Лучше использовать Quart + Disnake.

Quart - это асинхронная библиотека, копия Flask.
Disnake - Это стабильная и разрабатывающая библиотека, замена discord.

YouTube видео обзор
Build a Discord Bot with Python and Flask

[Vindicar]

Загир Меджидов, ну насколько я знаю, автор discord.py пришёл в себя и снова активно пилит проект. Так что лучше уж оригинал использовать.

[Tokishu]

Даниил Рыжов, Тут такая интересная ситуация, была изначальная задумка написать на Flask, Я начал писать и нашел эту проблему потом перешел на FastAPI потом меня попросили добавить ограничение на количество запросов я долго пытался и так не понял как это сделать. Подумал сделать это на Flask и опять перешел на него xD Если вы знаете как сделать ограничение запросов на не более чем 3 в сек буду благодарен

Это пишется не на заказ

import asyncio
from settings import settings
import discord
from discord.ext import commands
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.middleware.gzip import GZipMiddleware
from fastapi import Path

intents = discord.Intents.default()
intents.message_content = True
intents.members = True
bot = commands.Bot(command_prefix='>', intents=intents)

app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

app.add_middleware(GZipMiddleware, minimum_size=1000)

from fastapi import Depends

@app.get("/api/getUser/{user_id}")
async def get_user_route(user_id: str = Path(..., title="Discord User ID")):
    user_data = await get_user(user_id)
    print(user_data)
    if not user_data:
        return {"error": "User not found"}
    return user_data

async def get_user(user_id: str, rate_limit: Depends = Depends()):
    try:
        user = await bot.fetch_user(user_id)
        guild_id = settings['hakuServerId']
        guild = await bot.fetch_guild(guild_id)
        if guild:
            member = await guild.fetch_member(user.id)
            if member:
                avatar = user.avatar
                nickname = member.display_name
                banner = user.banner.url if user.banner else user.accent_color
                roles = [role.name for role in member.roles]
                banner = banner if banner is not None else user.accent_color

                return {
                    "avatar": str(avatar.url) if avatar else None,
                    "nickname": str(nickname),
                    "roles": roles,
                    "discord_id": user_id,
                    "banner": str(banner) if banner else None
                }



    except discord.NotFound:
        return None

@bot.event
async def on_ready():
    print(f'[{bot.user}] Успешная авторизация')

async def start_bot():
    await bot.start(settings['discordToken'])

async def start_api():
    import uvicorn
    config = uvicorn.Config(app, host="127.0.0.1", port=8000, log_level="info")
    server = uvicorn.Server(config)
    await server.serve()

if __name__ == "__main__":
    loop = asyncio.get_event_loop()
    loop.create_task(start_api())
    loop.run_until_complete(start_bot())

[Загир Меджидов]

Tokishu, к сожалению я очень слаб в FastAPI и не смогу ответить на ваш вопрос.

Answer 1

[Persifox]

Запускай не фласк, а Quart - тот же фласк, только async.

Бота через Thead, а Quart в основном потоке.

Примерно так реализовано у меня:

# main.py
bot = NoirBot(debug=True)

thread_bot = threading.Thread(target=bot.run, name="bot")

try:
    thread_bot.start()
    asyncio.run(bot.serve_app())

except Exception as exp:
    lprint(f"Connection failed: {exp}", Color.red)
finally:
    lprint(f"Stopped", Color.yellow)
    exit()

# Bot.py
# Run & Stop funcs
    def run(self) -> None:
        try:
            self.loop.run_until_complete(self.start())
        except KeyboardInterrupt:
            self.node.disconnect()
            self.loop.run_until_complete(self.close())
            # cancel all tasks lingering
        finally:
            self.loop.close()

    def start(self):
        if self._debug:
            return super().start(self._config.get("altsecrets", "token"))
        return super().start(self._config.get("secrets", "token"))

# Bot.py

# App
    async def serve_app(self):
        lprint("Checking routers", Color.blue, worker="APP")

        routersLoad(self)

        lprint("Loading Quart", Color.blue, worker="APP")

        config = hypercorn.Config()
        config.bind = ["0.0.0.0:5001"]
        config.use_reloader = True

        lprint("Done", Color.green, worker="APP")

        await serve(self._app, config)