mirror of
https://github.com/RGBCube/minearchy-bot
synced 2025-07-27 08:57:46 +00:00
Cleanup
This commit is contained in:
parent
5b7fb2deba
commit
5ab80712d0
6 changed files with 45 additions and 36 deletions
|
@ -2,12 +2,12 @@ from __future__ import annotations
|
|||
|
||||
__all__ = ("MinearchyBot",)
|
||||
|
||||
from asyncio import run as run_coro
|
||||
from inspect import cleandoc as strip_doc
|
||||
from itertools import chain as chain_iter
|
||||
import asyncio
|
||||
from inspect import cleandoc as strip
|
||||
from itertools import chain
|
||||
from pathlib import Path
|
||||
from time import time as current_time
|
||||
from traceback import format_exc as format_exit
|
||||
from time import time as get_time
|
||||
from traceback import format_exc as format_exception
|
||||
|
||||
from aiohttp import ClientSession as AIOHTTPSession
|
||||
from discord import AllowedMentions, Game, Intents, Webhook
|
||||
|
@ -58,7 +58,7 @@ class MinearchyBot(CommandsBot):
|
|||
@override
|
||||
async def on_ready(self) -> None:
|
||||
print(
|
||||
strip_doc(
|
||||
strip(
|
||||
f"""
|
||||
Connected to Discord!
|
||||
User: {self.user}
|
||||
|
@ -67,12 +67,12 @@ class MinearchyBot(CommandsBot):
|
|||
)
|
||||
)
|
||||
|
||||
self.ready_timestamp = current_time()
|
||||
self.ready_timestamp = get_time()
|
||||
await self.log_webhook.send("Bot is now online!")
|
||||
|
||||
async def load_extensions(self) -> None:
|
||||
cogs = Path(__file__).parent / "cogs"
|
||||
for file_name in chain_iter(
|
||||
for file_name in chain(
|
||||
map(
|
||||
lambda file_path: ".".join(file_path.relative_to(cogs.parent.parent).parts)[:-3],
|
||||
cogs.rglob("*.py"),
|
||||
|
@ -83,7 +83,7 @@ class MinearchyBot(CommandsBot):
|
|||
await self.load_extension(file_name)
|
||||
print(f"Loaded {file_name}")
|
||||
except (ExtensionFailed, NoEntryPointError):
|
||||
print(f"Couldn't load {file_name}:\n{format_exit()}")
|
||||
print(f"Couldn't load {file_name}:\n{format_exception()}")
|
||||
|
||||
@override
|
||||
def run(self) -> None:
|
||||
|
@ -96,6 +96,6 @@ class MinearchyBot(CommandsBot):
|
|||
await self.start(self.token, reconnect=True)
|
||||
|
||||
try:
|
||||
run_coro(runner())
|
||||
asyncio.run(runner())
|
||||
except KeyboardInterrupt:
|
||||
pass
|
||||
|
|
|
@ -1,23 +1,29 @@
|
|||
from __future__ import annotations
|
||||
|
||||
from json import load as parse_json
|
||||
import json
|
||||
from os import environ as env
|
||||
from pathlib import Path
|
||||
|
||||
from uvloop import install as install_uvloop
|
||||
import uvloop
|
||||
|
||||
from . import MinearchyBot
|
||||
|
||||
|
||||
def main() -> None:
|
||||
install_uvloop()
|
||||
uvloop.install()
|
||||
|
||||
with (Path(__file__).parent / "config.json").open() as f:
|
||||
config = parse_json(f)
|
||||
config = json.loads(
|
||||
(
|
||||
Path(__file__).parent / "config.json"
|
||||
).read_text()
|
||||
)
|
||||
|
||||
for key in ("HIDE", "NO_UNDERSCORE"):
|
||||
env[f"JISHAKU_{key}"] = "True"
|
||||
env[f"JISHAKU_HIDE"] = "True"
|
||||
env[f"JISHAKU_NO_UNDERSCORE"] = "True"
|
||||
|
||||
bot = MinearchyBot(token=config["BOT_TOKEN"], webhook_url=config["WEBHOOK_URL"])
|
||||
bot = MinearchyBot(
|
||||
token=config["BOT_TOKEN"],
|
||||
webhook_url=config["WEBHOOK_URL"]
|
||||
)
|
||||
|
||||
bot.run()
|
||||
|
|
|
@ -1,8 +1,8 @@
|
|||
from __future__ import annotations
|
||||
|
||||
from asyncio import gather as await_parallel
|
||||
import asyncio
|
||||
from contextlib import suppress as suppress_error
|
||||
from traceback import format_exception as format_exit
|
||||
from traceback import format_exception
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
from discord import HTTPException
|
||||
|
@ -61,10 +61,10 @@ class ErrorHandler(Cog):
|
|||
await ctx.reply("Invalid channel.")
|
||||
|
||||
else:
|
||||
trace = "".join(format_exit(type(error), error, error.__traceback__))
|
||||
trace = "".join(format_exception(type(error), error, error.__traceback__))
|
||||
print(f"Ignoring exception in command {ctx.command}:\n{trace}")
|
||||
|
||||
await await_parallel(
|
||||
await asyncio.gather(
|
||||
self.bot.log_webhook.send(f"<@512640455834337290>```{trace}```"),
|
||||
ctx.reply(
|
||||
"An error occurred while executing the command. The error has been reported."
|
||||
|
|
|
@ -2,7 +2,8 @@ from __future__ import annotations
|
|||
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
from discord.ext.commands import Cog, command, group
|
||||
from discord.ext import commands
|
||||
from discord.ext.commands import Cog, command
|
||||
from discord.ui import Button, View
|
||||
|
||||
if TYPE_CHECKING:
|
||||
|
@ -19,7 +20,7 @@ class MinecraftServer(
|
|||
def __init__(self, bot: MinearchyBot) -> None:
|
||||
self.bot = bot
|
||||
|
||||
@group(
|
||||
@commands.group(
|
||||
invoke_without_command=True,
|
||||
brief="Sends the server IP.",
|
||||
help="Sends the server IP.",
|
||||
|
|
|
@ -2,14 +2,15 @@ from __future__ import annotations
|
|||
|
||||
from collections import defaultdict as DefaultDict, deque as Deque
|
||||
from datetime import timedelta as TimeDelta
|
||||
from inspect import cleandoc as strip_doc
|
||||
from inspect import cleandoc as strip
|
||||
from io import BytesIO
|
||||
from platform import python_version
|
||||
from time import monotonic as ping_time, time as current_time
|
||||
from time import monotonic as get_monotonic, time as get_time
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
from discord import CategoryChannel, Color, Embed, File, TextChannel
|
||||
from discord.ext.commands import Cog, command, has_permissions
|
||||
from discord.ext import commands
|
||||
from discord.ext.commands import Cog, command
|
||||
from discord.utils import escape_markdown
|
||||
|
||||
from ..utils import override
|
||||
|
@ -52,9 +53,9 @@ class Miscellaneous(
|
|||
help="Sends the bots ping."
|
||||
)
|
||||
async def ping(self, ctx: Context) -> None:
|
||||
ts = ping_time()
|
||||
ts = get_monotonic()
|
||||
message = await ctx.reply("Pong!")
|
||||
ts = ping_time() - ts
|
||||
ts = get_monotonic() - ts
|
||||
await message.edit(content=f"Pong! `{int(ts * 1000)}ms`")
|
||||
|
||||
@command(
|
||||
|
@ -63,11 +64,11 @@ class Miscellaneous(
|
|||
)
|
||||
async def info(self, ctx: Context) -> None:
|
||||
await ctx.reply(
|
||||
strip_doc(
|
||||
strip(
|
||||
f"""
|
||||
__**Bot Info**__
|
||||
**Python Version:** v{python_version()}
|
||||
**Uptime:** `{TimeDelta(seconds=int(current_time() - self.bot.ready_timestamp))}`
|
||||
**Uptime:** `{TimeDelta(seconds=int(get_time() - self.bot.ready_timestamp))}`
|
||||
"""
|
||||
)
|
||||
)
|
||||
|
@ -155,7 +156,7 @@ class Miscellaneous(
|
|||
" isn't specified, it uses the current channel."
|
||||
),
|
||||
)
|
||||
@has_permissions(manage_messages=True) # needs to be able to delete messages to run the command
|
||||
@commands.has_permissions(manage_messages=True) # needs to be able to delete messages to run the command
|
||||
async def snipe(self, ctx: Context, channel: TextChannel | None = None) -> None:
|
||||
if channel is None:
|
||||
channel = ctx.channel
|
||||
|
@ -185,7 +186,7 @@ class Miscellaneous(
|
|||
|
||||
embed.add_field(
|
||||
name=str(i) + ("" if i else " (latest)"),
|
||||
value=strip_doc(
|
||||
value=strip(
|
||||
f"""
|
||||
Author: {message.author.mention} (ID: {message.author.id}, Plain: {escape_markdown(str(message.author))})
|
||||
Deleted at: <t:{ts}:F> (Relative: <t:{ts}:R>)
|
||||
|
@ -205,7 +206,7 @@ class Miscellaneous(
|
|||
if not message.guild:
|
||||
return
|
||||
|
||||
self.sniped[message.channel.id].appendleft((message, int(current_time())))
|
||||
self.sniped[message.channel.id].appendleft((message, int(get_time())))
|
||||
|
||||
while len(self.sniped[message.channel.id]) > 5:
|
||||
self.sniped[message.channel.id].pop()
|
||||
|
|
|
@ -3,7 +3,8 @@ from __future__ import annotations
|
|||
from datetime import timedelta as TimeDelta
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
from discord.ext.commands import Cog, command, has_permissions
|
||||
from discord.ext import commands
|
||||
from discord.ext.commands import Cog, command
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from discord import Member
|
||||
|
@ -27,7 +28,7 @@ class Moderation(Cog):
|
|||
brief="Times out a user.",
|
||||
help="Times out a user."
|
||||
)
|
||||
@has_permissions(manage_messages=True)
|
||||
@commands.has_permissions(manage_messages=True)
|
||||
async def timeout(self, ctx: Context, member: Member, duration: str = "1d") -> None:
|
||||
if duration[-1] not in self.time_values or len(duration) < 2:
|
||||
await ctx.reply("Invalid duration. Valid durations are: d, h, m, s.")
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue