market model

1 files changed, 15 insertions, 0 deletions

diff --git a/server/models/market.py b/server/models/market.py
index 072e40f..8606075 100644
--- a/server/models/market.py
+++ b/server/models/market.py
@@ -1,5 +1,7 @@
 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
@@ -7,3 +9,16 @@ class MarketData(BaseModel):
     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"