Files
sfgrid/strategy_db.py
T
2025-10-27 17:38:49 +08:00

20 lines
614 B
Python

from peewee import SqliteDatabase, Model, CharField, IntegerField, FloatField
# 连接到SQLite数据库
db = SqliteDatabase('example.db')
# 定义基础模型类
class BaseModel(Model):
class Meta:
database = db
# 定义Target类,对应targets表
class TradeTarget(BaseModel):
stock_code = CharField(unique=True)
stock_name = CharField()
current_position = IntegerField()
grid_index = IntegerField()
last_trade_price = FloatField()
current_buy_price = FloatField()
current_sell_price = FloatField()
status = IntegerField(default=1) # 1表示启用,0表示停止