Как правильно переопределить операции копирования / deepcopy для объекта в Python?

собирая ответ Алекса Мартелли и комментарий Роба Янга, вы получаете следующий код:

from copy import copy, deepcopy

class A(object):
    def __init__(self):
        print 'init'
        self.v = 10
        self.z = [2,3,4]

    def __copy__(self):
        cls = self.__class__
        result = cls.__new__(cls)
        result.__dict__.update(self.__dict__)
        return result

    def __deepcopy__(self, memo):
        cls = self.__class__
        result = cls.__new__(cls)
        memo[id(self)] = result
        for k, v in self.__dict__.items():
            setattr(result, k, deepcopy(v, memo))
        return result

a = A()
a.v = 11
b1, b2 = copy(a), deepcopy(a)
a.v = 12
a.z.append(5)
print b1.v, b1.z
print b2.v, b2.z

печать

init
11 [2, 3, 4, 5]
11 [2, 3, 4]

здесь __deepcopy__ заполняет memo dict, чтобы избежать избыточного копирования в случае, если сам объект с его членом.

я мог бы быть немного не в курсе специфики, но здесь идет;

С copy docs;

• мелкая копия создает новый составной объект, а затем (насколько это возможно) вставляет в него ссылки на объекты, найденные в оригинале.
• глубокая копия создает новый составной объект, а затем рекурсивно вставляет в него копии объектов, найденных в оригинале.

другими словами: copy() будет копировать только верхний элемент и оставить остальные в качестве указателей в исходную структуру. deepcopy() будет рекурсивно копировать все.

то есть deepcopy() это то, что вам нужно.

Если вам нужно сделать что-то действительно особенное, вы можете переопределить __copy__() или __deepcopy__(), как описано в руководстве. Лично я бы, вероятно, реализовал простую функцию (например,config.copy_config() или такой), чтобы было ясно, что это не стандарт Python поведение.

после отличный ответ Питера, чтобы реализовать пользовательскую deepcopy, с минимальным изменением реализации по умолчанию (например, просто изменение поля, как мне нужно):

class Foo(object):
    def __deepcopy__(self, memo):
        deepcopy_method = self.__deepcopy__
        self.__deepcopy__ = None
        cp = deepcopy(self, memo)
        self.__deepcopy__ = deepcopy_method

        # custom treatments
        # for instance: cp.id = None

        return cp

его не ясно из вашей проблемы, почему вам нужно переопределить эти методы, так как вы не хотите делать какие-либо настройки для методов копирования.

во всяком случае, если вы хотите настроить глубокую копию (например, путем совместного использования некоторых атрибутов и копирования других), вот решение:

from copy import deepcopy


def deepcopy_with_sharing(obj, shared_attribute_names, memo=None):
    '''
    Deepcopy an object, except for a given list of attributes, which should
    be shared between the original object and its copy.

    obj is some object
    shared_attribute_names: A list of strings identifying the attributes that
        should be shared between the original and its copy.
    memo is the dictionary passed into __deepcopy__.  Ignore this argument if
        not calling from within __deepcopy__.
    '''
    assert isinstance(shared_attribute_names, (list, tuple))
    shared_attributes = {k: getattr(obj, k) for k in shared_attribute_names}

    if hasattr(obj, '__deepcopy__'):
        # Do hack to prevent infinite recursion in call to deepcopy
        deepcopy_method = obj.__deepcopy__
        obj.__deepcopy__ = None

    for attr in shared_attribute_names:
        del obj.__dict__[attr]

    clone = deepcopy(obj)

    for attr, val in shared_attributes.iteritems():
        setattr(obj, attr, val)
        setattr(clone, attr, val)

    if hasattr(obj, '__deepcopy__'):
        # Undo hack
        obj.__deepcopy__ = deepcopy_method
        del clone.__deepcopy__

    return clone



class A(object):

    def __init__(self):
        self.copy_me = []
        self.share_me = []

    def __deepcopy__(self, memo):
        return deepcopy_with_sharing(self, shared_attribute_names = ['share_me'], memo=memo)

a = A()
b = deepcopy(a)
assert a.copy_me is not b.copy_me
assert a.share_me is b.share_me

c = deepcopy(b)
assert c.copy_me is not b.copy_me
assert c.share_me is b.share_me

основываясь на чистом ответе Энтони Хэтчкинса, вот моя версия, где рассматриваемый класс происходит от другого пользовательского класса (s. t. нам нужно вызвать super):

class Foo(FooBase):
    def __init__(self, param1, param2):
        self._base_params = [param1, param2]
        super(Foo, result).__init__(*self._base_params)

    def __copy__(self):
        cls = self.__class__
        result = cls.__new__(cls)
        result.__dict__.update(self.__dict__)
        super(Foo, result).__init__(*self._base_params)
        return result

    def __deepcopy__(self, memo):
        cls = self.__class__
        result = cls.__new__(cls)
        memo[id(self)] = result
        for k, v in self.__dict__.items():
            setattr(result, k, copy.deepcopy(v, memo))
        super(Foo, result).__init__(*self._base_params)
        return result

The copy модуль использует evantually __getstate__()/__setstate__()мариновать протоколом, так что это также допустимые цели для переопределения.

реализация по умолчанию просто возвращает и устанавливает __dict__ из класса, так что вам не нужно звонить super() и беспокоиться об умном трюке Эйно Гурдина,выше.