Как правильно сортировать экземпляр namedtuple

Question

Я изучаю, как использовать рассол. Я создал объект namedtuple, добавил его в список и попытался рассортировать этот список. Тем не менее, я получаю следующее сообщение об ошибке:

pickle.PicklingError: Can't pickle <class '__main__.P'>: it's not found as __main__.P 

Я обнаружил, что если я побежал код без окружив его внутри функции, она прекрасно работает. Есть ли дополнительный шаг, необходимый для рассортирования объекта при обертывании внутри функции?

Вот мой код:

from collections import namedtuple 
import pickle 

def pickle_test(): 
    P = namedtuple("P", "one two three four") 
    my_list = [] 
    abe = P("abraham", "lincoln", "vampire", "hunter") 
    my_list.append(abe) 
    f = open('abe.pickle', 'w') 
    pickle.dump(abe, f) 
    f.close() 

pickle_test() 

Answer 1

Создать именованный кортеж вне функции:

from collections import namedtuple 
import pickle 

P = namedtuple("P", "one two three four") 

def pickle_test(): 
    my_list = [] 
    abe = P("abraham", "lincoln", "vampire", "hunter") 
    my_list.append(abe) 
    f = open('abe.pickle', 'w') 
    pickle.dump(abe, f) 
    f.close() 

pickle_test() 

Теперь pickle можете найти его; теперь это модуль глобальный. При распаковке все модули pickle должны найти снова __main__.P. В вашей версии P является местным, к функции pickle_test(), и это не является интроспективным или импортируемым.

Важно помнить, что namedtuple() является фабрикой класса; вы даете ему параметры и возвращает объект класса для создания экземпляров. pickle хранит только данные , содержащиеся в экземплярах, а также ссылку на исходный класс для восстановления экземпляров.

Answer 2

После того, как я добавил свой вопрос в качестве комментария к основному ответу, я нашел способ решить проблему создания динамически созданного namedtuple разборки. Это необходимо в моем случае, потому что я обнаруживаю поля во время выполнения (после запроса БД).

Все, что я сделать, это обезьяны патчnamedtuple, эффективно перемещая его __main__ модуля:

def _CreateNamedOnMain(*args): 
    import __main__ 
    namedtupleClass = collections.namedtuple(*args) 
    setattr(__main__, namedtupleClass.__name__, namedtupleClass) 
    namedtupleClass.__module__ = "__main__" 
    return namedtupleClass 

По помнить, что namedtuple имя (которое предусмотрено args) может перезаписывать другой член в __main__ если вы не осторожны.

Answer 3

В качестве альтернативы, вы можете использовать cloudpickle или dill для сериализации:

from collections import namedtuple 

import cloudpickle 
import dill 



def dill_test(dynamic_names): 
    P = namedtuple('P', dynamic_names) 
    my_list = [] 
    abe = P("abraham", "lincoln", "vampire", "hunter") 
    my_list.append(abe) 
    with open('deleteme.cloudpickle', 'wb') as f: 
     cloudpickle.dump(abe, f) 
    with open('deleteme.dill', 'wb') as f: 
     dill.dump(abe, f) 


dill_test("one two three four") 

Answer 4

Я нашел this answer в другом потоке. Это все об именовании названного кортежа. Это сработало для меня:

group_t =   namedtuple('group_t', 'field1, field2') # this will work 
mismatched_group_t = namedtuple('group_t', 'field1, field2') # this will throw the error