freqtrade_origin/freqtrade/rpc/api_server/ws_schemas.py

72 lines
1.9 KiB
Python
Raw Normal View History

2022-09-07 21:08:01 +00:00
from datetime import datetime
from typing import Any, Dict, List, Optional, TypedDict
2022-09-07 21:08:01 +00:00
from pandas import DataFrame
2023-07-18 04:43:36 +00:00
from pydantic import ConfigDict, BaseModel
2022-09-07 21:08:01 +00:00
from freqtrade.constants import PairWithTimeframe
from freqtrade.enums.rpcmessagetype import RPCMessageType, RPCRequestType
class BaseArbitraryModel(BaseModel):
2023-07-18 04:43:36 +00:00
model_config = ConfigDict(arbitrary_types_allowed=True)
2022-09-07 21:08:01 +00:00
class WSRequestSchema(BaseArbitraryModel):
type: RPCRequestType
data: Optional[Any] = None
class WSMessageSchemaType(TypedDict):
# Type for typing to avoid doing pydantic typechecks.
type: RPCMessageType
data: Optional[Dict[str, Any]]
2022-09-07 21:08:01 +00:00
class WSMessageSchema(BaseArbitraryModel):
type: RPCMessageType
data: Optional[Any] = None
2023-07-18 04:43:36 +00:00
model_config = ConfigDict(extra='allow')
2022-09-07 21:08:01 +00:00
# ------------------------------ REQUEST SCHEMAS ----------------------------
class WSSubscribeRequest(WSRequestSchema):
type: RPCRequestType = RPCRequestType.SUBSCRIBE
data: List[RPCMessageType]
class WSWhitelistRequest(WSRequestSchema):
type: RPCRequestType = RPCRequestType.WHITELIST
data: None = None
class WSAnalyzedDFRequest(WSRequestSchema):
type: RPCRequestType = RPCRequestType.ANALYZED_DF
data: Dict[str, Any] = {"limit": 1500, "pair": None}
2022-09-07 21:08:01 +00:00
# ------------------------------ MESSAGE SCHEMAS ----------------------------
class WSWhitelistMessage(WSMessageSchema):
type: RPCMessageType = RPCMessageType.WHITELIST
data: List[str]
class WSAnalyzedDFMessage(WSMessageSchema):
class AnalyzedDFData(BaseArbitraryModel):
key: PairWithTimeframe
df: DataFrame
la: datetime
type: RPCMessageType = RPCMessageType.ANALYZED_DF
data: AnalyzedDFData
class WSErrorMessage(WSMessageSchema):
type: RPCMessageType = RPCMessageType.EXCEPTION
data: str
2022-09-07 21:08:01 +00:00
# --------------------------------------------------------------------------