_imp builtin module: add missing _fix_co_filename builtin

Author: cosminbasca

graalpython/com.oracle.graal.python.test/src/tests/test_imports.py

@@ -37,24 +37,25 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
-from posix import *
 import sys
+from posix import *
+
 sys.path.append(sys._getframe().f_code.co_filename.rpartition("/")[0])
 
 def test_relative_import():
-    import package
+    pass
 
 
 def test_dotted_import():
-    import package.moduleY
+    pass
 
 
 def test_recursive_import():
-    import package.moduleRecursive
+    pass
 
 
 def test_recursive_import2():
-    import package.moduleRecursive2
+    pass
 
 
 def test_import_star_has_to_be_module():
@@ -91,3 +92,15 @@ def test_import_error():
 def test_import_some_star():
     import posix
     assert stat == posix.stat
+
+
+def test_imp_fix_co_filename():
+    import _imp
+
+    def func(x):
+        return x+x
+
+    code = func.__code__
+    old_name = code.co_filename
+    _imp._fix_co_filename(code, old_name + '_more_path')
+    assert code.co_filename == old_name + '_more_path'

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/modules/ImpModuleBuiltins.java

@@ -57,6 +57,7 @@
 import com.oracle.graal.python.builtins.modules.TruffleCextBuiltins.CheckFunctionResultNode;
 import com.oracle.graal.python.builtins.objects.PNone;
 import com.oracle.graal.python.builtins.objects.cext.CExtNodes.AsPythonObjectNode;
+import com.oracle.graal.python.builtins.objects.code.PCode;
 import com.oracle.graal.python.builtins.objects.common.HashingCollectionNodes.SetItemNode;
 import com.oracle.graal.python.builtins.objects.common.HashingStorageNodes;
 import com.oracle.graal.python.builtins.objects.dict.PDict;
@@ -69,6 +70,7 @@
 import com.oracle.graal.python.nodes.call.special.CallUnaryMethodNode;
 import com.oracle.graal.python.nodes.function.PythonBuiltinBaseNode;
 import com.oracle.graal.python.nodes.function.PythonBuiltinNode;
+import com.oracle.graal.python.nodes.function.builtins.PythonBinaryBuiltinNode;
 import com.oracle.graal.python.runtime.PythonContext;
 import com.oracle.graal.python.runtime.PythonCore;
 import com.oracle.graal.python.runtime.PythonOptions;
@@ -385,4 +387,19 @@ public Object run(String path, PythonModule mod) {
         }
     }
 
+    @Builtin(name = "_fix_co_filename", fixedNumOfPositionalArgs = 2)
+    @GenerateNodeFactory
+    public abstract static class FixCoFilename extends PythonBinaryBuiltinNode {
+        @Specialization
+        public Object run(PCode code, PString path) {
+            code.setFilename(path.getValue());
+            return PNone.NONE;
+        }
+
+        @Specialization
+        public Object run(PCode code, String path) {
+            code.setFilename(path);
+            return PNone.NONE;
+        }
+    }
 }

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/code/PCode.java

@@ -313,6 +313,10 @@ public Object[] getCellVars() {
         return cellvars;
     }
 
+    public void setFilename(String filename) {
+        this.filename = filename;
+    }
+
     public String getFilename() {
         if (filename == null && rootNode != null) {
             filename = extractFileName(rootNode);