Conatum
a0cd2530c9
Added better error handling for message responses. Changed the `embed_reply` colour.
71 lines
1.8 KiB
Python
71 lines
1.8 KiB
Python
import discord
|
|
from cmdClient import Context
|
|
|
|
from data import tables
|
|
from core import Lion
|
|
from settings import GuildSettings, UserSettings
|
|
|
|
|
|
@Context.util
|
|
async def embed_reply(ctx, desc, colour=discord.Colour.orange(), **kwargs):
|
|
"""
|
|
Simple helper to embed replies.
|
|
All arguments are passed to the embed constructor.
|
|
`desc` is passed as the `description` kwarg.
|
|
"""
|
|
embed = discord.Embed(description=desc, colour=colour, **kwargs)
|
|
try:
|
|
return await ctx.reply(embed=embed, reference=ctx.msg)
|
|
except discord.NotFound:
|
|
return await ctx.reply(embed=embed)
|
|
|
|
|
|
@Context.util
|
|
async def error_reply(ctx, error_str, **kwargs):
|
|
"""
|
|
Notify the user of a user level error.
|
|
Typically, this will occur in a red embed, posted in the command channel.
|
|
"""
|
|
embed = discord.Embed(
|
|
colour=discord.Colour.red(),
|
|
description=error_str
|
|
)
|
|
message = None
|
|
try:
|
|
message = await ctx.ch.send(embed=embed, reference=ctx.msg, **kwargs)
|
|
except discord.NotFound:
|
|
message = await ctx.ch.send(embed=embed, **kwargs)
|
|
except discord.Forbidden:
|
|
message = await ctx.reply(error_str)
|
|
finally:
|
|
if message:
|
|
ctx.sent_messages.append(message)
|
|
return message
|
|
|
|
|
|
def context_property(func):
|
|
setattr(Context, func.__name__, property(func))
|
|
return func
|
|
|
|
|
|
@context_property
|
|
def best_prefix(ctx):
|
|
return ctx.client.prefix
|
|
|
|
|
|
@context_property
|
|
def guild_settings(ctx):
|
|
if ctx.guild:
|
|
tables.guild_config.fetch_or_create(ctx.guild.id)
|
|
return GuildSettings(ctx.guild.id if ctx.guild else 0)
|
|
|
|
|
|
@context_property
|
|
def author_settings(ctx):
|
|
return UserSettings(ctx.author.id)
|
|
|
|
|
|
@context_property
|
|
def alion(ctx):
|
|
return Lion.fetch(ctx.guild.id if ctx.guild else 0, ctx.author.id)
|