Unstable Kitsune
e0958f6f2a
Sets up RPG system as a Redbot cog with core mechanics Implements character creation, class selection, and stat management Adds action commands for attacking and healing Provides inventory and shop functionality for item management Introduces interactive UI menus for user engagement
94 lines
3.0 KiB
Python
94 lines
3.0 KiB
Python
from redbot.core import Config, commands
|
|
from collections.abc import Iterable
|
|
import validators as vals
|
|
import logging
|
|
import discord
|
|
|
|
|
|
log = logging.getLogger("red.rpg")
|
|
|
|
class Check:
|
|
|
|
def __init__(self, ctx_or_interaction, custom: Iterable = None, length: int = None):
|
|
self.ctx_or_interaction = ctx_or_interaction # Store the context or interaction object
|
|
self.custom = custom
|
|
self.length = length
|
|
|
|
def _get_author(self):
|
|
"""
|
|
Helper function to get the author from either ctx or interaction.
|
|
"""
|
|
if isinstance(self.ctx_or_interaction, commands.Context):
|
|
return self.ctx_or_interaction.author
|
|
elif isinstance(self.ctx_or_interaction, discord.Interaction):
|
|
return self.ctx_or_interaction.user
|
|
else:
|
|
log.error(f"Unexpected object type in Check._get_author: {type(self.ctx_or_interaction)}")
|
|
return None
|
|
|
|
def same(self, m):
|
|
author = self._get_author()
|
|
if author is None:
|
|
return False # Handle the case where author couldn't be determined
|
|
|
|
if isinstance(m, discord.Message):
|
|
return author == m.author
|
|
elif isinstance(m, discord.Interaction):
|
|
return author == m.user
|
|
else:
|
|
log.error(f"Unexpected object type in Check.same: {type(m)}")
|
|
return False
|
|
|
|
|
|
def comfirm(self, m):
|
|
return self.same(m) and m.content.lower() in ("yes", "no")
|
|
|
|
def valid_int(self, m):
|
|
return self.same and m.content.isdigit()
|
|
|
|
def valid_float(self, m):
|
|
try:
|
|
return self.same(m) and float(m.content) >= 1
|
|
except ValueError:
|
|
return False
|
|
|
|
def positive(self, m):
|
|
return self.same(m) and m.content.isdigit() and int(m.content) > 0
|
|
|
|
def role(self, m):
|
|
roles = [r.name for r in self.ctx.guild.roles if r.name != "Bot"]
|
|
return self.same(m) and m.content in roles
|
|
|
|
def member(self, m):
|
|
return self.same(m) and m.content in [x.name for x in self.ctx.guild.members]
|
|
|
|
def length_under(self, m):
|
|
try:
|
|
if isinstance(m, discord.Message):
|
|
content = m.content
|
|
elif isinstance(m, discord.Interaction):
|
|
content = m.data['components'][0]['components'][0]['value'] # Access modal input value
|
|
else:
|
|
raise ValueError("Unsupported object type for length_under check")
|
|
|
|
return self.same(m) and len(content) <= self.length
|
|
except TypeError:
|
|
raise ValueError("Length was not specified in Check")
|
|
|
|
def valid_image_url(self, m):
|
|
url = m.content.strip()
|
|
|
|
if not vals.url(url):
|
|
return False
|
|
|
|
valid_extensions = (".jpg", ".jpeg", ".png", ".gif")
|
|
if not any(url.lower().endswith(ext) for ext in valid_extensions):
|
|
return False
|
|
|
|
return True
|
|
|
|
def content(self, m):
|
|
try:
|
|
return self.same(m) and m.content in self.custom
|
|
except TypeError:
|
|
raise ValueError("A custom iterable was not set in Check") |