support NEWOBJ_EX

examples/objex.py

@@ -0,0 +1,11 @@
+class MyClass:
+    def __getnewargs_ex__(self):
+        return ('arg1', 345), {'base': 16, "test": True}
+
+import pickle
+
+x = MyClass()
+x.abc = 666
+
+filehandler = open(b"objex.pkl", "wb")
+pickle.dump(x, filehandler)

src/parser.ts

@@ -322,6 +322,17 @@ export class Parser {
                     stack.push(obj);
                     break;
                 }
+                case OP.NEWOBJ_EX: {
+                    const kwargs = stack.pop();
+                    const args = stack.pop();
+                    const cls = stack.pop();
+                    const obj = this.newObject(cls, ...args);
+                    if (obj.__setnewargs_ex__) {
+                        obj.__setnewargs_ex__(kwargs);
+                    }
+                    stack.push(obj);
+                    break;
+                }
                 case OP.PERSID:
                     stack.push(this.persistent_load(reader.line()));
                     break;

src/registry.ts

@@ -27,6 +27,14 @@ export class Registry extends Map<string, new (...args: any[]) => any> {
         } as unknown as new (...args: any[]) => any;
         PObject.prototype.module = module;
         PObject.prototype.name = name;
+        PObject.prototype.__setnewargs_ex__ = function (kwargs: any) {
+            Object.defineProperty(this, 'kwargs', {
+                value: kwargs,
+                enumerable: false,
+                configurable: false,
+                writable: false,
+            });
+        };
         return PObject;
     }
 }