forked from VimPlug/jedi
255186376e
Jedi passes pickles to subprocesses which are running the target version of Python and thus may not be the same as the version under which Jedi itself is running. In Python 3.13, pathlib is being refactored to allow for easier extension and has thus moved most of its internal implementation to a submodule. Unfortunately this changes the paths of the symbols, causing pickles of those types to fail to load in earlier versions of Python. This commit introduces a custom unpickler which accounts for this move, allowing bi-directional passing of pickles to work.
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
"""
|
|
This module is here to ensure compatibility of Windows/Linux/MacOS and
|
|
different Python versions.
|
|
"""
|
|
import errno
|
|
import sys
|
|
import pickle
|
|
from typing import Any
|
|
|
|
|
|
class Unpickler(pickle.Unpickler):
|
|
def find_class(self, module: str, name: str) -> Any:
|
|
# Python 3.13 moved pathlib implementation out of __init__.py as part of
|
|
# generalising its implementation. Ensure that we support loading
|
|
# pickles from 3.13 on older version of Python. Since 3.13 maintained a
|
|
# compatible API, pickles from older Python work natively on the newer
|
|
# version.
|
|
if module == 'pathlib._local':
|
|
module = 'pathlib'
|
|
return super().find_class(module, name)
|
|
|
|
|
|
def pickle_load(file):
|
|
try:
|
|
return Unpickler(file).load()
|
|
# Python on Windows don't throw EOF errors for pipes. So reraise them with
|
|
# the correct type, which is caught upwards.
|
|
except OSError:
|
|
if sys.platform == 'win32':
|
|
raise EOFError()
|
|
raise
|
|
|
|
|
|
def pickle_dump(data, file, protocol):
|
|
try:
|
|
pickle.dump(data, file, protocol)
|
|
# On Python 3.3 flush throws sometimes an error even though the writing
|
|
# operation should be completed.
|
|
file.flush()
|
|
# Python on Windows don't throw EPIPE errors for pipes. So reraise them with
|
|
# the correct type and error number.
|
|
except OSError:
|
|
if sys.platform == 'win32':
|
|
raise IOError(errno.EPIPE, "Broken pipe")
|
|
raise
|