From b41936b50900b94ff4d76c1fa5dbe5055908c515 Mon Sep 17 00:00:00 2001
From: wulan17 <wulan17@nusantararom.org>
Date: Fri, 20 Sep 2024 18:10:56 +0700
Subject: [PATCH] pyrofork: Ignore excluded plugins [2/2]

Signed-off-by: wulan17 <wulan17@nusantararom.org>
---
 pyrogram/client.py | 79 +++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 74 insertions(+), 5 deletions(-)

diff --git a/pyrogram/client.py b/pyrogram/client.py
index 29873334..1a183738 100644
--- a/pyrogram/client.py
+++ b/pyrogram/client.py
@@ -848,6 +848,12 @@ async def load_session(self):
                         except Exception as e:
                             print(e)
 
+    def is_excluded(self, exclude, module):
+        for e in exclude:
+            if module == e or module.startswith(e + '.'):
+                return True
+        return False
+
     def load_plugins(self):
         if self.plugins:
             plugins = self.plugins.copy()
@@ -952,7 +958,7 @@ def load_plugins(self):
             else:
                 for path, handlers in include:
                     module_path = root.replace("/", ".") + "." + path
-                    if module_path in exclude_plugins:
+                    if self.is_excluded(exclude_plugins, module_path):
                         log.warning(
                             '[%s] [LOAD] Ignoring namespace "%s"', self.name, module_path
                         )
@@ -971,10 +977,73 @@ def load_plugins(self):
                         continue
 
                     if "__path__" in dir(module):
-                        log.warning(
-                            '[%s] [LOAD] Ignoring namespace "%s"', self.name, module_path
-                        )
-                        continue
+                        for current_root, _, filenames in os.walk(module.replace(".", "/")):
+                            namespace = current_root.replace("/", ".")
+                            if "__pycache__" in namespace:
+                                continue
+                            if namespace in exclude_plugins:
+                                log.warning(
+                                    '[%s] [LOAD] Ignoring namespace "%s"', self.name, namespace
+                                )
+                                continue
+                            else:
+                                for filename in filenames:
+                                    if filename.endswith(".py"):
+                                        module_path = namespace + "." + filename[:-3]
+                                        if module_path in exclude_plugins:
+                                            log.warning(
+                                                '[%s] [LOAD] Ignoring namespace "%s"',
+                                                self.name,
+                                                module_path,
+                                            )
+                                            continue
+                                        else:
+                                            module = import_module(module_path)
+
+                                            for name in vars(module).keys():
+
+                                                # noinspection PyBroadException
+                                                try:
+                                                    for handler, group in getattr(
+                                                        module, name
+                                                    ).handlers:
+                                                        if isinstance(
+                                                            handler, Handler
+                                                        ) and isinstance(group, int):
+
+                                                            if (
+                                                                module_path in exclude_handlers
+                                                                and name
+                                                                in exclude_handlers[module_path]
+                                                            ):
+                                                                exclude_handlers[
+                                                                    module_path
+                                                                ].remove(name)
+                                                                log.warning(
+                                                                    '[{}] [LOAD] Ignoring function "{}" from group {} in "{}"'.format(
+                                                                        self.name,
+                                                                        name,
+                                                                        group,
+                                                                        module_path,
+                                                                    )
+                                                                )
+                                                                continue
+
+                                                            self.add_handler(handler, group)
+
+                                                            log.info(
+                                                                '[{}] [LOAD] {}("{}") in group {} from "{}"'.format(
+                                                                    self.name,
+                                                                    type(handler).__name__,
+                                                                    name,
+                                                                    group,
+                                                                    module_path,
+                                                                )
+                                                            )
+
+                                                            count += 1
+                                                except Exception as e:
+                                                    pass
 
                     if handlers is None:
                         handlers = vars(module).keys()