added minimal version checking

Author: itzlayz

teagram/dispatcher.py

@@ -27,7 +27,9 @@ async def check_filter(self, function: FunctionType, message: Message):
             if not coroutine:
                 return False
         else:
-            return message.outgoing
+            return message.outgoing or (
+                message.from_user and message.from_user.id == self.client.me.id
+            )
 
         return True
 

teagram/loader.py

@@ -4,6 +4,7 @@
 import sys
 import gc
 import os
+import re
 
 from importlib.machinery import ModuleSpec
 from importlib.util import spec_from_file_location, module_from_spec
@@ -14,9 +15,16 @@
 from typing import Final, List
 
 from .utils import BASE_PATH
+from . import __version__
 
 from .dispatcher import Dispatcher
-from .types import Module, StringLoader, ModuleException, ABCLoader
+from .types import (
+    Module,
+    StringLoader,
+    ModuleException,
+    ModuleVersionException,
+    ABCLoader,
+)
 
 from .inline import InlineDispatcher
 from .translator import Translator, ModuleTranslator
@@ -111,6 +119,9 @@ def __init__(self, client, database):
 
         self.translator = Translator(self.database)
 
+    def get(self, key: str):
+        return self.translator.get("loader", key)
+
     async def load(self):
         await self.load_modules()
 
@@ -173,11 +184,34 @@ async def load_module(
         )
 
         if not module_class:
+            sys.modules.pop(module_name, None)
+
             raise ModuleException("❌ Module class not found")
 
         module_class.__origin__ = origin
         name = getattr(module_class, "name", module_class.__class__.__name__)
 
+        min_version = module_class.MIN_VERSION
+        if min_version != "BETA":
+            if min_version != "Not specified":
+                exception = ""
+                if not re.fullmatch(r"\d+\.\d+\.\d+", min_version):
+                    exception = self.get("invalid_module_min_version").format(
+                        min_version
+                    )
+
+                current_version = tuple(map(int, __version__.split(".")))
+                required_version = tuple(map(int, min_version.split(".")))
+
+                if current_version < required_version:
+                    exception = self.get("incompatible_version").format(
+                        __version__, min_version
+                    )
+
+                if exception:
+                    sys.modules.pop(module_name, None)
+                    raise ModuleVersionException(exception)
+
         if self.lookup(name):
             raise ModuleException(f"❌ Module {name} has already loaded")
 

teagram/modules/manager.py

@@ -3,7 +3,7 @@
 from .. import loader, utils
 
 from ..translator import SUPPORTED_LANGUAGES
-from ..types import ModuleException
+from ..types import ModuleException, ModuleVersionException
 
 from time import time
 
@@ -179,8 +179,8 @@ async def loadmod(self, message: Message):
         try:
             module_name = await self.load_module(code)
             await utils.answer(message, self.get("load_success").format(module_name))
-        except ModuleException as error:
-            return await utils.answer(message, f"<b>{error}</b>")
+        except (ModuleException, ModuleVersionException) as error:
+            return await utils.answer(message, error)
 
     @loader.command(alias="ulm")
     async def unloadmod(self, message, args):

teagram/translations/en.yaml

@@ -26,4 +26,9 @@ help:
   modules: "<b>Teagram modules</b>\n\n"
 
 info:
-  ping: "<b>⚡ Ping: {}ms</b>"
+  ping: "<b>⚡ Ping: {}ms</b>"
+
+loader:
+  module_class_not_found: "<b>❌ Module's class not found</b>"
+  invalid_module_min_version: "<b>❌ Invalid module's minimal version (<code>{}</code>)</b>"
+  incompatible_version: "<b>❌ Userbot's version is lower than required ({} < {})</b>"

teagram/translations/ru.yaml

@@ -26,4 +26,9 @@ help:
   modules: "<b>Teagram модули</b>\n\n"
 
 info:
-  ping: "<b>⚡ Пинг: {}ms</b>"
+  ping: "<b>⚡ Пинг: {}ms</b>"
+
+loader:
+  module_class_not_found: "<b>❌ Класс модуля не найден</b>"
+  invalid_module_min_version: "<b>❌ Некорректная минимальная версия модуля (<code>{}</code>)</b>"
+  incompatible_version: "<b>❌ Версия юзербота ниже минимальной версии модуля ({} < {})</b>"

teagram/types.py

@@ -57,6 +57,10 @@ class ModuleException(Exception):
     pass
 
 
+class ModuleVersionException(Exception):
+    pass
+
+
 class Module:
     MIN_VERSION = "BETA"
     MODULE_VERSION = "Not specified"