Расширение модуля python черного ящика

Question

У меня есть модуль python «blackbox», который я хотел бы расширить. Модуль предоставляет класс class Foo без функции __init__ и вспомогательную функцию FooMaker, которая возвращает объекты типа Foo. Обычная стратегия проходящих модулей:

class ExtendedFoo(blackbox.Foo): 
    def __init__(self, x): 
     super(ExtendedFoo, self).__init__(x) 

не будет работать здесь, так как, как уже упоминалось выше, я должен полагаться на blackbox.FooMaker вместо __init__ функции Foo «s.

Любые идеи о том, как расширить модуль blackbox?

Answer 1

Оказалось, что Foo класс имел сеттер функцию, таким образом, что в конечном счете работал для меня был

class ExtendedFoo(blackbox.Foo): 
    pass 

def ExtendedFooMaker(*args): 
    _efoo = ExtendedFoo() 
    _efoo.set(FooMaker(*args).get()) 
    return _efoo 

Answer 2

Если модуль не делает серьезного взлома, вы можете просто заменить его элементы. Обратите внимание, что все ссылки и задания работают над именами blackbox., а не локальными именами!

Если вы хотите, чтобы ваш новый класс '__init__ был запущен внутри FooMaker, просто наследуйте и заменяйте Foo.

import blackbox 
class ExtendedFoo(blackbox.Foo): 
    def __init__(self, x): 
     super(ExtendedFoo, self).__init__(x) 

blackbox.Foo = ExtendedFoo 

Таким образом, FooMaker будет экземпляр класса только ExtendedFoo.

В качестве альтернативы вы можете заменить FooMaker своей собственной версией.

import blackbox 
_bbfoomaker = blackbox.FooMaker # keep reference so we can use it later 
def SuperFooMaker(*fooargs, **fookwargs): 
    new_instance = _bbfoomaker(*fooargs, **fookwargs) 
    new_instance.whitebox = True # or whatever 

blackbox.FooMaker = SuperFooMaker 

Ограничение на секунду, что если другой модуль делает что-то вроде from blackbox import FooMaker, ваш модуль должен быть запущен/импортирован ранее. В противном случае другой модуль все еще видит оригинальную фабрику, похожую на то, как у вас все еще есть _bbfoomaker.

Answer 3

Это должно сделать то же самое, как ваше решение, но немного короче:

from blackbox import Foo, FooMaker 

class ExtendedFoo(Foo): 

    def __init__(self, *args, **kwargs): 
     self.set(FooMaker(*args, **kwargs).get())