diff --git a/speedtest/__init__.py b/speedtest/__init__.py new file mode 100644 index 0000000..f91daf6 --- /dev/null +++ b/speedtest/__init__.py @@ -0,0 +1,5 @@ +from .speedtest import Speedtest + + +async def setup(bot): + await bot.add_cog(Speedtest(bot)) diff --git a/speedtest/info.json b/speedtest/info.json new file mode 100644 index 0000000..f331eac --- /dev/null +++ b/speedtest/info.json @@ -0,0 +1,12 @@ +{ + "author" : ["SeaswimmerTheFsh (seasw.)"], + "install_msg" : "Thank you for installing SeaUtils!\nYou can find the source code of this cog [here](https://coastalcommits.com/SeaswimmerTheFsh/SeaCogs).", + "name" : "SeaUtils", + "short" : "A collection of useful utilities.", + "description" : "A collection of useful utilities.", + "end_user_data_statement" : "This cog does not store end user data.", + "hidden": true, + "disabled": false, + "min_bot_version": "3.5.0", + "min_python_version": [3, 8, 0] +} diff --git a/speedtest/speedtest.py b/speedtest/speedtest.py new file mode 100644 index 0000000..bdfacda --- /dev/null +++ b/speedtest/speedtest.py @@ -0,0 +1,53 @@ +# _____ _ +# / ____| (_) +# | (___ ___ __ _ _____ ___ _ __ ___ _ __ ___ ___ _ __ +# \___ \ / _ \/ _` / __\ \ /\ / / | '_ ` _ \| '_ ` _ \ / _ \ '__| +# ____) | __/ (_| \__ \\ V V /| | | | | | | | | | | | __/ | +# |_____/ \___|\__,_|___/ \_/\_/ |_|_| |_| |_|_| |_| |_|\___|_| + +import json +import subprocess + +import discord +from redbot.core import commands +from redbot.core.bot import Red +from redbot.core.utils import chat_formatting as cf + + +class Speedtest(commands.Cog): + """A collection of random utilities.""" + + __author__ = ["SeaswimmerTheFsh"] + __version__ = "1.0.0" + + def __init__(self, bot: Red): + self.bot = bot + + def format_help_for_context(self, ctx: commands.Context) -> str: + pre_processed = super().format_help_for_context(ctx) or "" + n = "\n" if "\n\n" not in pre_processed else "" + text = [ + f"{pre_processed}{n}", + f"Cog Version: **{self.__version__}**", + f"Author: {cf.humanize_list(self.__author__)}" + ] + return "\n".join(text) + + @commands.command() + @commands.is_owner() + async def speedtest(self, ctx: commands.Context): + """Run a speedtest.""" + msg = await ctx.maybe_send_embed("Running speedtest...") + result = subprocess.run(["speedtest", "-f json", "--accept-license"], stdout=subprocess.PIPE) + r = json.loads(result.stdout.decode("utf-8")) + if await ctx.embed_requested(): + embed = discord.Embed(title="Speedtest Results", link=r["result"]["url"], color=await ctx.embed_color()) + embed.add_field(name="Bandwidth", value=f"{r['download']['bandwidth'] / 1_000_000:.2f} Mbps down, {r['upload']['bandwidth'] / 1_000_000:.2f} Mbps up") + embed.add_field(name="Ping", value=f"Base: {round(r['ping']['latency'])} ms \nDownload: {round(r['download']['latency']['igm'])} ms \nUpload: {round(r['upload']['latency']['igm'])} ms") + embed.set_image(url=r["result"]["image"] + ".png") + await msg.edit(embed=embed) + else: + await msg.edit(content=f"**Speedtest Results**\n" + f"**Bandwidth**: {r['download']['bandwidth'] / 1_000_000:.2f} Mbps down, {r['upload']['bandwidth'] / 1_000_000:.2f} Mbps up\n" + f"**Ping**: Base: {round(r['ping']['latency'])} ms, Download: {round(r['download']['latency']['igm'])} ms, Upload: {round(r['upload']['latency']['igm'])} ms\n" + f"**Results**: {r['result']['url']}")