28 lines
941 B
Python
28 lines
941 B
Python
from peewee import SqliteDatabase, Model, CharField, IntegerField, FloatField, BooleanField
|
|
|
|
# 连接到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=0) # 0表示新建，1表示已建初始仓
|
|
enabled = BooleanField(default=False) # 是否启动交易线程
|
|
|
|
class TradeRecord(BaseModel):
|
|
stock_code = CharField()
|
|
trade_type = CharField() # 'buy' 或 'sell'
|
|
price = FloatField()
|
|
volume = IntegerField()
|
|
trade_time = CharField() # 可以存储为字符串格式的时间 |