Improved performance of AutoUnload (approx. 35 times faster).

Author: jordanbriere

addons/source-python/packages/source-python/core/__init__.py

@@ -16,6 +16,7 @@
 import hashlib
 #   Inspect
 from inspect import getmodule
+from inspect import getmodulename
 from inspect import currentframe
 #   OS
 from os import sep
@@ -37,6 +38,7 @@
 # Source.Python Imports
 #   Paths
 from paths import GAME_PATH
+from paths import PLUGIN_PATH
 
 
 
@@ -105,11 +107,28 @@ def __new__(cls, *args, **kwargs):
         # Get the class instance
         self = super().__new__(cls)
 
-        # Get the calling module
-        caller = getmodule(currentframe().f_back)
+        # Get the calling frame
+        frame = currentframe().f_back
+
+        # Get the calling path
+        path = frame.f_code.co_filename
+
+        # Don't keep hostage instances that will never be unloaded
+        if not path.startswith(PLUGIN_PATH):
+            return self
+
+        # Resolve the calling module name
+        try:
+            name = frame.f_globals['__name__']
+        except KeyError:
+            try:
+                name = getmodule(frame).__name__
+            except AttributeError:
+                name = getmodulename(path)
 
         # Call class-specific logic for adding the instance.
-        self._add_instance(caller.__name__)
+        if name is not None:
+            self._add_instance(name)
 
         # Return the instance
         return self