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表示停止