continue on crud

backend/trading_journal/crud.py

@@ -1,12 +1,13 @@
+from datetime import datetime, timezone
 from typing import Mapping
 
 from sqlalchemy.exc import IntegrityError
-from sqlmodel import Session
+from sqlmodel import Session, select
 
 from trading_journal import models
 
 
-def _coerce_enum(enum_cls, value, field_name: str):
+def _check_enum(enum_cls, value, field_name: str):
     if value is None:
         raise ValueError(f"{field_name} is required")
     # already an enum member
@@ -29,29 +30,66 @@ def create_trade(session: Session, trade_data: Mapping) -> models.Trades:
         data = dict(trade_data)
     allowed = {c.name for c in models.Trades.__table__.columns}
     payload = {k: v for k, v in data.items() if k in allowed}
+    if "symbol" not in payload:
+        raise ValueError("symbol is required")
+    if "underlying_currency" not in payload:
+        raise ValueError("underlying_currency is required")
+    payload["underlying_currency"] = _check_enum(
+        models.UnderlyingCurrency, payload["underlying_currency"], "underlying_currency"
+    )
     if "trade_type" not in payload:
         raise ValueError("trade_type is required")
-    payload["trade_type"] = _coerce_enum(
+    payload["trade_type"] = _check_enum(
         models.TradeType, payload["trade_type"], "trade_type"
     )
-
     if "trade_strategy" not in payload:
         raise ValueError("trade_strategy is required")
-    payload["trade_strategy"] = _coerce_enum(
+    payload["trade_strategy"] = _check_enum(
         models.TradeStrategy, payload["trade_strategy"], "trade_strategy"
     )
+    now = datetime.now(timezone.utc)
+    payload.pop("trade_time_utc", None)
+    payload["trade_time_utc"] = now
+    if "trade_date" not in payload or payload.get("trade_date") is None:
+        payload["trade_date"] = payload["trade_time_utc"].date()
     cycle_id = payload.get("cycle_id")
     user_id = payload.get("user_id")
-
+    if "quantity" not in payload:
+        raise ValueError("quantity is required")
+    if "price_cents" not in payload:
+        raise ValueError("price_cents is required")
+    if "commission_cents" not in payload:
+        payload["commission_cents"] = 0
+    quantity: int = payload["quantity"]
+    price_cents: int = payload["price_cents"]
+    commission_cents: int = payload["commission_cents"]
+    if "gross_cash_flow_cents" not in payload:
+        payload["gross_cash_flow_cents"] = -quantity * price_cents
+    if "net_cash_flow_cents" not in payload:
+        payload["net_cash_flow_cents"] = (
+            payload["gross_cash_flow_cents"] - commission_cents
+        )
+    if cycle_id is None:
+        cycle_id = create_cycle(
+            session,
+            {
+                "user_id": user_id,
+                "symbol": payload["symbol"],
+                "underlying_currency": payload["underlying_currency"],
+                "friendly_name": "Auto-created Cycle by trade "
+                + payload.get("friendly_name", ""),
+                "status": models.CycleStatus.OPEN,
+                "start_date": payload["trade_date"],
+            },
+        ).id
+        payload["cycle_id"] = cycle_id
     if cycle_id is not None:
         cycle = session.get(models.Cycles, cycle_id)
         if cycle is None:
-            pass  # TODO: create a cycle with basic info here
+            raise ValueError("cycle_id does not exist")
         else:
             if cycle.user_id != user_id:
                 raise ValueError("cycle.user_id does not match trade.user_id")
-    else:
-        raise ValueError("trade must have a cycle_id.")
     t = models.Trades(**payload)
     session.add(t)
     try:
@@ -61,3 +99,75 @@ def create_trade(session: Session, trade_data: Mapping) -> models.Trades:
         raise ValueError("create_trade integrity error") from e
     session.refresh(t)
     return t
+
+
+def get_trade_by_id(session: Session, trade_id: int) -> models.Trades | None:
+    return session.get(models.Trades, trade_id)
+
+
+def get_trade_by_user_id_and_friendly_name(
+    session: Session, user_id: int, friendly_name: str
+) -> models.Trades | None:
+    statement = select(models.Trades).where(
+        models.Trades.user_id == user_id,
+        models.Trades.friendly_name == friendly_name,
+    )
+    return session.exec(statement).first()
+
+
+# Cycles
+def create_cycle(session: Session, cycle_data: Mapping) -> models.Cycles:
+    if hasattr(cycle_data, "dict"):
+        data = cycle_data.dict(exclude_unset=True)
+    else:
+        data = dict(cycle_data)
+    allowed = {c.name for c in models.Cycles.__table__.columns}
+    payload = {k: v for k, v in data.items() if k in allowed}
+    if "user_id" not in payload:
+        raise ValueError("user_id is required")
+    if "symbol" not in payload:
+        raise ValueError("symbol is required")
+    if "underlying_currency" not in payload:
+        raise ValueError("underlying_currency is required")
+    payload["underlying_currency"] = _check_enum(
+        models.UnderlyingCurrency, payload["underlying_currency"], "underlying_currency"
+    )
+    if "status" not in payload:
+        raise ValueError("status is required")
+    payload["status"] = _check_enum(models.CycleStatus, payload["status"], "status")
+    if "start_date" not in payload:
+        raise ValueError("start_date is required")
+
+    c = models.Cycles(**payload)
+    session.add(c)
+    try:
+        session.flush()
+    except IntegrityError as e:
+        session.rollback()
+        raise ValueError("create_cycle integrity error") from e
+    session.refresh(c)
+    return c
+
+
+# Users
+def create_user(session: Session, user_data: Mapping) -> models.Users:
+    if hasattr(user_data, "dict"):
+        data = user_data.dict(exclude_unset=True)
+    else:
+        data = dict(user_data)
+    allowed = {c.name for c in models.Users.__table__.columns}
+    payload = {k: v for k, v in data.items() if k in allowed}
+    if "username" not in payload:
+        raise ValueError("username is required")
+    if "password_hash" not in payload:
+        raise ValueError("password_hash is required")
+
+    u = models.Users(**payload)
+    session.add(u)
+    try:
+        session.flush()
+    except IntegrityError as e:
+        session.rollback()
+        raise ValueError("create_user integrity error") from e
+    session.refresh(u)
+    return u

backend/trading_journal/models.py

@@ -1,8 +1,17 @@
 from datetime import date, datetime  # noqa: TC003
 from enum import Enum
 
-from sqlalchemy import Date, Text, UniqueConstraint
-from sqlmodel import Column, DateTime, Field, Relationship, SQLModel
+from sqlmodel import (
+    Column,
+    Date,
+    DateTime,
+    Field,
+    Integer,
+    Relationship,
+    SQLModel,
+    Text,
+    UniqueConstraint,
+)
 
 
 class TradeType(str, Enum):
@@ -36,6 +45,18 @@ class CycleStatus(str, Enum):
     CLOSED = "CLOSED"
 
 
+class UnderlyingCurrency(str, Enum):
+    EUR = "EUR"
+    USD = "USD"
+    GBP = "GBP"
+    JPY = "JPY"
+    AUD = "AUD"
+    CAD = "CAD"
+    CHF = "CHF"
+    NZD = "NZD"
+    CNY = "CNY"
+
+
 class FundingSource(str, Enum):
     CASH = "CASH"
     MARGIN = "MARGIN"
@@ -57,19 +78,22 @@ class Trades(SQLModel, table=True):
         default=None, sa_column=Column(Text, nullable=True)
     )
     symbol: str = Field(sa_column=Column(Text, nullable=False))
-    underlying_currency: str = Field(sa_column=Column(Text, nullable=False))
+    underlying_currency: UnderlyingCurrency = Field(
+        sa_column=Column(Text, nullable=False)
+    )
     trade_type: TradeType = Field(sa_column=Column(Text, nullable=False))
     trade_strategy: TradeStrategy = Field(sa_column=Column(Text, nullable=False))
+    trade_date: date = Field(sa_column=Column(Date, nullable=False))
     trade_time_utc: datetime = Field(
         sa_column=Column(DateTime(timezone=True), nullable=False)
     )
     expiry_date: date | None = Field(default=None, nullable=True)
     strike_price_cents: int | None = Field(default=None, nullable=True)
-    quantity: int
-    price_cents: int
-    gross_cash_flow_cents: int
-    commission_cents: int
-    net_cash_flow_cents: int
+    quantity: int = Field(sa_column=Column(Integer, nullable=False))
+    price_cents: int = Field(sa_column=Column(Integer, nullable=False))
+    gross_cash_flow_cents: int = Field(sa_column=Column(Integer, nullable=False))
+    commission_cents: int = Field(sa_column=Column(Integer, nullable=False))
+    net_cash_flow_cents: int = Field(sa_column=Column(Integer, nullable=False))
     cycle_id: int | None = Field(
         default=None, foreign_key="cycles.id", nullable=True, index=True
     )
@@ -90,7 +114,9 @@ class Cycles(SQLModel, table=True):
         default=None, sa_column=Column(Text, nullable=True)
     )
     symbol: str = Field(sa_column=Column(Text, nullable=False))
