A helper to get code objects

2 files changed, 10 insertions, 5 deletions

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:
diff --git a/coverage/inorout.py b/coverage/inorout.py
index 2a19ba69..0eea62ae 100644
--- a/coverage/inorout.py
+++ b/coverage/inorout.py
@@ -14,6 +14,7 @@ import sys
 import traceback
 
 from coverage import env
+from coverage.backward import code_object
 from coverage.disposition import FileDisposition, disposition_init
 from coverage.files import TreeMatcher, FnmatchMatcher, ModuleMatcher
 from coverage.files import prep_patterns, find_python_files, canonical_filename
@@ -157,11 +158,7 @@ class InOrOut(object):
                 # objects still have the file names.  So dig into one to find
                 # the path to exclude.  The "filename" might be synthetic,
                 # don't be fooled by those.
-                structseq_new = _structseq.structseq_new
-                try:
-                    structseq_file = structseq_new.func_code.co_filename
-                except AttributeError:
-                    structseq_file = structseq_new.__code__.co_filename
+                structseq_file = code_object(_structseq.structseq_new).co_filename
                 if not structseq_file.startswith("<"):
                     self.pylib_paths.add(canonical_path(structseq_file))