CompanionMod

class CompanionMod (BaseTemplate):
    """
    Modifier for 'Companion' supported templates.

    Modifies:
        - 'is_hollow_crown': Enabled for Companion cards

    Adds:
        - 'companion_layer': Defines the Companion texture layer if card is Companion
        - 'enable_companion_layers': Called when card is a Companion
    """

    @auto_prop_cached
    def frame_layer_methods(self) -> list[Callable]:
        """Add companion layers step."""
        funcs = [self.enable_companion_layers] if self.is_companion else []
        return [*super().frame_layer_methods, *funcs]

    """
    * Bool
    """

    @auto_prop_cached
    def is_hollow_crown(self) -> bool:
        """Enable hollow crown for Companion cards."""
        if self.is_companion:
            return True
        return super().is_hollow_crown

    """
    * Layers
    """

    @auto_prop_cached
    def companion_layer(self) -> Optional[ArtLayer]:
        """Companion inner crown layer."""
        return psd.getLayer(self.pinlines, LAYERS.COMPANION)

    """
    * Companion Methods
    """

    def enable_companion_layers(self) -> None:
        """Enable Hollow Crown companion texture."""
        if self.is_legendary and self.companion_layer:
            self.companion_layer.visible = True