Add PyErr_GetExcInfo, make __builtins__ more similar to cpython

from_cpython/Include/pyerrors.h

@@ -83,6 +83,10 @@ PyAPI_FUNC(void) PyErr_Clear(void) PYSTON_NOEXCEPT;
 PyAPI_FUNC(void) PyErr_Fetch(PyObject **, PyObject **, PyObject **) PYSTON_NOEXCEPT;
 PyAPI_FUNC(void) PyErr_Restore(PyObject *, PyObject *, PyObject *) PYSTON_NOEXCEPT;
 
+// Pyton change: This functions are normally only available in CPython >= 3.3 and PyPy but Cython can use them.
+PyAPI_FUNC(void) PyErr_GetExcInfo(PyObject **ptype, PyObject **pvalue, PyObject **ptraceback) PYSTON_NOEXCEPT;
+PyAPI_FUNC(void) PyErr_SetExcInfo(PyObject *type, PyObject *value, PyObject *traceback) PYSTON_NOEXCEPT;
+
 #ifdef Py_DEBUG
 #define _PyErr_OCCURRED() PyErr_Occurred()
 #else

src/codegen/irgen/hooks.cpp

@@ -313,6 +313,8 @@ void compileAndRunModule(AST_Module* m, BoxedModule* bm) {
 
         std::unique_ptr<SourceInfo> si(new SourceInfo(bm, scoping, m, m->body, fn));
         bm->setattr("__doc__", si->getDocString(), NULL);
+        if (!bm->hasattr("__builtins__"))
+            bm->giveAttr("__builtins__", PyModule_GetDict(builtins_module));
 
         CLFunction* cl_f = new CLFunction(0, 0, false, false, std::move(si));
 

src/runtime/capi.cpp

@@ -24,6 +24,7 @@
 #include "llvm/Support/Path.h"
 
 #include "capi/types.h"
+#include "codegen/unwinding.h"
 #include "core/threading.h"
 #include "core/types.h"
 #include "runtime/classobj.h"
@@ -233,8 +234,14 @@ done:
 
 extern "C" PyObject* PyObject_GetAttr(PyObject* o, PyObject* attr_name) noexcept {
     if (!isSubclass(attr_name->cls, str_cls)) {
-        PyErr_Format(PyExc_TypeError, "attribute name must be string, not '%.200s'", Py_TYPE(attr_name)->tp_name);
-        return NULL;
+        if (PyUnicode_Check(attr_name)) {
+            attr_name = _PyUnicode_AsDefaultEncodedString(attr_name, NULL);
+            if (attr_name == NULL)
+                return NULL;
+        } else {
+            PyErr_Format(PyExc_TypeError, "attribute name must be string, not '%.200s'", Py_TYPE(attr_name)->tp_name);
+            return NULL;
+        }
     }
 
     try {
@@ -958,6 +965,20 @@ extern "C" void PyErr_Clear() noexcept {
     PyErr_Restore(NULL, NULL, NULL);
 }
 
+extern "C" void PyErr_GetExcInfo(PyObject** ptype, PyObject** pvalue, PyObject** ptraceback) noexcept {
+    ExcInfo* exc = getFrameExcInfo();
+    *ptype = exc->type;
+    *pvalue = exc->value;
+    *ptraceback = exc->traceback;
+}
+
+extern "C" void PyErr_SetExcInfo(PyObject* type, PyObject* value, PyObject* traceback) noexcept {
+    ExcInfo* exc = getFrameExcInfo();
+    exc->type = type;
+    exc->value = value;
+    exc->traceback = traceback;
+}
+
 extern "C" void PyErr_SetString(PyObject* exception, const char* string) noexcept {
     PyErr_SetObject(exception, boxStrConstant(string));
 }
@@ -1694,6 +1715,10 @@ extern "C" void _Py_FatalError(const char* fmt, const char* function, const char
     abort();
 }
 
+extern "C" PyObject* PyClassMethod_New(PyObject* callable) noexcept {
+    return new BoxedClassmethod(callable);
+}
+
 void setupCAPI() {
     capifunc_cls->giveAttr("__repr__",
                            new BoxedFunction(boxRTFunction((void*)BoxedCApiFunction::__repr__, UNKNOWN, 1)));

src/runtime/objmodel.cpp

@@ -4815,14 +4815,6 @@ extern "C" Box* getGlobal(Box* globals, const std::string* name) {
         static StatCounter stat_builtins("getglobal_builtins");
         stat_builtins.log();
 
-        if ((*name) == "__builtins__") {
-            if (rewriter.get()) {
-                RewriterVar* r_rtn = rewriter->loadConst((intptr_t)builtins_module, rewriter->getReturnDestination());
-                rewriter->commitReturning(r_rtn);
-            }
-            return builtins_module;
-        }
-
         Box* rtn;
         if (rewriter.get()) {
             RewriterVar* builtins = rewriter->loadConst((intptr_t)builtins_module, Location::any());

src/runtime/types.cpp

@@ -2803,6 +2803,8 @@ BoxedModule* createModule(const std::string& name, const char* fn, const char* d
         module->giveAttr("__file__", boxString(fn));
 
     d->d[b_name] = module;
+    if (name == "__main__")
+        module->giveAttr("__builtins__", builtins_module);
     return module;
 }
 

test/tests/builtins.py

@@ -12,6 +12,8 @@ print __builtins__
 __builtins__ = 2
 print __builtins__
 
+import builtins_getitem
+
 print all([]), all([True]), all([False]), all([None]), all([True, False, None])
 print any([]), any([True]), any([False]), any([None]), any([True, False, None])
 

test/tests/builtins_getitem.py

+# this file get's also included by the builtins.py to test that __builtins__ becomes a dict when getting imported
+try:
+    import types
+    print type(__builtins__) == types.ModuleType
+    __builtins__["all"]
+    print "No exception"
+except TypeError as e:
+    print e

test/tests/exec_in_test.py

@@ -178,3 +178,6 @@ print z
 """
 exec s in import_target.__dict__, {}
 exec s in globals(), {}
+
+exec "import builtins_getitem"
+exec "import builtins_getitem" in {}, {}