78 lines
1.9 KiB
Python
78 lines
1.9 KiB
Python
import asyncio
|
|
import functools
|
|
|
|
import aiohttp
|
|
from discord.ext import commands
|
|
|
|
from tuxbot.core.utils.functions.extra import ContextPlus
|
|
|
|
|
|
def upper_first(string: str) -> str:
|
|
return "".join(string[0].upper() + string[1:])
|
|
|
|
|
|
def typing(func):
|
|
@functools.wraps(func)
|
|
async def wrapped(*args, **kwargs):
|
|
context = (
|
|
args[0]
|
|
if isinstance(args[0], (commands.Context, ContextPlus))
|
|
else args[1]
|
|
)
|
|
|
|
async with context.typing():
|
|
await func(*args, **kwargs)
|
|
|
|
return wrapped
|
|
|
|
|
|
async def shorten(session, text: str, length: int) -> dict:
|
|
output = {"text": text[:length], "link": None}
|
|
|
|
if len(text) > length:
|
|
output["text"] += "[...]"
|
|
try:
|
|
async with session.post(
|
|
"https://paste.ramle.be/documents",
|
|
data=text.encode(),
|
|
timeout=aiohttp.ClientTimeout(total=2),
|
|
) as r:
|
|
output[
|
|
"link"
|
|
] = f"https://paste.ramle.be/{(await r.json())['key']}"
|
|
except (aiohttp.ClientError, asyncio.exceptions.TimeoutError):
|
|
pass
|
|
|
|
return output
|
|
|
|
|
|
def replace_in_dict(value: dict, search: str, replace: str) -> dict:
|
|
clean = {}
|
|
|
|
for k, v in value.items():
|
|
if isinstance(v, (str, bytes)):
|
|
v = v.replace(search, replace)
|
|
elif isinstance(v, list):
|
|
v = replace_in_list(v, search, replace)
|
|
elif isinstance(v, dict):
|
|
v = replace_in_dict(v, search, replace)
|
|
|
|
clean[k] = v
|
|
|
|
return clean
|
|
|
|
|
|
def replace_in_list(value: list, search: str, replace: str) -> list:
|
|
clean = []
|
|
|
|
for v in value:
|
|
if isinstance(v, (str, bytes)):
|
|
v = v.replace(search, replace)
|
|
elif isinstance(v, list):
|
|
v = replace_in_list(v, search, replace)
|
|
elif isinstance(v, dict):
|
|
v = replace_in_dict(v, search, replace)
|
|
|
|
clean.append(v)
|
|
|
|
return clean
|