Class SetAttr

mypyc/ir/ops.py:912–958

obj.attr = src (for a native object)

@final
class SetAttr(RegisterOp):
    """obj.attr = src (for a native object)"""

    error_kind = ERR_FALSE

    def __init__(self, obj: Value, attr: str, src: Value, line: int) -> None:
        super().__init__(line)
        self.obj = obj
        self.attr = attr
        self.src = src
        assert isinstance(obj.type, RInstance), "Attribute access not supported: %s" % obj.type
        self.class_type = obj.type
        self.type = bool_rprimitive
        # If True, we can safely assume that the attribute is previously undefined
        # and we don't use a setter
        self.is_init = False

        cl = self.class_type.class_ir
        is_propset = False
        for ir in cl.mro:
            propset = ir.method_decls.get(PROPSET_PREFIX + attr)
            if propset is not None:
                is_propset = not propset.implicit
                break
        # If True, this op represents calling a property setter.
        self.is_propset = is_propset

    def mark_as_initializer(self) -> None:
        self.is_init = True
        self.error_kind = ERR_NEVER
        self.type = void_rtype

    def sources(self) -> list[Value]:
        return [self.obj, self.src]

    def set_sources(self, new: list[Value]) -> None:
        self.obj, self.src = new

    def stolen(self) -> list[Value]:
        # The property setter method increfs the passed value so don't treat it as a steal
        # to avoid leaking.
        if self.is_propset:
            return []
        return [self.src]

    def accept(self, visitor: OpVisitor[T]) -> T:
        return visitor.visit_set_attr(self)


# Default name space for statics, variables