mirror of
https://github.com/status-im/status-python-sdk.git
synced 2026-08-31 06:01:19 +00:00
Nick Ninov
be0650d94e
New features: - Swap Tokens - Send Transactions - Search Transactions - Add payment requests to messages - Add payment requests to listening messages Related to: - https://github.com/status-im/status-python-sdk/issues/3
79 lines
5.0 KiB
Python
79 lines
5.0 KiB
Python
from pydantic import BaseModel, Field, AfterValidator
|
|
from typing import Optional, Literal, Annotated
|
|
import re
|
|
|
|
def validate_date(value: str) -> str:
|
|
if not re.match(r"^\d{4}-(0[1-9]|1[0-2])-(0[1-9]|[12]\d|3[01])$", value):
|
|
raise ValueError(f"Expected YYYY-MM-DD, got '{value}'")
|
|
return value
|
|
|
|
DateStr = Annotated[str, AfterValidator(validate_date)]
|
|
|
|
class AccountBalanceInput(BaseModel):
|
|
ccy: str = Field(description="ISO 4217 alpha code to represent the fiat currency", default="USD")
|
|
include_market: bool = Field(default=False, description="If market related information should be included per token in the final output")
|
|
|
|
class TokenSearchInput(BaseModel):
|
|
chain_id: Optional[int] = Field(description="Chain ID where the token exists.", default=None)
|
|
token_symbols: Optional[list[str]] = Field(description="Token Symbol for the given Chain ID", default=None)
|
|
|
|
class BalanceSearchInput(BaseModel):
|
|
chain_id: int = Field(description="Chain ID where the tokens exist.", default=1)
|
|
token_addresses: list[str] = Field(description=(
|
|
"Required. Non-empty list of token contract addresses to look up on the given chain. "
|
|
"This cannot be null or omitted. Use `get_token_info` first to resolve token "
|
|
"symbols (e.g. ETH, SNT) into addresses for the chain."
|
|
))
|
|
wallet_address: str = Field(description="The external wallet address (or ENS name) whose balance will be looked up.")
|
|
ccy: str = Field(description="ISO 4217 alpha code to represent the fiat currency", default="USD")
|
|
|
|
class AccountContactInput(BaseModel):
|
|
display_name: Optional[str] = Field(description="The current display name of the contact", default=None)
|
|
status: Literal['has_added_us', 'added', 'mutual', 'groups'] = Field(description=(
|
|
"If `has_added_us` then the contact request is pending on my side. "
|
|
"If `added` then the contact request is pending on the other user's side. "
|
|
"If `added` then both the account and other account are contacts. "
|
|
"If `group` then all group chats will be returned"
|
|
))
|
|
|
|
class AccountContactManagementInput(BaseModel):
|
|
public_key: str = Field(description="The contact's public key that will be modified")
|
|
action: Literal['accept', 'reject'] = Field(description=(
|
|
"If `accept` and the contact is pending then the contact request will be accepted. "
|
|
"If `accept` and there is no pending request then the contact will be sent a request. "
|
|
"If `reject` and the contact is pending then the contact request will be declined. "
|
|
"If `reject` and the contact exists then the contact will be removed."
|
|
))
|
|
display_name: Optional[str] = Field(description="Required only if a new contact is being created. If a request is pending to be accepted, then this field is not required.")
|
|
|
|
class MessageInput(BaseModel):
|
|
chat_id: str = Field(description="Required to send messages and fetch messages")
|
|
message: Optional[str] = Field(description="Required to send messages", default=None)
|
|
start_date: Optional[DateStr] = Field(description="Required to fetch chat messages from the specified date. Date should be in YYYY-MM-DD format.", default=None)
|
|
end_date: Optional[DateStr] = Field(description="Required to fetch chat messages to the specified date. Date should be in YYYY-MM-DD format.", default=None)
|
|
|
|
class TransactionSearchInput(BaseModel):
|
|
chain_ids: Optional[list[int]] = Field(description="Chain IDs where the token exists.", default=None)
|
|
token_symbols: Optional[list[str]] = Field(description="Token Symbol for the given Chain ID", default=None)
|
|
refresh: bool = Field(description="If `True` then the data will be refetched from scratch. If `False` then the data will be cached after the first call.")
|
|
start_date: Optional[DateStr] = Field(description="Required to fetch chat messages from the specified date. Date should be in YYYY-MM-DD format.", default=None)
|
|
end_date: Optional[DateStr] = Field(description="Required to fetch chat messages to the specified date. Date should be in YYYY-MM-DD format.", default=None)
|
|
|
|
class SendTransactionInput(BaseModel):
|
|
address: str = Field(description="The wallet address of the receiver")
|
|
symbol: str = Field(description="Either a valid Status token symbol (e.g. `ETH`, `SNT`) or its token address")
|
|
amount: float = Field(description="The amount of the token that will be sent to the receiver")
|
|
chain_id: int = Field(description="Chain ID where the token exists.", default=1)
|
|
|
|
class SwapTokensInput(BaseModel):
|
|
from_token: str = Field(description="The token to swap from. Either a valid Status token symbol (e.g. `ETH`, `SNT`) or its token address.")
|
|
to_token: str = Field(description="The token to swap to. Either a valid Status token symbol (e.g. `ETH`, `SNT`) or its token address.")
|
|
amount: float = Field(description="The amount of `from_token` to swap.")
|
|
chain_id: int = Field(description=(
|
|
"Chain ID where both tokens exist. The swap happens on a single chain, so `from_token` and `to_token` "
|
|
"must be on the same chain."
|
|
), default=1)
|
|
|
|
class NoArgs(BaseModel):
|
|
pass
|