mirror of
https://github.com/nexy7574/LCC-bot.git
synced 2024-09-19 18:16:34 +01:00
104 lines
3 KiB
Python
104 lines
3 KiB
Python
import time
|
|
from typing import TYPE_CHECKING, Optional
|
|
from urllib.parse import urlparse
|
|
|
|
from discord.ext import commands
|
|
|
|
from .console import *
|
|
from .db import *
|
|
from .views import *
|
|
|
|
if TYPE_CHECKING:
|
|
from datetime import datetime
|
|
|
|
|
|
class JimmyBanException(discord.CheckFailure):
|
|
def __init__(self, until: "datetime", reason: str):
|
|
super().__init__(reason)
|
|
self.until = until
|
|
self.reason = reason
|
|
|
|
def __str__(self):
|
|
ok = discord.utils.format_dt(self.until, "R")
|
|
return (
|
|
f"\N{cross mark} You are not allowed to use commands right now. You will be unbanned {ok}.\n"
|
|
f"Ban reason:\n>>> {self.reason}"
|
|
)
|
|
|
|
def __repr__(self):
|
|
return f"<JimmyBanException until={self.until!r} reason={self.reason!r}>"
|
|
|
|
|
|
class Timer:
|
|
def __init__(self, target: dict = None, name: str = ...):
|
|
self.target = target
|
|
self.name = name
|
|
self._start_time = None
|
|
self._end_time = None
|
|
|
|
@property
|
|
def total(self) -> float:
|
|
if not self._start_time:
|
|
raise RuntimeError("Timer has not been started.")
|
|
if not self._end_time:
|
|
return time.time() - self._start_time
|
|
return self._end_time - self._start_time
|
|
|
|
def __enter__(self):
|
|
self._start_time = time.time()
|
|
return self
|
|
|
|
def __exit__(self, exc_type, exc_val, exc_tb):
|
|
self._end_time = time.time()
|
|
if self.target and self.name is not ...:
|
|
self.target[self.name] = self.total
|
|
|
|
|
|
def simple_embed_paginator(
|
|
lines: list[str], *, assert_ten: bool = False, empty_is_none: bool = True, **kwargs
|
|
) -> Optional[list[discord.Embed]]:
|
|
"""Paginates x lines into x embeds."""
|
|
if not lines and empty_is_none is True:
|
|
return
|
|
|
|
kwargs.setdefault("description", "")
|
|
embeds = [discord.Embed(**kwargs)]
|
|
for line in lines:
|
|
embed = embeds[-1]
|
|
total_length = len(embed)
|
|
description_length = len(embed.description)
|
|
if total_length + len(line) > 6000 or description_length + len(line) > 4096:
|
|
embed = discord.Embed(**kwargs)
|
|
embed.description += line + "\n"
|
|
embeds.append(embed)
|
|
else:
|
|
embed.description += line + "\n"
|
|
|
|
if assert_ten:
|
|
assert len(embeds) <= 10, "Too many embeds."
|
|
return embeds
|
|
|
|
|
|
def hyperlink(url: str, *, text: str = None, max_length: int = None) -> str:
|
|
if max_length < len(url):
|
|
raise ValueError(f"Max length ({max_length}) is too low for provided URL ({len(url)}). Hyperlink impossible.")
|
|
|
|
fmt = "[{}]({})"
|
|
if text:
|
|
fmt = fmt.format(text, url)
|
|
else:
|
|
parsed = urlparse(url)
|
|
fmt = fmt.format(parsed.hostname, url)
|
|
|
|
if len(fmt) > max_length:
|
|
return url
|
|
return fmt
|
|
|
|
|
|
def owner_or_admin():
|
|
async def predicate(ctx: commands.Context):
|
|
if ctx.author.guild_permissions.administrator or await ctx.bot.is_owner(ctx.author):
|
|
return True
|
|
raise commands.MissingPermissions(["administrator"])
|
|
|
|
return commands.check(predicate)
|