commit c75b39fc4de8e181a030aecbf8648284294fcba6
author Joshua Haberman <jhaberman@gmail.com> Sun Jan 09 21:48:26 2022 -0800
committer Joshua Haberman <jhaberman@gmail.com> Sun Jan 09 21:48:26 2022 -0800
tree 2ad15c9fb572110eb7b7a6fb05c3013c4b7fb2fc
parent 61d6eff5c4cb4e80b00423b59dcf0a72bfe7a1f5

Accept bytes object as field name.

python/message.c

diff --git a/python/message.c b/python/message.c
index 46212ac..02fb902 100644
--- a/python/message.c
+++ b/python/message.c
@@ -243,7 +243,12 @@
                                       PyObject* exc_type) {
   assert(f || o);
   Py_ssize_t size;
-  const char* name = PyUnicode_AsUTF8AndSize(py_name, &size);
+  const char* name = NULL;
+  if (PyUnicode_Check(py_name)) {
+    name = PyUnicode_AsUTF8AndSize(py_name, &size);
+  } else if (PyBytes_Check(py_name)) {
+    PyBytes_AsStringAndSize(py_name, (char**)&name, &size);
+  }
   if (!name) return NULL;
   const upb_msgdef* msgdef = _PyUpb_CMessage_GetMsgdef(self);