Implement C API function PyUnicode_AsEncodedString.

Author: fangerer

graalpython/com.oracle.graal.python.cext/src/unicodeobject.c

@@ -497,3 +497,8 @@ UPCALL_ID(PyUnicode_Tailmatch);
 Py_ssize_t PyUnicode_Tailmatch(PyObject *str, PyObject *substr, Py_ssize_t start, Py_ssize_t end, int direction) {
 	return UPCALL_CEXT_L(_jls_PyUnicode_Tailmatch, native_to_java(str), native_to_java(substr), start, end, direction);
 }
+
+UPCALL_ID(PyUnicode_AsEncodedString);
+PyObject * PyUnicode_AsEncodedString(PyObject *unicode, const char *encoding, const char *errors) {
+	return UPCALL_CEXT_O(_jls_PyUnicode_AsEncodedString, native_to_java(unicode), polyglot_from_string(encoding, SRC_CS), polyglot_from_string(errors, SRC_CS));
+}

graalpython/com.oracle.graal.python.test/src/tests/cpyext/test_unicode.py

@@ -495,3 +495,21 @@ def compile_module(self, name):
         arguments=["PyObject* str"],
         cmpfunc=unhandled_error_compare
     )
+
+
+    test_PyUnicode_AsEncodedString = CPyExtFunction(
+        lambda args: args[0].encode(args[1], args[2]),
+        lambda: (
+            ("abcd", "ascii", "report"),
+            ("abcd", "utf8", "report"),
+            ("öüä", "ascii", "report"),
+            ("öüä", "utf8", "report"),
+            ("öüä", "ascii", "ignore"),
+            ("öüä", "ascii", "replace"),
+        ),
+        resultspec="O",
+        argspec='Oss',
+        arguments=["PyObject* str", "const char* encoding", "const char* errors"],
+        cmpfunc=unhandled_error_compare
+    )
+

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

@@ -57,7 +57,6 @@
 import java.nio.charset.CoderResult;
 import java.nio.charset.CodingErrorAction;
 import java.nio.charset.StandardCharsets;
-import java.nio.charset.UnsupportedCharsetException;
 import java.text.DecimalFormat;
 import java.text.ParseException;
 import java.text.ParsePosition;
@@ -2635,9 +2634,9 @@ abstract static class PyUnicode_Decode extends NativeUnicodeBuiltin {
 
         @Specialization
         Object doDecode(VirtualFrame frame, Object module, Object cByteArray, long size, String encoding, String errors,
-                            @Cached CExtNodes.ToSulongNode toSulongNode,
-                            @Cached GetByteArrayNode getByteArrayNode,
-                            @Cached GetNativeNullNode getNativeNullNode) {
+                        @Cached CExtNodes.ToSulongNode toSulongNode,
+                        @Cached GetByteArrayNode getByteArrayNode,
+                        @Cached GetNativeNullNode getNativeNullNode) {
 
             try {
                 ByteBuffer inputBuffer = wrap(getByteArrayNode.execute(frame, cByteArray, size));

graalpython/lib-graalpython/python_cext.py

@@ -784,6 +784,11 @@ def PyUnicode_Tailmatch(s, substr, start, end, direction):
     return 1 if s[start:end].startswith(substr) else 0
 
 
+@may_raise
+def PyUnicode_AsEncodedString(s, encoding, errors):
+    return s.encode(encoding, errors)
+
+
 ##################### CAPSULE