hyhhyhybboooo / app.py
coollsd's picture
Update app.py
4338078 verified
limport discord
from discord import app_commands
from fastapi import FastAPI
import uvicorn
import asyncio
import os
import aiohttp
app = FastAPI()
TOKEN = os.environ['TOKEN']
SERVER_ID = 1128366741995659296
CHANNEL_ID = 1296102118326931596
API_URL = "https://fruitblox.deno.dev/"
intents = discord.Intents.default()
intents.message_content = True
bot = discord.Client(intents=intents)
tree = app_commands.CommandTree(bot)
api_data = {}
previous_data = None
async def format_and_send_message(data, channel):
message = []
if "normal" in data and data["normal"]:
message.append("Normal")
for fruit in data["normal"]:
message.append(fruit)
message.append("")
if "mirage" in data and data["mirage"]:
message.append("Mirage")
for fruit in data["mirage"]:
message.append(fruit)
formatted_message = "\n".join(message)
await channel.send(formatted_message)
async def check_api_changes():
global previous_data
channel = bot.get_channel(CHANNEL_ID)
if not channel:
return
async with aiohttp.ClientSession() as session:
try:
async with session.get(API_URL) as response:
if response.status == 200:
current_data = await response.json()
if previous_data is None:
# First run, post initial state
await format_and_send_message(current_data, channel)
elif current_data != previous_data:
# Data has changed, post update
await format_and_send_message(current_data, channel)
previous_data = current_data
except Exception as e:
print(f"Error fetching API: {e}")
async def schedule_api_checks():
while True:
await check_api_changes()
await asyncio.sleep(30) # Check every 30 seconds
@app.get("/")
async def read_root():
return api_data
@bot.event
async def on_ready():
await tree.sync()
print(f"{bot.user} is now online!")
asyncio.create_task(schedule_api_checks())
@bot.event
async def on_message(message):
if message.author == bot.user or message.guild.id != SERVER_ID or message.channel.id != CHANNEL_ID:
return
content = message.content.strip().split('\n')
current_category = None
updated = False
api_data.clear()
for line in content:
line = line.strip()
if line.lower() in ["normal", "mirage"]:
current_category = line.lower()
if current_category not in api_data:
api_data[current_category] = []
updated = True
elif current_category and line:
api_data[current_category].append(line)
updated = True
if updated:
await message.channel.send("Posted in API")
@tree.command(name="clear", description="clear api")
async def clear_api(interaction: discord.Interaction):
global api_data
api_data.clear()
await interaction.response.send_message("cleared")
@tree.command(name="forceupdate", description="force stock/check ")
async def force_update(interaction: discord.Interaction):
await check_api_changes()
await interaction.response.send_message("completed")
async def run_bot():
await bot.start(TOKEN)
@app.on_event("startup")
async def startup_event():
asyncio.create_task(run_bot())
if __name__ == "__main__":
uvicorn.run("app:app", host="0.0.0.0", port=7860)