76 lines
4.7 KiB
Python
76 lines
4.7 KiB
Python
from sqlalchemy import create_engine, MetaData
|
||
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column, Session
|
||
from sqlalchemy import Integer, String, Float, DateTime, DECIMAL, func
|
||
from datetime import datetime
|
||
from typing import Optional
|
||
import json
|
||
|
||
class Base(DeclarativeBase):
|
||
"""SQLAlchemy基础类"""
|
||
pass
|
||
|
||
class StrategyPosition(Base):
|
||
"""持仓表模型"""
|
||
__tablename__ = 'deh_strategy_position_new'
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True, comment='自增id')
|
||
st_id: Mapped[int] = mapped_column(Integer, nullable=False, comment='策略id')
|
||
k_id: Mapped[int] = mapped_column(Integer, default=0, comment='对应strategy_key 的ID')
|
||
asset: Mapped[str] = mapped_column(String(32), default='USDT', comment='使用资产名称')
|
||
symbol: Mapped[str] = mapped_column(String(50), nullable=False, comment='交易对')
|
||
price: Mapped[Optional[float]] = mapped_column(Float, comment='持仓均价')
|
||
side: Mapped[str] = mapped_column(String(10), nullable=False, comment='方向')
|
||
|
||
# 注意：这里属性名是 qty，但对应数据库字段是 sum
|
||
qty: Mapped[float] = mapped_column("sum", Float, nullable=False, comment='仓位(张数)')
|
||
|
||
asset_num: Mapped[float] = mapped_column(DECIMAL(20, 8), default=0.00000000, comment='资产数量')
|
||
asset_profit: Mapped[Optional[float]] = mapped_column(DECIMAL(20, 8), comment='利润数量')
|
||
leverage: Mapped[int] = mapped_column(Integer, default=0, comment='杠杆倍数')
|
||
uptime: Mapped[int] = mapped_column(Integer, nullable=False, comment='更新时间')
|
||
profit_price: Mapped[float] = mapped_column(DECIMAL(20, 8), default=0.00000000, comment='止盈价格')
|
||
stop_price: Mapped[float] = mapped_column(DECIMAL(20, 8), default=0.00000000, comment='止损价格')
|
||
liquidation_price: Mapped[float] = mapped_column(DECIMAL(20, 8), default=0.00000000, comment='强平价格')
|
||
|
||
def __repr__(self) -> str:
|
||
return f"StrategyPosition(id={self.id!r}, k_id={self.k_id!r}, symbol={self.symbol!r})"
|
||
|
||
class StrategyOrder(Base):
|
||
"""订单表模型"""
|
||
__tablename__ = 'deh_strategy_order_new'
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
||
st_id: Mapped[Optional[int]] = mapped_column(Integer, comment='策略id')
|
||
k_id: Mapped[int] = mapped_column(Integer, default=0, comment='对应strategy_key 的ID')
|
||
asset: Mapped[Optional[str]] = mapped_column(String(32), default='USDT', comment='资产名称')
|
||
order_id: Mapped[Optional[str]] = mapped_column(String(765), comment='订单id')
|
||
symbol: Mapped[Optional[str]] = mapped_column(String(120), comment='交易对')
|
||
side: Mapped[Optional[str]] = mapped_column(String(120), comment='订单方向')
|
||
price: Mapped[Optional[float]] = mapped_column(Float, comment='订单价格')
|
||
time: Mapped[Optional[int]] = mapped_column(Integer, comment='订单时间')
|
||
order_qty: Mapped[Optional[float]] = mapped_column(Float, comment='订单挂单数量')
|
||
last_qty: Mapped[Optional[float]] = mapped_column(Float, comment='订单成交数量')
|
||
avg_price: Mapped[Optional[float]] = mapped_column(Float, comment='订单成交均价')
|
||
exchange_id: Mapped[Optional[int]] = mapped_column(Integer, comment='交易所id')
|
||
|
||
def __repr__(self) -> str:
|
||
return f"StrategyOrder(id={self.id!r}, order_id={self.order_id!r})"
|
||
|
||
class StrategyKX(Base):
|
||
"""账户信息表模型"""
|
||
__tablename__ = 'deh_strategy_kx_new'
|
||
|
||
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True, comment='自增id')
|
||
st_id: Mapped[Optional[int]] = mapped_column(Integer, comment='策略id')
|
||
k_id: Mapped[int] = mapped_column(Integer, default=0, comment='对应strategy_key 的ID')
|
||
asset: Mapped[Optional[str]] = mapped_column(String(32), default='USDT', comment='资产名')
|
||
balance: Mapped[float] = mapped_column(DECIMAL(20, 8), default=0.00000000, comment='当日账户金额')
|
||
withdrawal: Mapped[float] = mapped_column(DECIMAL(20, 8), default=0.00000000, comment='当日提现')
|
||
deposit: Mapped[float] = mapped_column(DECIMAL(20, 8), default=0.00000000, comment='当日充值')
|
||
other: Mapped[float] = mapped_column(DECIMAL(20, 8), default=0.00000000, comment='当日其他')
|
||
profit: Mapped[float] = mapped_column(DECIMAL(20, 8), default=0.00000000, comment='当日利润')
|
||
time: Mapped[Optional[int]] = mapped_column(Integer, comment='时间')
|
||
up_time: Mapped[Optional[datetime]] = mapped_column(DateTime, default=func.now(), onupdate=func.now(), comment='最后更新时间')
|
||
|
||
def __repr__(self) -> str:
|
||
return f"StrategyKX(id={self.id!r}, k_id={self.k_id!r}, time={self.time!r})" |