Module orca_whirlpool.internal.anchor.accounts.position_bundle

Expand source code
import typing
from dataclasses import dataclass
from solders.pubkey import Pubkey
from solana.rpc.async_api import AsyncClient
from solana.rpc.commitment import Commitment
import borsh_construct as borsh
from anchorpy.coder.accounts import ACCOUNT_DISCRIMINATOR_SIZE
from anchorpy.error import AccountInvalidDiscriminator
from anchorpy.utils.rpc import get_multiple_accounts
from anchorpy.borsh_extension import BorshPubkey
from ..program_id import PROGRAM_ID


class PositionBundleJSON(typing.TypedDict):
    position_bundle_mint: str
    position_bitmap: list[int]


@dataclass
class PositionBundle:
    discriminator: typing.ClassVar = b"\x81\xa9\xafA\xb9_ d"
    layout: typing.ClassVar = borsh.CStruct(
        "position_bundle_mint" / BorshPubkey, "position_bitmap" / borsh.U8[32]
    )
    position_bundle_mint: Pubkey
    position_bitmap: list[int]

    @classmethod
    async def fetch(
        cls,
        conn: AsyncClient,
        address: Pubkey,
        commitment: typing.Optional[Commitment] = None,
        program_id: Pubkey = PROGRAM_ID,
    ) -> typing.Optional["PositionBundle"]:
        resp = await conn.get_account_info(address, commitment=commitment)
        info = resp.value
        if info is None:
            return None
        if info.owner != program_id:
            raise ValueError("Account does not belong to this program")
        bytes_data = info.data
        return cls.decode(bytes_data)

    @classmethod
    async def fetch_multiple(
        cls,
        conn: AsyncClient,
        addresses: list[Pubkey],
        commitment: typing.Optional[Commitment] = None,
        program_id: Pubkey = PROGRAM_ID,
    ) -> typing.List[typing.Optional["PositionBundle"]]:
        infos = await get_multiple_accounts(conn, addresses, commitment=commitment)
        res: typing.List[typing.Optional["PositionBundle"]] = []
        for info in infos:
            if info is None:
                res.append(None)
                continue
            if info.account.owner != program_id:
                raise ValueError("Account does not belong to this program")
            res.append(cls.decode(info.account.data))
        return res

    @classmethod
    def decode(cls, data: bytes) -> "PositionBundle":
        if data[:ACCOUNT_DISCRIMINATOR_SIZE] != cls.discriminator:
            raise AccountInvalidDiscriminator(
                "The discriminator for this account is invalid"
            )
        dec = PositionBundle.layout.parse(data[ACCOUNT_DISCRIMINATOR_SIZE:])
        return cls(
            position_bundle_mint=dec.position_bundle_mint,
            position_bitmap=dec.position_bitmap,
        )

    def to_json(self) -> PositionBundleJSON:
        return {
            "position_bundle_mint": str(self.position_bundle_mint),
            "position_bitmap": self.position_bitmap,
        }

    @classmethod
    def from_json(cls, obj: PositionBundleJSON) -> "PositionBundle":
        return cls(
            position_bundle_mint=Pubkey.from_string(obj["position_bundle_mint"]),
            position_bitmap=obj["position_bitmap"],
        )

Classes

class PositionBundle (position_bundle_mint: solders.pubkey.Pubkey, position_bitmap: list[int])

PositionBundle(position_bundle_mint: solders.pubkey.Pubkey, position_bitmap: list[int])

Expand source code
@dataclass
class PositionBundle:
    discriminator: typing.ClassVar = b"\x81\xa9\xafA\xb9_ d"
    layout: typing.ClassVar = borsh.CStruct(
        "position_bundle_mint" / BorshPubkey, "position_bitmap" / borsh.U8[32]
    )
    position_bundle_mint: Pubkey
    position_bitmap: list[int]

    @classmethod
    async def fetch(
        cls,
        conn: AsyncClient,
        address: Pubkey,
        commitment: typing.Optional[Commitment] = None,
        program_id: Pubkey = PROGRAM_ID,
    ) -> typing.Optional["PositionBundle"]:
        resp = await conn.get_account_info(address, commitment=commitment)
        info = resp.value
        if info is None:
            return None
        if info.owner != program_id:
            raise ValueError("Account does not belong to this program")
        bytes_data = info.data
        return cls.decode(bytes_data)

    @classmethod
    async def fetch_multiple(
        cls,
        conn: AsyncClient,
        addresses: list[Pubkey],
        commitment: typing.Optional[Commitment] = None,
        program_id: Pubkey = PROGRAM_ID,
    ) -> typing.List[typing.Optional["PositionBundle"]]:
        infos = await get_multiple_accounts(conn, addresses, commitment=commitment)
        res: typing.List[typing.Optional["PositionBundle"]] = []
        for info in infos:
            if info is None:
                res.append(None)
                continue
            if info.account.owner != program_id:
                raise ValueError("Account does not belong to this program")
            res.append(cls.decode(info.account.data))
        return res

    @classmethod
    def decode(cls, data: bytes) -> "PositionBundle":
        if data[:ACCOUNT_DISCRIMINATOR_SIZE] != cls.discriminator:
            raise AccountInvalidDiscriminator(
                "The discriminator for this account is invalid"
            )
        dec = PositionBundle.layout.parse(data[ACCOUNT_DISCRIMINATOR_SIZE:])
        return cls(
            position_bundle_mint=dec.position_bundle_mint,
            position_bitmap=dec.position_bitmap,
        )

    def to_json(self) -> PositionBundleJSON:
        return {
            "position_bundle_mint": str(self.position_bundle_mint),
            "position_bitmap": self.position_bitmap,
        }

    @classmethod
    def from_json(cls, obj: PositionBundleJSON) -> "PositionBundle":
        return cls(
            position_bundle_mint=Pubkey.from_string(obj["position_bundle_mint"]),
            position_bitmap=obj["position_bitmap"],
        )

