EmoteManager/utils/errors.py

# © 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.')