pythongh-117500: Store interactive source code in an alternative location in linecache

Author: pablogsal

Lib/linecache.py

@@ -14,6 +14,7 @@
 # The cache. Maps filenames to either a thunk which will provide source code,
 # or a tuple (size, mtime, lines, fullname) once loaded.
 cache = {}
+_interactive_cache = {}
 
 
 def clearcache():
@@ -35,10 +36,11 @@ def getlines(filename, module_globals=None):
     """Get the lines for a Python source file from the cache.
     Update the cache if it doesn't contain an entry for this file already."""
 
-    if filename in cache:
-        entry = cache[filename]
-        if len(entry) != 1:
-            return cache[filename][2]
+    for the_cache in (cache, _interactive_cache):
+        if filename in the_cache:
+            entry = the_cache[filename]
+            if len(entry) != 1:
+                return the_cache[filename][2]
 
     try:
         return updatecache(filename, module_globals)
@@ -159,11 +161,12 @@ def lazycache(filename, module_globals):
         get_source method must be found, the filename must be a cacheable
         filename, and the filename must not be already cached.
     """
-    if filename in cache:
-        if len(cache[filename]) == 1:
-            return True
-        else:
-            return False
+    for the_cache in (cache, _interactive_cache):
+        if filename in the_cache:
+            if len(the_cache[filename]) == 1:
+                return True
+            else:
+                return False
     if not filename or (filename.startswith('<') and filename.endswith('>')):
         return False
     # Try for a __loader__, if available
@@ -183,8 +186,8 @@ def get_lines(name=name, *args, **kwargs):
     return False
 
 
-def _register_code(code, string, name):
-    cache[code] = (
+def _register_code(code, filename, string, name):
+    _interactive_cache[filename] = (
             len(string),
             None,
             [line + '\n' for line in string.splitlines()],

Python/pythonrun.c

@@ -1351,7 +1351,8 @@ run_mod(mod_ty mod, PyObject *filename, PyObject *globals, PyObject *locals,
         }
 
         PyObject* result = PyObject_CallFunction(
-            print_tb_func, "OOO",
+            print_tb_func, "OOOO",
+            co,
             interactive_filename,
             interactive_src,
             filename