<reponame>nftlabs/nftlabs-sdk-python
from typing import Generic
from thirdweb.core.classes.contract_metadata import ContractMetadata
from thirdweb.core.classes.contract_wrapper import ContractWrapper
from thirdweb.types.contract import TRoyaltyABI
from thirdweb.types.settings.metadata import ContractRoyaltySchema
from web3.eth import TxReceipt


class ContractRoyalty(Generic[TRoyaltyABI]):
    _contract_wrapper: ContractWrapper[TRoyaltyABI]
    _metadata: ContractMetadata

    def __init__(
        self, contract_wrapper: ContractWrapper[TRoyaltyABI], metadata: ContractMetadata
    ):
        self._contract_wrapper = contract_wrapper
        self._metadata = metadata

    def get_default_royalty_info(self) -> ContractRoyaltySchema:
        """
        Get the default royalty information for this contract.

        :returns: the default royalty information.
        """

        (
            royalty_recipient,
            royalty_bps,
        ) = self._contract_wrapper._contract_abi.get_default_royalty_info.call()
        return ContractRoyaltySchema(royalty_recipient, royalty_bps)

    def get_token_royalty_info(self, token_id: int) -> ContractRoyaltySchema:
        """
        Get the royalty information for a specific token.

        :param token_id: the id of the token.
        :returns: the royalty information for the token.
        """

        (
            royalty_recipient,
            royalty_bps,
        ) = self._contract_wrapper._contract_abi.get_royalty_info_for_token.call(
            token_id
        )
        return ContractRoyaltySchema(royalty_recipient, royalty_bps)

    def set_default_royalty_info(
        self, royalty_data: ContractRoyaltySchema
    ) -> TxReceipt:
        """
        Set the default royalty information for this contract.

        :param royalty_data: the default royalty information.
        :returns: the transaction receipt of setting the royalty.
        """

        metadata = self._metadata.get()
        metadata.fee_recipient = royalty_data.fee_recipient
        metadata.seller_fee_basis_points = royalty_data.seller_fee_basis_points

        contract_uri = self._metadata._parse_and_upload_metadata(metadata)

        encoded = [
            self._contract_wrapper.get_contract_interface().encodeABI(
                "set_default_royalty_info",
                [metadata.fee_recipient, metadata.seller_fee_basis_points],
            ),
            self._contract_wrapper.get_contract_interface().encodeABI(
                "set_contract_uri", [contract_uri]
            ),
        ]

        return self._contract_wrapper.multi_call(encoded)

    def set_token_royalty_info(
        self, token_id: int, royalty_data: ContractRoyaltySchema
    ) -> TxReceipt:
        """
        Set the royalty information for a specific token.

        :param token_id: the id of the token.
        :param royalty_data: the royalty information for the token.
        :returns: the transaction receipt of setting the royalty.
        """

        return self._contract_wrapper.send_transaction(
            "set_royalty_info_for_token",
            [
                token_id,
                royalty_data.fee_recipient,
                royalty_data.seller_fee_basis_points,
            ],
        )

from typing import Generic
from thirdweb.core.classes.contract_metadata import ContractMetadata
from thirdweb.core.classes.contract_wrapper import ContractWrapper
from thirdweb.types.contract import TRoyaltyABI
from thirdweb.types.settings.metadata import ContractRoyaltySchema
from web3.eth import TxReceipt


class ContractRoyalty(Generic[TRoyaltyABI]):
    _contract_wrapper: ContractWrapper[TRoyaltyABI]
    _metadata: ContractMetadata

    def __init__(
        self, contract_wrapper: ContractWrapper[TRoyaltyABI], metadata: ContractMetadata
    ):
        self._contract_wrapper = contract_wrapper
        self._metadata = metadata

    def get_default_royalty_info(self) -> ContractRoyaltySchema:
        """
        Get the default royalty information for this contract.

        :returns: the default royalty information.
        """

        (
            royalty_recipient,
            royalty_bps,
        ) = self._contract_wrapper._contract_abi.get_default_royalty_info.call()
        return ContractRoyaltySchema(royalty_recipient, royalty_bps)

    def get_token_royalty_info(self, token_id: int) -> ContractRoyaltySchema:
        """
        Get the royalty information for a specific token.

        :param token_id: the id of the token.
        :returns: the royalty information for the token.
        """

        (
            royalty_recipient,
            royalty_bps,
        ) = self._contract_wrapper._contract_abi.get_royalty_info_for_token.call(
            token_id
        )
        return ContractRoyaltySchema(royalty_recipient, royalty_bps)

    def set_default_royalty_info(
        self, royalty_data: ContractRoyaltySchema
    ) -> TxReceipt:
        """
        Set the default royalty information for this contract.

        :param royalty_data: the default royalty information.
        :returns: the transaction receipt of setting the royalty.
        """

        metadata = self._metadata.get()
        metadata.fee_recipient = royalty_data.fee_recipient
        metadata.seller_fee_basis_points = royalty_data.seller_fee_basis_points

        contract_uri = self._metadata._parse_and_upload_metadata(metadata)

        encoded = [
            self._contract_wrapper.get_contract_interface().encodeABI(
                "set_default_royalty_info",
                [metadata.fee_recipient, metadata.seller_fee_basis_points],
            ),
            self._contract_wrapper.get_contract_interface().encodeABI(
                "set_contract_uri", [contract_uri]
            ),
        ]

        return self._contract_wrapper.multi_call(encoded)

    def set_token_royalty_info(
        self, token_id: int, royalty_data: ContractRoyaltySchema
    ) -> TxReceipt:
        """
        Set the royalty information for a specific token.

        :param token_id: the id of the token.
        :param royalty_data: the royalty information for the token.
        :returns: the transaction receipt of setting the royalty.
        """

        return self._contract_wrapper.send_transaction(
            "set_royalty_info_for_token",
            [
                token_id,
                royalty_data.fee_recipient,
                royalty_data.seller_fee_basis_points,
            ],
        )
Get the default royalty information for this contract.          :returns: the default royalty information. Get the royalty information for a specific token.          :param token_id: the id of the token.         :returns: the royalty information for the token. Set the default royalty information for this contract.          :param royalty_data: the default royalty information.         :returns: the transaction receipt of setting the royalty. Set the royalty information for a specific token.          :param token_id: the id of the token.         :param royalty_data: the royalty information for the token.         :returns: the transaction receipt of setting the royalty.