simplify finding all modules

ubot/modules/_sudo.py

@@ -0,0 +1,166 @@
+# SPDX-License-Identifier: GPL-2.0-or-later
+
+import asyncio
+import os
+from platform import python_version
+
+import psutil
+from telethon import version
+
+from ubot.micro_bot import ldr, micro_bot
+
+
+@ldr.add("reload", sudo=True, hide_help=True)
+async def reload_modules(event):
+    reload_msg = await event.reply("Reloading modules…")
+
+    errors = ldr.reload_all_modules()
+
+    if errors:
+        await reload_msg.edit(errors)
+    else:
+        try:
+            await reload_msg.edit("Successfully reloaded.")
+        except:
+            pass
+
+
+@ldr.add("sysd", sudo=True, hide_help=True)
+async def sysd(event):
+    try:
+        neo = "neofetch --stdout"
+
+        fetch = await asyncio.create_subprocess_shell(
+            neo,
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE,
+        )
+
+        stdout, stderr = await fetch.communicate()
+
+        await event.reply(f"`{stdout.decode().strip()}{stderr.decode().strip()}`")
+    except FileNotFoundError:
+        await event.reply("Neofetch not found!")
+
+
+@ldr.add("alive", sudo=True, hide_help=True)
+async def alive(event):
+    alive_format = "**Telethon version:** {0}\n" \
+                   "**Python version:** {1}\n" \
+                   "**Memory usage:** {2}MiB"
+
+    mem_usage = int(psutil.Process(os.getpid()).memory_info().rss / 1048576)
+
+    await event.reply(alive_format.format(version.__version__, python_version(), mem_usage))
+
+
+@ldr.add("shutdown", owner=True, hide_help=True)
+async def shutdown(event):
+    await event.reply("Goodbye…")
+    await micro_bot.stop_client()
+
+
+@ldr.add("blacklist", sudo=True, hide_help=True)
+async def add_blacklist(event):
+    user_entity = await get_user(event)
+
+    if not user_entity:
+        return
+
+    userid = user_entity.id
+    userfullname = f"{user_entity.first_name} {user_entity.last_name or ''}".strip()
+
+    ldr.settings.add_to_list("blacklisted_users", userid)
+    await event.reply(f"Successfully blacklisted **{userfullname}** (`{userid}`)")
+
+
+@ldr.add("unblacklist", sudo=True, hide_help=True)
+async def rem_blacklist(event):
+    user_entity = await get_user(event)
+
+    if not user_entity:
+        return
+
+    userid = user_entity.id
+    userfullname = f"{user_entity.first_name} {user_entity.last_name or ''}".strip()
+
+    ldr.settings.remove_from_list("blacklisted_users", userid)
+    await event.reply(f"Successfully unblacklisted **{userfullname}** (`{userid}`)")
+
+
+@ldr.add("showblacklist", sudo=True, hide_help=True)
+async def show_blacklist(event):
+    blacklist_string = ""
+
+    for i in ldr.settings.get_list('blacklisted_users'):
+        blacklist_string += f"\n{i}"
+    
+    await event.reply(f"**Blacklisted users:**\n{blacklist_string}")
+
+
+@ldr.add("sudo", owner=True, hide_help=True)
+async def add_sudo(event):
+    user_entity = await get_user(event)
+
+    if not user_entity:
+        return
+
+    userid = user_entity.id
+    userfullname = f"{user_entity.first_name} {user_entity.last_name or ''}".strip()
+
+    ldr.settings.add_to_list("sudo_users", userid)
+    await event.reply(f"Successfully sudo'd **{userfullname}** (`{userid}`)")
+
+
+@ldr.add("unsudo", owner=True, hide_help=True)
+async def rem_sudo(event):
+    user_entity = await get_user(event)
+
+    if not user_entity:
+        return
+
+    userid = user_entity.id
+    userfullname = f"{user_entity.first_name} {user_entity.last_name or ''}".strip()
+
+    ldr.settings.remove_from_list("sudo_users", userid)
+    await event.reply(f"Successfully unsudo'd **{userfullname}** (`{userid}`)")
+
+
+@ldr.add("showsudo", sudo=True, hide_help=True)
+async def show_sudo(event):
+    sudo_string = ""
+
+    for i in ldr.settings.get_list("sudo_users"):
+        sudo_string += f"\n{i}"
+    
+    await event.reply(f"**Sudo users:**\n{sudo_string}")
+
+
+async def get_user(event):
+    if event.args:
+        try:
+            event.args = int(event.args)
+        except:
+            pass
+
+        try:
+            return await event.client.get_entity(event.args)
+        except (ValueError, TypeError):
+            await event.reply("The ID or username you provided was invalid!")
+            return
+    elif event.is_reply:
+        reply = await event.get_reply_message()
+        reply_id = reply.from_id
+
+        if reply_id:
+            try:
+                return await event.client.get_entity(reply_id)
+            except (ValueError, TypeError):
+                await event.reply("There was an error getting the user's ID!")
+                return
+        else:
+            await event.reply("Sudoing failed!")
+            return
+    else:
+        await event.reply("Give me a user ID, username or reply!")
+        return