如何使用类装饰器包装器？

Question

我有一个data_class对象，希望使用装饰器添加一些功能。它可以工作，但是类名丢失，成为DefaultInstance. .subclass 。

我知道functools.wraps可以修复这个问题，但是应该在哪里/如何使用它呢？

from dataclasses import dataclass, fields, is_dataclass
   
def defaultinstance(object):
  
    class SubClass(object):
        @classmethod
        def woof(cls):
            print("woof")

    return SubClass


@defaultinstance
@dataclass
class Dog:
    name: str
    paws: int


doggo = Dog(name="jack", paws=4)

print(doggo)
# defaultinstance.<locals>.SubClass(name='jack', paws=4)

Answer 1

您可以让装饰器像这样修改类

from dataclasses import dataclass, fields, is_dataclass
   
def defaultinstance(object):
    """
    creates a class method 'woof' and adds it to object
    """
    @classmethod
    def woof(cls):
        print("woof")
    
    setattr(object, 'woof', woof)
    return object


@defaultinstance
@dataclass
class Dog:
    name: str
    paws: int


doggo = Dog(name="jack", paws=4)

print(doggo)

有结果

Dog(name='jack', paws=4)

这样，您就不必创建一个需要伪装成旧类的新类，而旧类从未被使用过。