Травление статического метода в Python - PullRequest
Новая
       11

Травление статического метода в Python

2 голосов
/ 16 декабря 2009

Я пытался выбрать объект, который содержит ссылки на методы статического класса. Pickle завершается неудачно (например, на module.MyClass.foo), заявляя, что его невозможно засолить, так как module.foo не существует.
Я пришел к следующему решению, используя объект-оболочку для определения местоположения функции при вызове, сохраняя класс контейнера и имя функции: 

class PicklableStaticMethod(object):
    """Picklable version of a static method.
    Typical usage:
        class MyClass:
            @staticmethod
            def doit():
                print "done"
        # This cannot be pickled:
        non_picklable = MyClass.doit
        # This can be pickled:
        picklable = PicklableStaticMethod(MyClass.doit, MyClass)
    """
    def __init__(self, func, parent_class):
        self.func_name = func.func_name
        self.parent_class = parent_class
    def __call__(self, *args, **kwargs):
        func = getattr(self.parent_class, self.func_name)
        return func(*args, **kwargs)

Мне вот интересно, есть ли лучший - более стандартный способ - засолить такой объект? Я не хочу вносить изменения в глобальный процесс pickle (используя, например, copy_reg), но следующий шаблон был бы полезен: Класс MyClass (объект): @picklable_staticmethod def foo (): печать "готово".

Мои попытки не увенчались успехом, особенно потому, что я не смог извлечь класс владельца из функции foo. Я даже был готов согласиться на явную спецификацию (например, @picklable_staticmethod(MyClass)), но я не знаю, как можно ссылаться на класс MyClass там, где он определяется.

Любые идеи будут великолепны!

Йонатан

Ответы [ 2 ]

5 голосов
/ 16 декабря 2009

Это похоже на работу. 

class PickleableStaticMethod(object):
    def __init__(self, fn, cls=None):
        self.cls = cls
        self.fn = fn
    def __call__(self, *args, **kwargs):
        return self.fn(*args, **kwargs)
    def __get__(self, obj, cls):
        return PickleableStaticMethod(self.fn, cls)
    def __getstate__(self):
        return (self.cls, self.fn.__name__)
    def __setstate__(self, state):
        self.cls, name = state
        self.fn = getattr(self.cls, name).fn

Хитрость заключается в том, чтобы поймать класс, когда от него получен статический метод.

Альтернативы: Вы можете использовать метаклассирование, чтобы дать всем вашим статическим методам атрибут .__parentclass__. Затем вы могли бы создать подкласс Pickler и дать каждому экземпляру подкласса свою собственную таблицу .dispatch, которую затем можно изменить, не затрагивая глобальную таблицу диспетчеризации (Pickler.dispatch). Защипывание, расслоение и вызов метода могут быть немного быстрее.

0 голосов
/ 16 декабря 2009

РЕДАКТИРОВАТЬ: изменено после комментария Джейсона.

Я думаю, что Python является правильным, не позволяя выбирать объект staticmethod - так как невозможно выбрать методы экземпляра или класса! Такой объект не имеет большого смысла вне контекста:

Проверьте это: Учебник по дескриптору 

import pickle

def dosomething(a, b):
    print a, b

class MyClass(object):
    dosomething = staticmethod(dosomething) 

o = MyClass()

pickled = pickle.dumps(dosomething)

Это работает, и это то, что нужно сделать - определить функцию, выделить ее и использовать такую ​​функцию как статический метод в определенном классе.

Если у вас есть вариант использования для ваших нужд, запишите его, и я буду рад обсудить его.

...