gh-128118: Speed up copy.copy with fast lookup for atomic and container types (#128119)

This commit is contained in:
Pieter Eendebak 2024-12-30 18:18:42 +01:00 committed by GitHub
parent 3480124321
commit 34b85ef26c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 11 additions and 18 deletions

View file

@ -67,13 +67,15 @@ def copy(x):
cls = type(x)
copier = _copy_dispatch.get(cls)
if copier:
return copier(x)
if cls in _copy_atomic_types:
return x
if cls in _copy_builtin_containers:
return cls.copy(x)
if issubclass(cls, type):
# treat it as a regular class:
return _copy_immutable(x)
return x
copier = getattr(cls, "__copy__", None)
if copier is not None:
@ -98,23 +100,12 @@ def copy(x):
return _reconstruct(x, None, *rv)
_copy_dispatch = d = {}
def _copy_immutable(x):
return x
for t in (types.NoneType, int, float, bool, complex, str, tuple,
_copy_atomic_types = {types.NoneType, int, float, bool, complex, str, tuple,
bytes, frozenset, type, range, slice, property,
types.BuiltinFunctionType, types.EllipsisType,
types.NotImplementedType, types.FunctionType, types.CodeType,
weakref.ref, super):
d[t] = _copy_immutable
d[list] = list.copy
d[dict] = dict.copy
d[set] = set.copy
d[bytearray] = bytearray.copy
del d, t
weakref.ref, super}
_copy_builtin_containers = {list, dict, set, bytearray}
def deepcopy(x, memo=None, _nil=[]):
"""Deep copy operation on arbitrary Python objects.

View file

@ -0,0 +1,2 @@
Improve performance of :func:`copy.copy` by 30% via
a fast path for atomic types and container types.