Adds sys.audit event for import_module.

Author: Lisa Carrier

Lib/importlib/__init__.py

@@ -85,7 +85,18 @@ def import_module(name, package=None):
             if character != '.':
                 break
             level += 1
-    return _bootstrap._gcd_import(name[level:], package, level)
+    module = _bootstrap._gcd_import(name[level:], package, level)
+    if module:
+        sys.audit(
+            "import",
+            name,
+            # We could try to grab __file__ here but it breaks LazyLoader
+            None,
+            sys.path,
+            sys.meta_path,
+            sys.path_hooks
+        )
+    return module
 
 
 _RELOADING = {}

Lib/test/audit-tests.py

@@ -672,6 +672,22 @@ def hook(event, args):
         assertEqual(event_script_path, tmp_file.name)
         assertEqual(remote_event_script_path, tmp_file.name)
 
+def test_import_module():
+    import importlib
+
+    with TestHook() as hook:
+        importlib.import_module("os")  # random stdlib
+        importlib.import_module("pythoninfo")  # random module
+
+    actual = [a[0] for e, a in hook.seen if e == "import"]
+    assertSequenceEqual(
+        [
+            "os",
+            "pythoninfo",
+        ],
+        actual,
+    )
+
 if __name__ == "__main__":
     from test.support import suppress_msvcrt_asserts
 

Lib/test/test_audit.py

@@ -331,5 +331,8 @@ def test_sys_remote_exec(self):
         if returncode:
             self.fail(stderr)
 
+    def test_import_module(self):
+        self.do_test("test_import_module")
+
 if __name__ == "__main__":
     unittest.main()