from datetime import datetime
from pydantic import BaseModel
from sqlalchemy import Column, Integer, String, Float, DateTime
from db import Base

class MarketData(BaseModel):
    timestamp: datetime
    lmp: float
    energy: float
    congestion: float
    loss: float

# New DB model
class MarketDataDB(Base):
    __tablename__ = "market_data"

    id = Column(Integer, primary_key=True, index=True)
    timestamp = Column(DateTime, index=True)
    lmp = Column(Float)
    energy = Column(Float)
    congestion = Column(Float)
    loss = Column(Float)
    market = Column(String, index=True)   # eg. "ISONE"
    type = Column(String, index=True)     # "REALTIME" or "DAYAHEAD"