Definir un decorador en una clase, que también se puede usar dentro de la definición de clase

Question

Estoy tratando de implementar un sistema de "subcomando" como una clase heredable en Python. Mi caso de uso esperado es algo así como:

from command import Command 
import sys 

class MyCommand(Command): 
    @Command.subcommand 
    def foo(self): 
     print "this can be run as a subcommand" 

    def bar(self): 
     print "but this is a plain method and isn't exposed to the CLI" 

MyCommand()(*sys.argv) 

# at the command line, the user runs "mycommand.py foo" 

he implementado Command.subcommand como un método estático y todo funcionaba bien hasta que he intentado añadir un subcomando a la clase padre, que me consiguió TypeError: 'staticmethod' object is not callable. En retrospectiva, es obvio que no va a funcionar this:

class Command(object): 
    @staticmethod 
    def subcommand(method): 
     method.is_subcommand = True 

     return method 

    @subcommand 
    def common(self): 
     print "this subcommand is available to all child classes" 

La única alternativa que he encontrado hasta ahora es declarar el decorador subcommand fuera de la clase padre, y luego inyectarla después de la definición de clase se ha completado.

def subcommand(method): 
    method.is_subcommand = True 

    return method 

class Command(object): 
    @subcommand 
    def common(self): 
     print "this subcommand is available to all child classes" 

Command.subcommand = staticmethod(subcommand) 
del subcommand 

Sin embargo, como alguien que nunca utiliza Python antes de añadir decoradores, esto se siente muy torpe para mí. ¿Hay una forma más elegante de lograr esto?

Answer 1

Hay dos soluciones a este problema que se me ocurren. El más simple es hacer que sea un método estático después haya terminado de usarlo en la clase padre:

class Command(object): 
    def subcommand(method): # Regular function in class definition scope. 
     method.is_subcommand = True 

     return method 

    @subcommand 
    def common(self): 
     print "this subcommand is available to all child classes" 

    subcommand = staticmethod(subcommand) 
    # Now a static method. Can no longer be called during class definition phase. 

Esto es algo frágil en el que no se puede utilizar en la clase padre que después de realizar un método estático. La forma más robusta para hacer esto es añadir una clase intermedia:

class Command(object): 
    @staticmethod 
    def subcommand(method): 
     method.is_subcommand = True 

     return method 

class CommandBase(Command): 

    @Command.subcommand 
    def common(self): 
     print "this subcommand is available to all child classes" 

Ahora puede heredar todas sus clases de CommandBase en lugar de Command.