Create RFT.py
Discord bot that I plan to use for various raid shadows related tasks
This commit is contained in:
commit
b1ae43c052
1 changed files with 43 additions and 0 deletions
43
RFT.py
Normal file
43
RFT.py
Normal file
|
|
@ -0,0 +1,43 @@
|
|||
import discord
|
||||
from discord.ext import commands
|
||||
import pandas as pd
|
||||
|
||||
intents = discord.Intents.default()
|
||||
bot = commands.Bot(command_prefix="!", intents=intents)
|
||||
|
||||
# Replace this with your authorized channel ID
|
||||
AUTHORIZED_CHANNEL_ID = # TODO: input channel ID here
|
||||
|
||||
@bot.event
|
||||
async def on_ready():
|
||||
print(f"Bot connected as {bot.user}")
|
||||
|
||||
def read_google_sheet(url):
|
||||
file_id = url.split("/d/")[1].split("/")[0]
|
||||
csv_url = f"https://docs.google.com/spreadsheets/d/{file_id}/export?format=csv"
|
||||
df = pd.read_csv(csv_url)
|
||||
return df
|
||||
|
||||
@bot.command()
|
||||
async def siege(ctx):
|
||||
# If not in the authorized channel, silently ignore the command
|
||||
if ctx.channel.id != AUTHORIZED_CHANNEL_ID:
|
||||
return
|
||||
|
||||
try:
|
||||
df = read_google_sheet("") # TODO: set public sheet url
|
||||
|
||||
user_pseudo = ctx.author.name.lower()
|
||||
first_column = df.columns[0]
|
||||
|
||||
matching_row = df[df[first_column].str.lower() == user_pseudo]
|
||||
|
||||
if matching_row.empty:
|
||||
await ctx.send("No data found for your username.")
|
||||
else:
|
||||
info = matching_row.iloc[0].to_dict()
|
||||
response = "\n".join([f"**{col}**: {val}" for col, val in info.items() if col != first_column])
|
||||
await ctx.send(f"Data for **{ctx.author.name}**:\n{response}")
|
||||
|
||||
except Exception as e:
|
||||
await ctx.send(f"Error: {e}")
|
||||
Loading…
Reference in a new issue