From ae25eefbf48b9e6897af37903ecd697eb19f337f Mon Sep 17 00:00:00 2001 From: Ned Batchelder Date: Mon, 2 Dec 2019 20:15:32 -0500 Subject: A helper to get code objects --- coverage/backward.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'coverage/backward.py') diff --git a/coverage/backward.py b/coverage/backward.py index 17f04219..37b49167 100644 --- a/coverage/backward.py +++ b/coverage/backward.py @@ -191,6 +191,14 @@ except AttributeError: PYC_MAGIC_NUMBER = imp.get_magic() +def code_object(fn): + """Get the code object from a function.""" + try: + return fn.func_code + except AttributeError: + return fn.__code__ + + try: from types import SimpleNamespace except ImportError: -- cgit v1.2.1