Class variables

var discriminator : ClassVar
var layout : ClassVar
var position_bitmap : list[int]
var position_bundle_mint : solders.pubkey.Pubkey

Static methods

def decode(data: bytes) ‑> PositionBundle
Expand source code
@classmethod
def decode(cls, data: bytes) -> "PositionBundle":
    if data[:ACCOUNT_DISCRIMINATOR_SIZE] != cls.discriminator:
        raise AccountInvalidDiscriminator(
            "The discriminator for this account is invalid"
        )
    dec = PositionBundle.layout.parse(data[ACCOUNT_DISCRIMINATOR_SIZE:])
    return cls(
        position_bundle_mint=dec.position_bundle_mint,
        position_bitmap=dec.position_bitmap,
    )
async def fetch(conn: solana.rpc.async_api.AsyncClient, address: solders.pubkey.Pubkey, commitment: Optional[solana.rpc.commitment.Commitment] = None, program_id: solders.pubkey.Pubkey = Pubkey( whirLbMiicVdio4qvUfM5KAg6Ct8VwpYzGff3uctyCc, )) ‑> Optional[PositionBundle]
Expand source code
@classmethod
async def fetch(
    cls,
    conn: AsyncClient,
    address: Pubkey,
    commitment: typing.Optional[Commitment] = None,
    program_id: Pubkey = PROGRAM_ID,
) -> typing.Optional["PositionBundle"]:
    resp = await conn.get_account_info(address, commitment=commitment)
    info = resp.value
    if info is None:
        return None
    if info.owner != program_id:
        raise ValueError("Account does not belong to this program")
    bytes_data = info.data
    return cls.decode(bytes_data)
async def fetch_multiple(conn: solana.rpc.async_api.AsyncClient, addresses: list[solders.pubkey.Pubkey], commitment: Optional[solana.rpc.commitment.Commitment] = None, program_id: solders.pubkey.Pubkey = Pubkey( whirLbMiicVdio4qvUfM5KAg6Ct8VwpYzGff3uctyCc, )) ‑> List[Optional[PositionBundle]]
Expand source code
@classmethod
async def fetch_multiple(
    cls,
    conn: AsyncClient,
    addresses: list[Pubkey],
    commitment: typing.Optional[Commitment] = None,
    program_id: Pubkey = PROGRAM_ID,
) -> typing.List[typing.Optional["PositionBundle"]]:
    infos = await get_multiple_accounts(conn, addresses, commitment=commitment)
    res: typing.List[typing.Optional["PositionBundle"]] = []
    for info in infos:
        if info is None:
            res.append(None)
            continue
        if info.account.owner != program_id:
            raise ValueError("Account does not belong to this program")
        res.append(cls.decode(info.account.data))
    return res
def from_json(obj: PositionBundleJSON) ‑> PositionBundle
Expand source code
@classmethod
def from_json(cls, obj: PositionBundleJSON) -> "PositionBundle":
    return cls(
        position_bundle_mint=Pubkey.from_string(obj["position_bundle_mint"]),
        position_bitmap=obj["position_bitmap"],
    )

Methods

def to_json(self) ‑> PositionBundleJSON
Expand source code
def to_json(self) -> PositionBundleJSON:
    return {
        "position_bundle_mint": str(self.position_bundle_mint),
        "position_bitmap": self.position_bitmap,
    }
class PositionBundleJSON (*args, **kwargs)

dict() -> new empty dictionary dict(mapping) -> new dictionary initialized from a mapping object's (key, value) pairs dict(iterable) -> new dictionary initialized as if via: d = {} for k, v in iterable: d[k] = v dict(**kwargs) -> new dictionary initialized with the name=value pairs in the keyword argument list. For example: dict(one=1, two=2)

Expand source code
class PositionBundleJSON(typing.TypedDict):
    position_bundle_mint: str
    position_bitmap: list[int]

Ancestors

  • builtins.dict

Class variables

var position_bitmap : list[int]
var position_bundle_mint : str