squash

ubot/modules/system.py

@@ -0,0 +1,94 @@
+# Copyright (C) 2019 Nick Filmer (nick80835@gmail.com)
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as
+# published by the Free Software Foundation, either version 3 of the
+# License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+from platform import python_version, uname
+from time import time_ns
+
+from telethon import version
+
+from ubot.micro_bot import micro_bot
+
+ldr = micro_bot.loader
+
+
+#@ldr.add(pattern="reload")
+async def reload_modules(event):
+    await event.edit("`Reloading modules…`")
+
+    errors = ldr.reload_all_modules()
+
+    if errors:
+        await event.edit(errors)
+    else:
+        try:
+            await event.delete()
+        except:
+            pass
+
+
+@ldr.add(pattern="alive")
+async def alive(event):
+    alive_format = "`μBot is running under {0}.\n\n" \
+                   "Version: {1}\n" \
+                   "Telethon: {2}\n" \
+                   "Python: {3}`"
+
+    await event.reply(alive_format.format(uname().node, ldr.botversion, version.__version__, python_version()))
+
+
+#@ldr.add(pattern="shutdown")
+async def shutdown(event):
+    await event.edit("`Goodbye…`")
+    await micro_bot.stop_client()
+
+
+@ldr.add(pattern="ping")
+async def ping(event):
+    start = time_ns()
+    reply = await event.reply("`Ping…`")
+    time_taken_ms = (time_ns() - start) / 1000000
+    await reply.edit(f"`Ping… Pong! -> `**{time_taken_ms}**`ms`")
+
+
+#@ldr.add(pattern="prefix")
+async def change_prefix(event):
+    new_prefix = event.pattern_match.group(1)
+
+    if not new_prefix:
+        await event.edit("`Please specify a valid command prefix!`")
+        return
+
+    micro_bot.settings.set_config("cmd_prefix", new_prefix)
+    errors = ldr.reload_all_modules()
+
+    if errors:
+        await event.edit(f"`Command prefix successfully changed to `**{new_prefix}**` but there were errors:`\n\n{errors}")
+    else:
+        await event.edit(f"`Command prefix successfully changed to `**{new_prefix}**`!`")
+
+
+@ldr.add(pattern="repo")
+async def bot_repo(event):
+    await event.reply("https://github.com/Nick80835/microbot")
+
+
+#@ldr.add(pattern="toggleincoming")
+async def toggleincoming(event):
+    if micro_bot.settings.get_bool("incoming_allowed"):
+        micro_bot.settings.set_config("incoming_allowed", "False")
+        await event.edit("`Successfully disabled incoming commands!`")
+    else:
+        micro_bot.settings.set_config("incoming_allowed", "True")
+        await event.edit("`Successfully enabled incoming commands!`")