python set_cls_attrs wrapper

set_cls_attrs wrapper

def set_cls_attrs(**injected_attrs):
    def wrap_cls(cls):
        # inject new class attr
        for _attr_name, _attr_val in injected_attrs.items():
            setattr(cls, _attr_name, _attr_val)

        # wrapper class based on original cls
        class wrapper_cls(cls):
            def __init__(self, *args, **kw):
                self.original_cls_instance = cls(*args, **kw)
                # some magic from functools
                functools.update_wrapper(self, self.original_cls_instance)

        # for CLI compatible
        wrapper_cls.__name__ = cls.__name__
        return wrapper_cls

    return wrap_cls