Fix plugin modules not being properly reloaded from disk

delivrance · delivrance · commit ac8fad3a1868 · 2020-01-16T22:01:10.000+01:00
When using importlib.import_module, Python loads the module from disk
only once and any subsequent call to this method will just re-import
the already loaded module from RAM. Wrapping importlib.import_module
with importlib.reload will make Python force-reload the module from
disk.
diff --git a/pyrogram/client/client.py b/pyrogram/client/client.py
@@ -26,7 +26,7 @@
 import time
 from configparser import ConfigParser
 from hashlib import sha256, md5
-from importlib import import_module
+from importlib import import_module, reload
 from pathlib import Path
 from signal import signal, SIGINT, SIGTERM, SIGABRT
 from threading import Thread
@@ -1531,7 +1531,7 @@ def load_plugins(self):
             if not include:
                 for path in sorted(Path(root).rglob("*.py")):
                     module_path = '.'.join(path.parent.parts + (path.stem,))
-                    module = import_module(module_path)
+                    module = reload(import_module(module_path))
 
                     for name in vars(module).keys():
                         # noinspection PyBroadException
@@ -1553,7 +1553,7 @@ def load_plugins(self):
                     warn_non_existent_functions = True
 
                     try:
-                        module = import_module(module_path)
+                        module = reload(import_module(module_path))
                     except ImportError:
                         log.warning('[{}] [LOAD] Ignoring non-existent module "{}"'.format(
                             self.session_name, module_path))
@@ -1591,7 +1591,7 @@ def load_plugins(self):
                     warn_non_existent_functions = True
 
                     try:
-                        module = import_module(module_path)
+                        module = reload(import_module(module_path))
                     except ImportError:
                         log.warning('[{}] [UNLOAD] Ignoring non-existent module "{}"'.format(
                             self.session_name, module_path))
