90 lines
3.6 KiB
Python
90 lines
3.6 KiB
Python
# © 2018–2020 io mintz <io@mintz.cc>
|
||
#
|
||
# Emote Manager is free software: you can redistribute it and/or modify
|
||
# it under the terms of the GNU Affero General Public License as
|
||
# published by the Free Software Foundation, either version 3 of the
|
||
# License, or (at your option) any later version.
|
||
#
|
||
# Emote Manager is distributed in the hope that it will be useful,
|
||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
# GNU Affero General Public License for more details.
|
||
#
|
||
# You should have received a copy of the GNU Affero General Public License
|
||
# along with Emote Manager. If not, see <https://www.gnu.org/licenses/>.
|
||
|
||
import utils
|
||
import asyncio
|
||
import humanize
|
||
import datetime
|
||
from discord.ext import commands
|
||
|
||
class MissingManageEmojisPermission(commands.MissingPermissions):
|
||
"""The invoker or the bot doesn't have permissions to manage server emojis."""
|
||
|
||
def __init__(self):
|
||
super(Exception, self).__init__(
|
||
f'{utils.SUCCESS_EMOJIS[False]} '
|
||
"Sorry, you don't have enough permissions to run this command. "
|
||
'You and I both need the Manage Emojis permission.')
|
||
|
||
class EmoteManagerError(commands.CommandError):
|
||
"""Generic error with the bot. This can be used to catch all bot errors."""
|
||
pass
|
||
|
||
class ImageProcessingTimeoutError(EmoteManagerError, asyncio.TimeoutError):
|
||
pass
|
||
|
||
class ImageResizeTimeoutError(ImageProcessingTimeoutError):
|
||
"""Resizing the image took too long."""
|
||
def __init__(self):
|
||
super().__init__('Error: resizing the image took too long.')
|
||
|
||
class ImageConversionTimeoutError(ImageProcessingTimeoutError):
|
||
def __init__(self):
|
||
super().__init__('Error: converting the image to a GIF took too long.')
|
||
|
||
class HTTPException(EmoteManagerError):
|
||
"""The server did not respond with an OK status code. This is only for non-Discord HTTP requests."""
|
||
def __init__(self, status):
|
||
super().__init__(f'URL error: server returned error code {status}')
|
||
|
||
class RateLimitedError(EmoteManagerError):
|
||
def __init__(self, retry_at):
|
||
if isinstance(retry_at, float):
|
||
# it took me about an HOUR to realize i had to pass tz because utcfromtimestamp returns a NAÏVE time obj!
|
||
retry_at = datetime.datetime.fromtimestamp(retry_at, tz=datetime.timezone.utc)
|
||
# humanize.naturaltime is annoying to work with due to timezones so we use this
|
||
delta = humanize.naturaldelta(retry_at, when=datetime.datetime.now(tz=datetime.timezone.utc))
|
||
super().__init__(f'Error: Discord told me to slow down! Please retry this command in {delta}.')
|
||
|
||
class EmoteNotFoundError(EmoteManagerError):
|
||
"""An emote with that name was not found"""
|
||
def __init__(self, name):
|
||
super().__init__(f'An emote called `{name}` does not exist in this server.')
|
||
|
||
class FileTooBigError(EmoteManagerError):
|
||
def __init__(self, size, limit):
|
||
self.size = size
|
||
self.limit = limit
|
||
|
||
class InvalidFileError(EmoteManagerError):
|
||
"""The file is not a zip, tar, GIF, PNG, JPG, or WEBP file."""
|
||
def __init__(self):
|
||
super().__init__('Invalid file given.')
|
||
|
||
class InvalidImageError(InvalidFileError):
|
||
"""The image is not a GIF, PNG, or JPG"""
|
||
def __init__(self):
|
||
super(Exception, self).__init__('The image supplied was not a GIF, PNG, JPG, or WEBP file.')
|
||
|
||
class PermissionDeniedError(EmoteManagerError):
|
||
"""Raised when a user tries to modify an emote without the Manage Emojis permission"""
|
||
def __init__(self, name):
|
||
super().__init__(f"You're not authorized to modify `{name}`.")
|
||
|
||
class DiscordError(Exception):
|
||
"""Usually raised when the client cache is being baka"""
|
||
def __init__(self):
|
||
super().__init__('Discord seems to be having issues right now, please try again later.')
|