tuxbot-bot/tuxbot/cogs/Utils/utils.py

255 lines
8.3 KiB
Python

import inspect
import logging
import os
import platform
import discord
import humanize
import psutil
from discord.ext import commands
from tuxbot import version_info, __version__
from tuxbot.core.utils.functions.extra import command_extra, ContextPlus
from tuxbot.core.bot import Tux
from tuxbot.core.i18n import (
Translator,
)
from .functions.info import fetch_info
log = logging.getLogger("tuxbot.cogs.Utils")
_ = Translator("Utils", __file__)
class Utils(commands.Cog, name="Utils"):
def __init__(self, bot: Tux):
self.bot = bot
# =========================================================================
# =========================================================================
@command_extra(name="info", aliases=["about"])
async def _info(self, ctx: ContextPlus):
proc = psutil.Process()
infos = fetch_info()
with proc.oneshot():
mem = proc.memory_full_info()
e = discord.Embed(
title=_("Information about TuxBot", ctx, self.bot.config),
color=0x89C4F9,
)
e.add_field(
name=_(
"__:busts_in_silhouette: Development__",
ctx,
self.bot.config,
),
value="**Romain#5117:** [git](https://git.gnous.eu/Romain)\n"
"**Outout#4039:** [git](https://git.gnous.eu/mael)\n",
inline=True,
)
e.add_field(
name="__<:python:596577462335307777> Python__",
value=f"**python** `{platform.python_version()}`\n"
f"**discord.py** `{discord.__version__}`",
inline=True,
)
e.add_field(
name="__:gear: Usage__",
value=_(
"**{}** physical memory\n**{}** virtual memory",
ctx,
self.bot.config,
).format(
humanize.naturalsize(mem.rss),
humanize.naturalsize(mem.vms),
),
inline=True,
)
e.add_field(
name=_("__Servers count__", ctx, self.bot.config),
value=str(len(self.bot.guilds)),
inline=True,
)
e.add_field(
name=_("__Channels count__", ctx, self.bot.config),
value=str(len(list(self.bot.get_all_channels()))),
inline=True,
)
e.add_field(
name=_("__Members count__", ctx, self.bot.config),
value=str(len(list(self.bot.get_all_members()))),
inline=True,
)
e.add_field(
name=_("__:file_folder: Files__", ctx, self.bot.config),
value=f"{infos.get('file_amount')} "
f"*({infos.get('python_file_amount')}"
f" <:python:596577462335307777>)*",
inline=True,
)
e.add_field(
name=_("__¶ Lines__", ctx, self.bot.config),
value=f"{infos.get('total_lines')} "
f"*({infos.get('total_python_class')} class,"
f" {infos.get('total_python_functions')} functions,"
f" {infos.get('total_python_coroutines')} coroutines,"
f" {infos.get('total_python_comments')} comments)*",
inline=True,
)
e.add_field(
name=_("__Latest changes__", ctx, self.bot.config),
value=version_info.info,
inline=False,
)
e.add_field(
name=_("__:link: Links__", ctx, self.bot.config),
value="[tuxbot.gnous.eu](https://tuxbot.gnous.eu/) "
"| [gnous.eu](https://gnous.eu/) "
"| [git](https://git.gnous.eu/gnouseu/tuxbot-bot) "
"| [status](https://status.gnous.eu/check/154250) "
+ _("| [Invite]", ctx, self.bot.config)
+ "(https://discordapp.com/oauth2/authorize?client_id="
"301062143942590465&scope=bot&permissions=268749888)",
inline=False,
)
e.set_footer(text=f"version: {__version__} • prefix: {ctx.prefix}")
await ctx.send(embed=e)
# =========================================================================
@command_extra(name="credits", aliases=["contributors", "authors"])
async def _credits(self, ctx: ContextPlus):
e = discord.Embed(
title=_("Contributors", ctx, self.bot.config), color=0x36393F
)
e.add_field(
name="**Romain#5117** ",
value="• [github](https://github.com/Rom1-J)\n"
"• [gitea](https://git.gnous.eu/Romain)\n"
"• romain@gnous.eu",
inline=True,
)
e.add_field(
name="**Outout#4039** ",
value="• [gitea](https://git.gnous.eu/mael)\n"
"• [@outoutxyz](https://twitter.com/outouxyz)\n"
"• mael@gnous.eu",
inline=True,
)
await ctx.send(embed=e)
# =========================================================================
@command_extra(name="invite")
async def _invite(self, ctx: ContextPlus):
basic_perms = discord.Permissions(
add_reactions=True,
read_messages=True,
send_messages=True,
manage_messages=True,
embed_links=True,
attach_files=True,
read_message_history=True,
external_emojis=True,
connect=True,
speak=True,
manage_roles=True,
)
admin_perms = discord.Permissions(
create_instant_invite=True,
kick_members=True,
ban_members=True,
add_reactions=True,
read_messages=True,
send_messages=True,
manage_messages=True,
embed_links=True,
attach_files=True,
read_message_history=True,
external_emojis=True,
connect=True,
speak=True,
manage_roles=True,
)
e = discord.Embed(
title=_("Invite", ctx, self.bot.config), color=0x36393F
)
e.add_field(
name=_("Minimal", ctx, self.bot.config),
value=_(
"The minimum permissions include the strict requirements for "
"the proper functioning of all basics commands.\n",
ctx,
self.bot.config,
)
+ _("[Add!]", ctx, self.bot.config)
+ f"({discord.utils.oauth_url(self.bot.user.id, basic_perms)})",
inline=False,
)
e.add_field(
name=_("Admin", ctx, self.bot.config),
value=_(
"All minimal permissions + extra permissions for admin "
"commands such as kick and ban\n",
ctx,
self.bot.config,
)
+ _("[Add!]", ctx, self.bot.config)
+ f"({discord.utils.oauth_url(self.bot.user.id, admin_perms)})",
inline=False,
)
await ctx.send(embed=e)
# =========================================================================
@command_extra(name="source")
async def _source(self, ctx: ContextPlus, *, name=None):
base_url = "https://github.com/Rom1-J/tuxbot-bot"
if not name:
return await ctx.send(f"<{base_url}>")
cmd = self.bot.get_command(name)
if cmd:
src = cmd.callback.__code__
rpath = src.co_filename
else:
return await ctx.send(
_("Unable to find `{}`", ctx, self.bot.config).format(name)
)
try:
lines, start_line = inspect.getsourcelines(src)
except OSError:
return await ctx.send(
_(
"Unable to fetch lines for `{}`", ctx, self.bot.config
).format(name)
)
location = (
os.path.relpath(rpath)
.replace("\\", "/")
.split("site-packages/")[1]
)
final_url = (
f"<{base_url}/tree/master/{location}#L{start_line}"
f"-L{start_line + len(lines) - 1}>"
)
await ctx.send(final_url)