Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-10-19 19:10:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

support object pickle

Browse Source
This commit is contained in:
blueloveTH 2024-12-16 17:53:29 +08:00
parent 1b4902dbd3
commit 58b5455871
2 changed files with 27 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/modules/pickle.c
View File

@ -345,9 +345,12 @@ static bool pkl__write_object(PickleObject* buf, py_TValue* obj) {
buf->used_types[obj->type] = true;
return true;
}
// try memo for `is_ptr=true` objects
if(pkl__try_memo(buf, obj->_obj)) return true;
py_TypeInfo* ti = pk__type_info(obj->type);
py_Ref f_reduce = py_tpfindmagic(obj->type, __reduce__);
if(!py_isnil(f_reduce)) {
if(f_reduce != NULL) {
if(!py_call(f_reduce, 1, obj)) return false;
// expected: (callable, args)
py_Ref reduced = py_retval();
@ -364,9 +367,18 @@ static bool pkl__write_object(PickleObject* buf, py_TValue* obj) {
}
pkl__emit_op(buf, PKL_CALL);
pkl__emit_int(buf, args_length);
// store memo
pkl__store_memo(buf, obj->_obj);
return true;
}
if(ti->is_python) {
pkl__emit_op(buf, PKL_OBJECT);
pkl__emit_int(buf, obj->type);
buf->used_types[obj->type] = true;
// store memo
pkl__store_memo(buf, obj->_obj);
return true;
}
if(ti->is_python) { return true; }
return TypeError("'%t' object is not picklable", obj->type);
}
}
@ -648,7 +660,10 @@ bool py_pickle_loads_body(const unsigned char* p, int memo_length, c11_smallmap_
break;
}
case PKL_OBJECT: {
c11__abort("PKL_OBJECT is not implemented");
py_Type type = (py_Type)pkl__read_int(&p);
type = pkl__fix_type(type, type_mapping);
if(!py_tpcall(type, 0, NULL)) return false;
py_push(py_retval());
break;
}
case PKL_EOF: {

10
tests/90_pickle.py
View File

@ -119,7 +119,15 @@ class A:
print('__reduce__() called')
return A, (self.seed,)
test(A(1))
test([A(1)]*10)
class Simple:
def __init__(self): pass
def __eq__(self, other): return True
def __ne__(self, other): return False
test(Simple())
test([Simple()]*10)
exit()