Update autoreload.py

jaysarva · web-flow · commit 5790fcc6506f · 2024-08-19T16:42:11.000-07:00
diff --git a/IPython/extensions/autoreload.py b/IPython/extensions/autoreload.py
@@ -107,7 +107,7 @@
 
 from IPython.core import magic_arguments
 from IPython.core.magic import Magics, magics_class, line_magic
-
+from IPython.extensions.deduperreload.deduperreload import DeduperReloader
 __skip_doctest__ = True
 
 # -----------------------------------------------------------------------------
@@ -167,6 +167,9 @@ def __init__(self, shell=None):
         # Reporting callable for verbosity
         self._report = lambda msg: None  # by default, be quiet.
 
+        # Deduper reloader
+        self.deduper_reloader = DeduperReloader()
+
         # Cache module modification times
         self.check(check_all=True, do_reload=False)
 
@@ -233,7 +236,7 @@ def filename_and_mtime(self, module):
 
         return py_filename, pymtime
 
-    def check(self, check_all=False, do_reload=True):
+    def check(self, check_all=False, do_reload=True, use_deduper_reload=True):
         """Check whether some modules need to be reloaded."""
 
         if not self.enabled and not check_all:
@@ -270,7 +273,9 @@ def check(self, check_all=False, do_reload=True):
             if do_reload:
                 self._report(f"Reloading '{modname}'.")
                 try:
-                    if self.autoload_obj:
+                    if use_deduper_reload and self.deduper_reloader.maybe_reload_module(m):
+                        pass
+                    elif self.autoload_obj:
                         superreload(m, reload, self.old_objects, self.shell)
                     else:
                         superreload(m, reload, self.old_objects)
@@ -285,6 +290,8 @@ def check(self, check_all=False, do_reload=True):
                             file=sys.stderr,
                         )
                     self.failed[py_filename] = pymtime
+        if use_deduper_reload:
+            self.deduper_reloader.update_sources()
 
 
 # ------------------------------------------------------------------------------
