Come creare dinamicamente funzioni di livello di modulo da metodi in una classe

Question

Sto cercando di creare dinamicamente le funzioni a livello di modulo dai metodi in una classe. Quindi per ogni metodo in una classe, voglio creare una funzione con lo stesso nome che crea un'istanza della classe e poi chiama il metodo.

Il motivo che voglio fare questo è così posso prendere un approccio orientato agli oggetti per la creazione di file in tessuto. Dal tessuto chiamerà funzioni di livello di modulo, ma non i metodi di una classe, questo è il mio lavoro-around.

Ho usato i link seguenti per farmi iniziare

• Come faccio ad avere l'elenco dei metodi di in una classe Python?
• dinamicamente l'aggiunta di funzioni ad un modulo Python
• Come chiamo setattr () sul modulo corrente ?
• http://effbot.org/zone/python-getattr.htm
• Chiamata di una funzione di un modulo da una stringa con il nome della funzione in Python
• Come modificare lo spazio dei nomi locali in Python

E devo venire con il seguente codice

import inspect
import sys
import types

class TestClass(object):
    def __init__(self):
        pass

    def method1(self, arg1):
        print 'method 1 %s' % arg1

    def method2(self):
        print 'method 2'

def fabric_class_to_function_magic(module_name):
    # get the module as an object
    print module_name
    module_obj = sys.modules[module_name]
    print dir(module_obj)

    # Iterate over the methods of the class and dynamically create a function
    # for each method that calls the method and add it to the current module
    for method in inspect.getmembers(TestClass, predicate=inspect.ismethod):
        print
        print method
        method_name, method_obj = method

        # create a new template function which calls the method
        def newfunc_template(*args, **kwargs):
            tc = TestClass()
            func = getattr(tc, method_name)
            return func(*args, **kwargs)

        # create the actual function
        print 'code: ', newfunc_template.func_code
        print 'method_name: ', method_name
        newfunc = types.FunctionType(newfunc_template.func_code,
                                     {'TestClass': TestClass,
                                      'getattr': getattr,
                                      'method_name': method_name,
                                      },
                                     name=method_name,
                                     argdefs=newfunc_template.func_defaults,
                                     closure=newfunc_template.func_closure,
                                     )

        # add the new function to the current module
        setattr(module_obj, method_name, newfunc)

# test the dynamically created module level function
thismodule = sys.modules[__name__]
print dir(thismodule)
fabric_class_to_function_magic(__name__)
print dir(thismodule)
method1('arg1')
method2()

E ho il seguente errore

['TestClass', '__builtins__', '__doc__', '__file__', '__name__', '__package__', 'fabric_class_to_function_magic', 'inspect', 'sys', 'thismodule', 'types']
__main__
['TestClass', '__builtins__', '__doc__', '__file__', '__name__', '__package__', 'fabric_class_to_function_magic', 'inspect', 'sys', 'thismodule', 'types']

('__init__', <unbound method TestClass.__init__>)
code:  <code object newfunc_template at 0x7f8800a28d50, file "test.py", line 85>
method_name:  __init__

('method1', <unbound method TestClass.method1>)
code:  <code object newfunc_template at 0x7f8800a28d50, file "test.py", line 85>
method_name:  method1

('method2', <unbound method TestClass.method2>)
code:  <code object newfunc_template at 0x7f8800a28d50, file "test.py", line 85>
method_name:  method2
['TestClass', '__builtins__', '__doc__', '__file__', '__init__', '__name__', '__package__', 'fabric_class_to_function_magic', 'inspect', 'method1', 'method2', 'sys', 'thismodule', 'types']
Traceback (most recent call last):
  File "test.py", line 111, in <module>
    method1('arg1')
  File "test.py", line 88, in newfunc_template
    return func(*args, **kwargs)
TypeError: method2() takes exactly 1 argument (2 given)

Sembra essere riutilizzare il riferimento alla funzione? Tutte le idee?

UPDATE: Ecco il codice di lavoro con correzione di Ned Batchelder

def fabric_class_to_function_magic(module_name):
    # get the module as an object
    module_obj = sys.modules[module_name]

    # Iterate over the methods of the class and dynamically create a function
    # for each method that calls the method and add it to the current module
    for method in inspect.getmembers(TestClass, predicate=inspect.ismethod):
        method_name, method_obj = method

        # get the bound method
        tc = TestClass()
        func = getattr(tc, method_name)

        # add the function to the current module
        setattr(module_obj, method_name, func)

UPDATE 2: Ecco il mio post sul blog sul tema: http://www.saltycrane.com/blog/2010/09/class-based-fabric-scripts-metaprogramming-hack/

Answer 1

Stai pensando troppo la vostra soluzione. Modificare la fine del fabric_class_to_function_magic essere questo:

    tc = TestClass()
    func = getattr(tc, method_name)

    # add the new function to the current module
    setattr(module_obj, method_name, func)

e funziona benissimo. Non c'è bisogno di fare un nuovo oggetto funzione, è già uno restituito dal getattr sul vostro oggetto. Il metodo vincolato restituito da getattr è una cosa callable. Basta assegnarlo al proprio attributo del modulo, e siete a posto.

Answer 2

in realtà il codice è giusto, ma quando ritorno func (* args, kwargs **) esegue, args passeranno tupla vuota come () e non ci sono parametri nel method2, quindi solleva tale eccezione,

una soluzione rapida verso il vostro problema sarebbe, come

class TestClass(object):
    def __init__(self):
        pass

    def method1(self, arg1):
        print 'method 1 %s' % arg1

    def method2(self, *args, **kw):
        print 'method 2'