pyrofork: Ignore excluded plugins

Signed-off-by: wulan17 <[email protected]>
Mayuri-Chan · Sep 10, 2024 · 732aec9 · 732aec9
1 parent 0f3313a
commit 732aec9
Showing 1 changed file with 15 additions and 35 deletions.
diff --git a/pyrogram/client.py b/pyrogram/client.py
@@ -865,12 +865,24 @@ def load_plugins(self):
             root = plugins["root"]
             include = plugins.get("include", [])
             exclude = plugins.get("exclude", [])
+            exclude_plugins = []
+            if exclude:
+                for i in exclude:
+                    plugin = root.replace(".", "/") + "/" + i.replace(".", "/")
+                    if Path(plugin).is_dir():
+                        for j in sorted(Path(plugin).rglob("*.py")):
+                            exclude_plugins.append(str(j).replace("/", ".")[:-3])
+                    else:
+                        exclude_plugins.append(plugin.replace("/", "."))
 
             count = 0
 
             if not include:
                 for path in sorted(Path(root.replace(".", "/")).rglob("*.py")):
                     module_path = '.'.join(path.parent.parts + (path.stem,))
+                    if exclude_plugins and module_path in exclude_plugins:
+                        log.info(f"[{self.name}] [LOAD] Ignoring excluded plugins {module_path}")
+                        continue
                     module = import_module(module_path)
 
                     for name in vars(module).keys():
@@ -889,6 +901,9 @@ def load_plugins(self):
             else:
                 for path, handlers in include:
                     module_path = root.replace("/",".") + "." + path
+                    if exclude_plugins and module_path in exclude_plugins:
+                        log.info(f"[{self.name}] [LOAD] Ignoring excluded plugins {module_path}")
+                        continue
                     warn_non_existent_functions = True
 
                     try:
@@ -921,41 +936,6 @@ def load_plugins(self):
                                 log.warning('[{}] [LOAD] Ignoring non-existent function "{}" from "{}"'.format(
                                     self.name, name, module_path))
 
-            if exclude:
-                for path, handlers in exclude:
-                    module_path = root.replace("/",".") + "." + path
-                    warn_non_existent_functions = True
-
-                    try:
-                        module = import_module(module_path)
-                    except ImportError:
-                        log.warning('[%s] [UNLOAD] Ignoring non-existent module "%s"', self.name, module_path)
-                        continue
-
-                    if "__path__" in dir(module):
-                        log.warning('[%s] [UNLOAD] Ignoring namespace "%s"', self.name, module_path)
-                        continue
-
-                    if handlers is None:
-                        handlers = vars(module).keys()
-                        warn_non_existent_functions = False
-
-                    for name in handlers:
-                        # noinspection PyBroadException
-                        try:
-                            for handler, group in getattr(module, name).handlers:
-                                if isinstance(handler, Handler) and isinstance(group, int):
-                                    self.remove_handler(handler, group)
-
-                                    log.info('[{}] [UNLOAD] {}("{}") from group {} in "{}"'.format(
-                                        self.name, type(handler).__name__, name, group, module_path))
-
-                                    count -= 1
-                        except Exception:
-                            if warn_non_existent_functions:
-                                log.warning('[{}] [UNLOAD] Ignoring non-existent function "{}" from "{}"'.format(
-                                    self.name, name, module_path))
-
             if count > 0:
                 log.info('[{}] Successfully loaded {} plugin{} from "{}"'.format(
                     self.name, count, "s" if count > 1 else "", root))