-    underlying_currency: str = Field(sa_column=Column(Text, nullable=False))
+    underlying_currency: UnderlyingCurrency = Field(
+        sa_column=Column(Text, nullable=False)
+    )
     status: CycleStatus = Field(sa_column=Column(Text, nullable=False))
     funding_source: FundingSource = Field(sa_column=Column(Text, nullable=True))
     capital_exposure_cents: int | None = Field(default=None, nullable=True)

backend/trading_journal/models_v1.py

@@ -36,6 +36,18 @@ class CycleStatus(str, Enum):
     CLOSED = "CLOSED"
 
 
+class UnderlyingCurrency(str, Enum):
+    EUR = "EUR"
+    USD = "USD"
+    GBP = "GBP"
+    JPY = "JPY"
+    AUD = "AUD"
+    CAD = "CAD"
+    CHF = "CHF"
+    NZD = "NZD"
+    CNY = "CNY"
+
+
 class FundingSource(str, Enum):
     CASH = "CASH"
     MARGIN = "MARGIN"
@@ -57,9 +69,12 @@ class Trades(SQLModel, table=True):
         default=None, sa_column=Column(Text, nullable=True)
     )
     symbol: str = Field(sa_column=Column(Text, nullable=False))
-    underlying_currency: str = Field(sa_column=Column(Text, nullable=False))
+    underlying_currency: UnderlyingCurrency = Field(
+        sa_column=Column(Text, nullable=False)
+    )
     trade_type: TradeType = Field(sa_column=Column(Text, nullable=False))
     trade_strategy: TradeStrategy = Field(sa_column=Column(Text, nullable=False))
+    trade_date: date = Field(sa_column=Column(Date, nullable=False))
     trade_time_utc: datetime = Field(
         sa_column=Column(DateTime(timezone=True), nullable=False)
     )
@@ -90,7 +105,9 @@ class Cycles(SQLModel, table=True):
         default=None, sa_column=Column(Text, nullable=True)
     )
     symbol: str = Field(sa_column=Column(Text, nullable=False))
-    underlying_currency: str = Field(sa_column=Column(Text, nullable=False))
+    underlying_currency: UnderlyingCurrency = Field(
+        sa_column=Column(Text, nullable=False)
+    )
     status: CycleStatus = Field(sa_column=Column(Text, nullable=False))
     funding_source: FundingSource = Field(sa_column=Column(Text, nullable=True))
     capital_exposure_cents: int | None = Field(default=None, nullable=True)