522e3448df
added units for the singular versions of days, hours, minutes, and seconds - also added abbreivations
42 lines
1.8 KiB
Python
42 lines
1.8 KiB
Python
from datetime import timedelta
|
|
import revolt
|
|
from revolt.ext import commands
|
|
|
|
class Moderation(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
def parse_timedelta(self, input_str):
|
|
# Split the string into its components (e.g., "1 day 3 hours" becomes ["1", "day", "3", "hours"])
|
|
components = input_str.split()
|
|
|
|
# Define a dictionary to map time units to their corresponding `timedelta` attribute
|
|
units = {"days": "days", "hours": "hours", "minutes": "minutes", "seconds": "seconds", "day": "days", "hour": "hours", "minute": "minutes", "second": "seconds", "d": "days", "h": "hours", "m": "minutes", "s": "seconds"}
|
|
|
|
# Iterate over the components, taking pairs of values and units
|
|
values_units = zip(components[::2], components[1::2])
|
|
|
|
# Initialize a dictionary to store the values for each unit
|
|
values = {}
|
|
|
|
# Parse the values and units into the dictionary
|
|
for value, unit in values_units:
|
|
# Convert the value to an integer
|
|
value = int(value)
|
|
# Map the unit to the corresponding `timedelta` attribute and store the value
|
|
values[units[unit]] = value
|
|
|
|
# Create and return the `timedelta` object
|
|
return timedelta(**values)
|
|
|
|
|
|
@commands.command(name="mute", aliases="timeout")
|
|
async def mute(self, ctx, target: revolt.Member, duration: str = "1 hour"):
|
|
parsed_time = Moderation.parse_timedelta(self, duration)
|
|
await target.timeout(parsed_time)
|
|
await ctx.message.reply(f"{target.mention} has been timed out for {str(parsed_time)}!")
|
|
|
|
@commands.command()
|
|
async def timedeltaconvert(self, ctx, *, duration: str = "1 hour"):
|
|
parsed_time = Moderation.parse_timedelta(self, duration)
|
|
await ctx.send(str(parsed_time))
|