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()  # 可以存储为字符串格式的时间