Big refactoring
This commit is contained in:
@@ -1,6 +1,8 @@
|
||||
from datetime import datetime
|
||||
|
||||
from sqlalchemy import DateTime, Float, ForeignKey, String
|
||||
import json
|
||||
|
||||
from sqlalchemy import DateTime, Float, ForeignKey, String, Text
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.database import Base
|
||||
@@ -23,4 +25,34 @@ class TradeSetup(Base):
|
||||
DateTime(timezone=True), nullable=False
|
||||
)
|
||||
|
||||
confidence_score: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||||
targets_json: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||||
conflict_flags_json: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||||
recommended_action: Mapped[str | None] = mapped_column(String(20), nullable=True)
|
||||
reasoning: Mapped[str | None] = mapped_column(Text, nullable=True)
|
||||
risk_level: Mapped[str | None] = mapped_column(String(10), nullable=True)
|
||||
actual_outcome: Mapped[str | None] = mapped_column(String(20), nullable=True)
|
||||
|
||||
ticker = relationship("Ticker", back_populates="trade_setups")
|
||||
|
||||
@property
|
||||
def targets(self) -> list[dict]:
|
||||
if not self.targets_json:
|
||||
return []
|
||||
try:
|
||||
parsed = json.loads(self.targets_json)
|
||||
except (TypeError, ValueError):
|
||||
return []
|
||||
return parsed if isinstance(parsed, list) else []
|
||||
|
||||
@property
|
||||
def conflict_flags(self) -> list[str]:
|
||||
if not self.conflict_flags_json:
|
||||
return []
|
||||
try:
|
||||
parsed = json.loads(self.conflict_flags_json)
|
||||
except (TypeError, ValueError):
|
||||
return []
|
||||
if not isinstance(parsed, list):
|
||||
return []
|
||||
return [str(item) for item in parsed]
|
||||
|
||||
Reference in New Issue
Block a user