bbgo_origin/python/bbgo/data/trade.py
2022-04-22 21:32:04 +08:00

42 lines
955 B
Python

from __future__ import annotations
from dataclasses import dataclass
from datetime import datetime
import bbgo_pb2
from ..enums import SideType
from ..utils import parse_float
from ..utils import parse_time
@dataclass
class Trade:
session: str
exchange: str
symbol: str
trade_id: str
price: float
quantity: float
created_at: datetime
side: SideType
fee_currency: str
fee: float
maker: bool
@classmethod
def from_pb(cls, obj: bbgo_pb2.Trade) -> Trade:
return cls(
session=obj.session,
exchange=obj.exchange,
symbol=obj.symbol,
trade_id=obj.id,
price=parse_float(obj.price),
quantity=parse_float(obj.quantity),
created_at=parse_time(obj.created_at),
side=SideType(obj.side),
fee_currency=obj.fee_currency,
fee=parse_float(obj.fee),
maker=obj.maker,
)