This commit is contained in:
@@ -52,7 +52,9 @@ def _data_to_dict(data: AnyModel) -> dict[str, AnyModel]:
|
||||
|
||||
|
||||
# Trades
|
||||
def create_trade(session: Session, trade_data: Mapping[str, Any] | BaseModel) -> models.Trades:
|
||||
def create_trade(
|
||||
session: Session, trade_data: Mapping[str, Any] | BaseModel
|
||||
) -> models.Trades:
|
||||
data = _data_to_dict(trade_data)
|
||||
allowed = _allowed_columns(models.Trades)
|
||||
payload = {k: v for k, v in data.items() if k in allowed}
|
||||
@@ -72,13 +74,19 @@ def create_trade(session: Session, trade_data: Mapping[str, Any] | BaseModel) ->
|
||||
raise ValueError("exchange.user_id does not match trade.user_id")
|
||||
if "underlying_currency" not in payload:
|
||||
raise ValueError("underlying_currency is required")
|
||||
payload["underlying_currency"] = _check_enum(models.UnderlyingCurrency, payload["underlying_currency"], "underlying_currency")
|
||||
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"] = _check_enum(models.TradeType, payload["trade_type"], "trade_type")
|
||||
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"] = _check_enum(models.TradeStrategy, payload["trade_strategy"], "trade_strategy")
|
||||
payload["trade_strategy"] = _check_enum(
|
||||
models.TradeStrategy, payload["trade_strategy"], "trade_strategy"
|
||||
)
|
||||
# trade_time_utc is the creation moment: always set to now (caller shouldn't provide)
|
||||
now = datetime.now(timezone.utc)
|
||||
payload.pop("trade_time_utc", None)
|
||||
@@ -105,7 +113,8 @@ def create_trade(session: Session, trade_data: Mapping[str, Any] | BaseModel) ->
|
||||
"symbol": payload["symbol"],
|
||||
"exchange_id": payload["exchange_id"],
|
||||
"underlying_currency": payload["underlying_currency"],
|
||||
"friendly_name": "Auto-created Cycle by trade " + payload.get("friendly_name", ""),
|
||||
"friendly_name": "Auto-created Cycle by trade "
|
||||
+ payload.get("friendly_name", ""),
|
||||
"status": models.CycleStatus.OPEN,
|
||||
"start_date": payload["trade_date"],
|
||||
}
|
||||
@@ -119,7 +128,9 @@ def create_trade(session: Session, trade_data: Mapping[str, Any] | BaseModel) ->
|
||||
|
||||
if cycle is None:
|
||||
raise ValueError("cycle_id does not exist")
|
||||
payload.pop("exchange_id", None) # ignore exchange_id if provided; use cycle's exchange_id
|
||||
payload.pop(
|
||||
"exchange_id", None
|
||||
) # ignore exchange_id if provided; use cycle's exchange_id
|
||||
payload["exchange_id"] = cycle.exchange_id
|
||||
if cycle.user_id != user_id:
|
||||
raise ValueError("cycle.user_id does not match trade.user_id")
|
||||
@@ -147,7 +158,9 @@ 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:
|
||||
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,
|
||||
@@ -162,7 +175,9 @@ def get_trades_by_user_id(session: Session, user_id: int) -> list[models.Trades]
|
||||
return list(session.exec(statement).all())
|
||||
|
||||
|
||||
def update_trade_friendly_name(session: Session, trade_id: int, friendly_name: str) -> models.Trades:
|
||||
def update_trade_friendly_name(
|
||||
session: Session, trade_id: int, friendly_name: str
|
||||
) -> models.Trades:
|
||||
trade: models.Trades | None = session.get(models.Trades, trade_id)
|
||||
if trade is None:
|
||||
raise ValueError("trade_id does not exist")
|
||||
@@ -210,7 +225,9 @@ def invalidate_trade(session: Session, trade_id: int) -> models.Trades:
|
||||
return trade
|
||||
|
||||
|
||||
def replace_trade(session: Session, old_trade_id: int, new_trade_data: Mapping[str, Any] | BaseModel) -> models.Trades:
|
||||
def replace_trade(
|
||||
session: Session, old_trade_id: int, new_trade_data: Mapping[str, Any] | BaseModel
|
||||
) -> models.Trades:
|
||||
invalidate_trade(session, old_trade_id)
|
||||
data = _data_to_dict(new_trade_data)
|
||||
data["replaced_by_trade_id"] = old_trade_id
|
||||
@@ -218,7 +235,9 @@ def replace_trade(session: Session, old_trade_id: int, new_trade_data: Mapping[s
|
||||
|
||||
|
||||
# Cycles
|
||||
def create_cycle(session: Session, cycle_data: Mapping[str, Any] | BaseModel) -> models.Cycles:
|
||||
def create_cycle(
|
||||
session: Session, cycle_data: Mapping[str, Any] | BaseModel
|
||||
) -> models.Cycles:
|
||||
data = _data_to_dict(cycle_data)
|
||||
allowed = _allowed_columns(models.Cycles)
|
||||
payload = {k: v for k, v in data.items() if k in allowed}
|
||||
@@ -236,7 +255,9 @@ def create_cycle(session: Session, cycle_data: Mapping[str, Any] | BaseModel) ->
|
||||
raise ValueError("exchange.user_id does not match cycle.user_id")
|
||||
if "underlying_currency" not in payload:
|
||||
raise ValueError("underlying_currency is required")
|
||||
payload["underlying_currency"] = _check_enum(models.UnderlyingCurrency, payload["underlying_currency"], "underlying_currency")
|
||||
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")
|
||||
@@ -268,7 +289,9 @@ def get_cycles_by_user_id(session: Session, user_id: int) -> list[models.Cycles]
|
||||
return list(session.exec(statement).all())
|
||||
|
||||
|
||||
def update_cycle(session: Session, cycle_id: int, update_data: Mapping[str, Any] | BaseModel) -> models.Cycles:
|
||||
def update_cycle(
|
||||
session: Session, cycle_id: int, update_data: Mapping[str, Any] | BaseModel
|
||||
) -> models.Cycles:
|
||||
cycle: models.Cycles | None = session.get(models.Cycles, cycle_id)
|
||||
if cycle is None:
|
||||
raise ValueError("cycle_id does not exist")
|
||||
@@ -304,7 +327,9 @@ def update_cycle(session: Session, cycle_id: int, update_data: Mapping[str, Any]
|
||||
|
||||
|
||||
# Cycle loan and interest
|
||||
def create_cycle_loan_event(session: Session, loan_data: Mapping[str, Any] | BaseModel) -> models.CycleLoanChangeEvents:
|
||||
def create_cycle_loan_event(
|
||||
session: Session, loan_data: Mapping[str, Any] | BaseModel
|
||||
) -> models.CycleLoanChangeEvents:
|
||||
data = _data_to_dict(loan_data)
|
||||
allowed = _allowed_columns(models.CycleLoanChangeEvents)
|
||||
payload = {k: v for k, v in data.items() if k in allowed}
|
||||
@@ -314,7 +339,9 @@ def create_cycle_loan_event(session: Session, loan_data: Mapping[str, Any] | Bas
|
||||
if cycle is None:
|
||||
raise ValueError("cycle_id does not exist")
|
||||
|
||||
payload["effective_date"] = payload.get("effective_date") or datetime.now(timezone.utc).date()
|
||||
payload["effective_date"] = (
|
||||
payload.get("effective_date") or datetime.now(timezone.utc).date()
|
||||
)
|
||||
payload["created_at"] = datetime.now(timezone.utc)
|
||||
cle = models.CycleLoanChangeEvents(**payload)
|
||||
session.add(cle)
|
||||
@@ -327,7 +354,9 @@ def create_cycle_loan_event(session: Session, loan_data: Mapping[str, Any] | Bas
|
||||
return cle
|
||||
|
||||
|
||||
def get_loan_events_by_cycle_id(session: Session, cycle_id: int) -> list[models.CycleLoanChangeEvents]:
|
||||
def get_loan_events_by_cycle_id(
|
||||
session: Session, cycle_id: int
|
||||
) -> list[models.CycleLoanChangeEvents]:
|
||||
eff_col = cast("ColumnElement", models.CycleLoanChangeEvents.effective_date)
|
||||
id_col = cast("ColumnElement", models.CycleLoanChangeEvents.id)
|
||||
statement = (
|
||||
@@ -340,7 +369,46 @@ def get_loan_events_by_cycle_id(session: Session, cycle_id: int) -> list[models.
|
||||
return list(session.exec(statement).all())
|
||||
|
||||
|
||||
def create_cycle_daily_accrual(session: Session, cycle_id: int, accrual_date: date, accrual_amount_cents: int) -> models.CycleDailyAccrual:
|
||||
def get_loan_event_by_cycle_id_and_effective_date(
|
||||
session: Session, cycle_id: int, effective_date: date
|
||||
) -> models.CycleLoanChangeEvents | None:
|
||||
statement = select(models.CycleLoanChangeEvents).where(
|
||||
models.CycleLoanChangeEvents.cycle_id == cycle_id,
|
||||
models.CycleLoanChangeEvents.effective_date == effective_date,
|
||||
)
|
||||
return session.exec(statement).first()
|
||||
|
||||
|
||||
def update_cycle_loan_event(
|
||||
session: Session, event_id: int, update_data: Mapping[str, Any] | BaseModel
|
||||
) -> models.CycleLoanChangeEvents:
|
||||
event: models.CycleLoanChangeEvents | None = session.get(
|
||||
models.CycleLoanChangeEvents, event_id
|
||||
)
|
||||
if event is None:
|
||||
raise ValueError("event_id does not exist")
|
||||
data = _data_to_dict(update_data)
|
||||
|
||||
allowed = _allowed_columns(models.CycleLoanChangeEvents)
|
||||
for k, v in data.items():
|
||||
if k in {"id", "cycle_id", "effective_date", "created_at"}:
|
||||
raise ValueError(f"field {k!r} is immutable")
|
||||
if k not in allowed:
|
||||
continue
|
||||
setattr(event, k, v)
|
||||
session.add(event)
|
||||
try:
|
||||
session.flush()
|
||||
except IntegrityError as e:
|
||||
session.rollback()
|
||||
raise ValueError("update_cycle_loan_event integrity error") from e
|
||||
session.refresh(event)
|
||||
return event
|
||||
|
||||
|
||||
def create_cycle_daily_accrual(
|
||||
session: Session, cycle_id: int, accrual_date: date, accrual_amount_cents: int
|
||||
) -> models.CycleDailyAccrual:
|
||||
cycle = session.get(models.Cycles, cycle_id)
|
||||
if cycle is None:
|
||||
raise ValueError("cycle_id does not exist")
|
||||
@@ -370,7 +438,9 @@ def create_cycle_daily_accrual(session: Session, cycle_id: int, accrual_date: da
|
||||
return row
|
||||
|
||||
|
||||
def get_cycle_daily_accruals_by_cycle_id(session: Session, cycle_id: int) -> list[models.CycleDailyAccrual]:
|
||||
def get_cycle_daily_accruals_by_cycle_id(
|
||||
session: Session, cycle_id: int
|
||||
) -> list[models.CycleDailyAccrual]:
|
||||
date_col = cast("ColumnElement", models.CycleDailyAccrual.accrual_date)
|
||||
statement = (
|
||||
select(models.CycleDailyAccrual)
|
||||
@@ -382,7 +452,9 @@ def get_cycle_daily_accruals_by_cycle_id(session: Session, cycle_id: int) -> lis
|
||||
return list(session.exec(statement).all())
|
||||
|
||||
|
||||
def get_cycle_daily_accrual_by_cycle_id_and_date(session: Session, cycle_id: int, accrual_date: date) -> models.CycleDailyAccrual | None:
|
||||
def get_cycle_daily_accrual_by_cycle_id_and_date(
|
||||
session: Session, cycle_id: int, accrual_date: date
|
||||
) -> models.CycleDailyAccrual | None:
|
||||
statement = select(models.CycleDailyAccrual).where(
|
||||
models.CycleDailyAccrual.cycle_id == cycle_id,
|
||||
models.CycleDailyAccrual.accrual_date == accrual_date,
|
||||
@@ -394,7 +466,9 @@ def get_cycle_daily_accrual_by_cycle_id_and_date(session: Session, cycle_id: int
|
||||
IMMUTABLE_EXCHANGE_FIELDS = {"id"}
|
||||
|
||||
|
||||
def create_exchange(session: Session, exchange_data: Mapping[str, Any] | BaseModel) -> models.Exchanges:
|
||||
def create_exchange(
|
||||
session: Session, exchange_data: Mapping[str, Any] | BaseModel
|
||||
) -> models.Exchanges:
|
||||
data = _data_to_dict(exchange_data)
|
||||
allowed = _allowed_columns(models.Exchanges)
|
||||
payload = {k: v for k, v in data.items() if k in allowed}
|
||||
@@ -416,7 +490,9 @@ def get_exchange_by_id(session: Session, exchange_id: int) -> models.Exchanges |
|
||||
return session.get(models.Exchanges, exchange_id)
|
||||
|
||||
|
||||
def get_exchange_by_name_and_user_id(session: Session, name: str, user_id: int) -> models.Exchanges | None:
|
||||
def get_exchange_by_name_and_user_id(
|
||||
session: Session, name: str, user_id: int
|
||||
) -> models.Exchanges | None:
|
||||
statement = select(models.Exchanges).where(
|
||||
models.Exchanges.name == name,
|
||||
models.Exchanges.user_id == user_id,
|
||||
@@ -429,14 +505,18 @@ def get_all_exchanges(session: Session) -> list[models.Exchanges]:
|
||||
return list(session.exec(statement).all())
|
||||
|
||||
|
||||
def get_all_exchanges_by_user_id(session: Session, user_id: int) -> list[models.Exchanges]:
|
||||
def get_all_exchanges_by_user_id(
|
||||
session: Session, user_id: int
|
||||
) -> list[models.Exchanges]:
|
||||
statement = select(models.Exchanges).where(
|
||||
models.Exchanges.user_id == user_id,
|
||||
)
|
||||
return list(session.exec(statement).all())
|
||||
|
||||
|
||||
def update_exchange(session: Session, exchange_id: int, update_data: Mapping[str, Any] | BaseModel) -> models.Exchanges:
|
||||
def update_exchange(
|
||||
session: Session, exchange_id: int, update_data: Mapping[str, Any] | BaseModel
|
||||
) -> models.Exchanges:
|
||||
exchange: models.Exchanges | None = session.get(models.Exchanges, exchange_id)
|
||||
if exchange is None:
|
||||
raise ValueError("exchange_id does not exist")
|
||||
@@ -473,7 +553,9 @@ def delete_exchange(session: Session, exchange_id: int) -> None:
|
||||
IMMUTABLE_USER_FIELDS = {"id", "username", "created_at"}
|
||||
|
||||
|
||||
def create_user(session: Session, user_data: Mapping[str, Any] | BaseModel) -> models.Users:
|
||||
def create_user(
|
||||
session: Session, user_data: Mapping[str, Any] | BaseModel
|
||||
) -> models.Users:
|
||||
data = _data_to_dict(user_data)
|
||||
allowed = _allowed_columns(models.Users)
|
||||
payload = {k: v for k, v in data.items() if k in allowed}
|
||||
@@ -504,7 +586,9 @@ def get_user_by_username(session: Session, username: str) -> models.Users | None
|
||||
return session.exec(statement).first()
|
||||
|
||||
|
||||
def update_user(session: Session, user_id: int, update_data: Mapping[str, Any] | BaseModel) -> models.Users:
|
||||
def update_user(
|
||||
session: Session, user_id: int, update_data: Mapping[str, Any] | BaseModel
|
||||
) -> models.Users:
|
||||
user: models.Users | None = session.get(models.Users, user_id)
|
||||
if user is None:
|
||||
raise ValueError("user_id does not exist")
|
||||
@@ -561,7 +645,9 @@ def create_login_session(
|
||||
return s
|
||||
|
||||
|
||||
def get_login_session_by_token_hash_and_user_id(session: Session, session_token_hash: str, user_id: int) -> models.Sessions | None:
|
||||
def get_login_session_by_token_hash_and_user_id(
|
||||
session: Session, session_token_hash: str, user_id: int
|
||||
) -> models.Sessions | None:
|
||||
statement = select(models.Sessions).where(
|
||||
models.Sessions.session_token_hash == session_token_hash,
|
||||
models.Sessions.user_id == user_id,
|
||||
@@ -571,7 +657,9 @@ def get_login_session_by_token_hash_and_user_id(session: Session, session_token_
|
||||
return session.exec(statement).first()
|
||||
|
||||
|
||||
def get_login_session_by_token_hash(session: Session, session_token_hash: str) -> models.Sessions | None:
|
||||
def get_login_session_by_token_hash(
|
||||
session: Session, session_token_hash: str
|
||||
) -> models.Sessions | None:
|
||||
statement = select(models.Sessions).where(
|
||||
models.Sessions.session_token_hash == session_token_hash,
|
||||
models.Sessions.expires_at > datetime.now(timezone.utc),
|
||||
@@ -583,7 +671,11 @@ def get_login_session_by_token_hash(session: Session, session_token_hash: str) -
|
||||
IMMUTABLE_SESSION_FIELDS = {"id", "user_id", "session_token_hash", "created_at"}
|
||||
|
||||
|
||||
def update_login_session(session: Session, session_token_hashed: str, update_session: Mapping[str, Any] | BaseModel) -> models.Sessions | None:
|
||||
def update_login_session(
|
||||
session: Session,
|
||||
session_token_hashed: str,
|
||||
update_session: Mapping[str, Any] | BaseModel,
|
||||
) -> models.Sessions | None:
|
||||
login_session: models.Sessions | None = session.exec(
|
||||
select(models.Sessions).where(
|
||||
models.Sessions.session_token_hash == session_token_hashed,
|
||||
|
||||
@@ -1,11 +1,15 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from datetime import date, datetime # noqa: TC003
|
||||
from datetime import date, datetime
|
||||
|
||||
from pydantic import BaseModel
|
||||
from sqlmodel import SQLModel
|
||||
|
||||
from trading_journal.models import TradeStrategy, TradeType, UnderlyingCurrency # noqa: TC001
|
||||
from trading_journal.models import (
|
||||
TradeStrategy,
|
||||
TradeType,
|
||||
UnderlyingCurrency,
|
||||
)
|
||||
|
||||
|
||||
class UserBase(SQLModel):
|
||||
@@ -90,6 +94,32 @@ class CycleRead(CycleCreate):
|
||||
id: int
|
||||
|
||||
|
||||
class CycleLoanChangeEventBase(SQLModel):
|
||||
cycle_id: int
|
||||
effective_date: date
|
||||
loan_amount_cents: int | None = None
|
||||
loan_interest_rate_tenth_bps: int | None = None
|
||||
related_trade_id: int | None = None
|
||||
notes: str | None = None
|
||||
created_at: datetime
|
||||
|
||||
|
||||
class CycleLoanChangeEventCreate(CycleLoanChangeEventBase):
|
||||
created_at: datetime
|
||||
|
||||
|
||||
class CycleLoanChangeEventRead(CycleLoanChangeEventBase):
|
||||
id: int
|
||||
created_at: datetime
|
||||
|
||||
|
||||
class CycleInterestAccrualBase(SQLModel):
|
||||
cycle_id: int
|
||||
accrual_date: date
|
||||
accrual_amount_cents: int
|
||||
created_at: datetime
|
||||
|
||||
|
||||
class TradeBase(SQLModel):
|
||||
friendly_name: str | None = None
|
||||
symbol: str
|
||||
|
||||
@@ -69,33 +69,51 @@ class FundingSource(str, Enum):
|
||||
|
||||
class Trades(SQLModel, table=True):
|
||||
__tablename__ = "trades" # type: ignore[attr-defined]
|
||||
__table_args__ = (UniqueConstraint("user_id", "friendly_name", name="uq_trades_user_friendly_name"),)
|
||||
__table_args__ = (
|
||||
UniqueConstraint(
|
||||
"user_id", "friendly_name", name="uq_trades_user_friendly_name"
|
||||
),
|
||||
)
|
||||
|
||||
id: int | None = Field(default=None, primary_key=True)
|
||||
user_id: int = Field(foreign_key="users.id", nullable=False, index=True)
|
||||
# allow null while user may omit friendly_name; uniqueness enforced per-user by constraint
|
||||
friendly_name: str | None = Field(default=None, sa_column=Column(Text, nullable=True))
|
||||
friendly_name: str | None = Field(
|
||||
default=None, sa_column=Column(Text, nullable=True)
|
||||
)
|
||||
symbol: str = Field(sa_column=Column(Text, nullable=False))
|
||||
exchange_id: int = Field(foreign_key="exchanges.id", nullable=False, index=True)
|
||||
exchange: "Exchanges" = Relationship(back_populates="trades")
|
||||
underlying_currency: UnderlyingCurrency = 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))
|
||||
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 = Field(sa_column=Column(Integer, nullable=False))
|
||||
quantity_multiplier: int = Field(sa_column=Column(Integer, nullable=False), default=1)
|
||||
quantity_multiplier: int = Field(
|
||||
sa_column=Column(Integer, nullable=False), default=1
|
||||
)
|
||||
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))
|
||||
is_invalidated: bool = Field(default=False, nullable=False)
|
||||
invalidated_at: datetime | None = Field(default=None, sa_column=Column(DateTime(timezone=True), nullable=True))
|
||||
replaced_by_trade_id: int | None = Field(default=None, foreign_key="trades.id", nullable=True)
|
||||
invalidated_at: datetime | None = Field(
|
||||
default=None, sa_column=Column(DateTime(timezone=True), nullable=True)
|
||||
)
|
||||
replaced_by_trade_id: int | None = Field(
|
||||
default=None, foreign_key="trades.id", nullable=True
|
||||
)
|
||||
notes: str | None = Field(default=None, sa_column=Column(Text, nullable=True))
|
||||
cycle_id: int | None = Field(default=None, foreign_key="cycles.id", nullable=True, index=True)
|
||||
cycle_id: int | None = Field(
|
||||
default=None, foreign_key="cycles.id", nullable=True, index=True
|
||||
)
|
||||
|
||||
cycle: "Cycles" = Relationship(back_populates="trades")
|
||||
|
||||
@@ -107,15 +125,23 @@ class Trades(SQLModel, table=True):
|
||||
|
||||
class Cycles(SQLModel, table=True):
|
||||
__tablename__ = "cycles" # type: ignore[attr-defined]
|
||||
__table_args__ = (UniqueConstraint("user_id", "friendly_name", name="uq_cycles_user_friendly_name"),)
|
||||
__table_args__ = (
|
||||
UniqueConstraint(
|
||||
"user_id", "friendly_name", name="uq_cycles_user_friendly_name"
|
||||
),
|
||||
)
|
||||
|
||||
id: int | None = Field(default=None, primary_key=True)
|
||||
user_id: int = Field(foreign_key="users.id", nullable=False, index=True)
|
||||
friendly_name: str | None = Field(default=None, sa_column=Column(Text, nullable=True))
|
||||
friendly_name: str | None = Field(
|
||||
default=None, sa_column=Column(Text, nullable=True)
|
||||
)
|
||||
symbol: str = Field(sa_column=Column(Text, nullable=False))
|
||||
exchange_id: int = Field(foreign_key="exchanges.id", nullable=False, index=True)
|
||||
exchange: "Exchanges" = Relationship(back_populates="cycles")
|
||||
underlying_currency: UnderlyingCurrency = 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)
|
||||
@@ -127,23 +153,51 @@ class Cycles(SQLModel, table=True):
|
||||
loan_amount_cents: int | None = Field(default=None, nullable=True)
|
||||
loan_interest_rate_tenth_bps: int | None = Field(default=None, nullable=True)
|
||||
|
||||
latest_interest_accrued_date: date | None = Field(default=None, sa_column=Column(Date, nullable=True))
|
||||
total_accrued_amount_cents: int = Field(default=0, sa_column=Column(Integer, nullable=False))
|
||||
latest_interest_accrued_date: date | None = Field(
|
||||
default=None, sa_column=Column(Date, nullable=True)
|
||||
)
|
||||
total_accrued_amount_cents: int = Field(
|
||||
default=0, sa_column=Column(Integer, nullable=False)
|
||||
)
|
||||
|
||||
loan_change_events: list["CycleLoanChangeEvents"] = Relationship(back_populates="cycle")
|
||||
loan_change_events: list["CycleLoanChangeEvents"] = Relationship(
|
||||
back_populates="cycle"
|
||||
)
|
||||
daily_accruals: list["CycleDailyAccrual"] = Relationship(back_populates="cycle")
|
||||
|
||||
|
||||
class CycleLoanChangeEvents(SQLModel, table=True):
|
||||
__tablename__ = "cycle_loan_change_events" # type: ignore[attr-defined]
|
||||
__table_args__ = (
|
||||
UniqueConstraint(
|
||||
"cycle_id", "effective_date", name="uq_cycle_loan_change_cycle_date"
|
||||
),
|
||||
)
|
||||
|
||||
id: int | None = Field(default=None, primary_key=True)
|
||||
cycle_id: int = Field(sa_column=Column(Integer, ForeignKey("cycles.id", ondelete="CASCADE"), nullable=False, index=True))
|
||||
cycle_id: int = Field(
|
||||
sa_column=Column(
|
||||
Integer,
|
||||
ForeignKey("cycles.id", ondelete="CASCADE"),
|
||||
nullable=False,
|
||||
index=True,
|
||||
)
|
||||
)
|
||||
effective_date: date = Field(sa_column=Column(Date, nullable=False))
|
||||
loan_amount_cents: int | None = Field(default=None, sa_column=Column(Integer, nullable=True))
|
||||
loan_interest_rate_tenth_bps: int | None = Field(default=None, sa_column=Column(Integer, nullable=True))
|
||||
related_trade_id: int | None = Field(default=None, sa_column=Column(Integer, ForeignKey("trades.id"), nullable=True, unique=True))
|
||||
loan_amount_cents: int | None = Field(
|
||||
default=None, sa_column=Column(Integer, nullable=True)
|
||||
)
|
||||
loan_interest_rate_tenth_bps: int | None = Field(
|
||||
default=None, sa_column=Column(Integer, nullable=True)
|
||||
)
|
||||
related_trade_id: int | None = Field(
|
||||
default=None,
|
||||
sa_column=Column(Integer, ForeignKey("trades.id"), nullable=True, unique=True),
|
||||
) # Not used for now.
|
||||
notes: str | None = Field(default=None, sa_column=Column(Text, nullable=True))
|
||||
created_at: datetime = Field(sa_column=Column(DateTime(timezone=True), nullable=False))
|
||||
created_at: datetime = Field(
|
||||
sa_column=Column(DateTime(timezone=True), nullable=False)
|
||||
)
|
||||
|
||||
cycle: "Cycles" = Relationship(back_populates="loan_change_events")
|
||||
trade: Optional["Trades"] = Relationship(back_populates="related_loan_change_event")
|
||||
@@ -151,20 +205,35 @@ class CycleLoanChangeEvents(SQLModel, table=True):
|
||||
|
||||
class CycleDailyAccrual(SQLModel, table=True):
|
||||
__tablename__ = "cycle_daily_accrual" # type: ignore[attr-defined]
|
||||
__table_args__ = (UniqueConstraint("cycle_id", "accrual_date", name="uq_cycle_daily_accruals_cycle_date"),)
|
||||
__table_args__ = (
|
||||
UniqueConstraint(
|
||||
"cycle_id", "accrual_date", name="uq_cycle_daily_accruals_cycle_date"
|
||||
),
|
||||
)
|
||||
|
||||
id: int | None = Field(default=None, primary_key=True)
|
||||
cycle_id: int = Field(sa_column=Column(Integer, ForeignKey("cycles.id", ondelete="CASCADE"), nullable=False, index=True))
|
||||
cycle_id: int = Field(
|
||||
sa_column=Column(
|
||||
Integer,
|
||||
ForeignKey("cycles.id", ondelete="CASCADE"),
|
||||
nullable=False,
|
||||
index=True,
|
||||
)
|
||||
)
|
||||
accrual_date: date = Field(sa_column=Column(Date, nullable=False))
|
||||
accrual_amount_cents: int = Field(sa_column=Column(Integer, nullable=False))
|
||||
created_at: datetime = Field(sa_column=Column(DateTime(timezone=True), nullable=False))
|
||||
created_at: datetime = Field(
|
||||
sa_column=Column(DateTime(timezone=True), nullable=False)
|
||||
)
|
||||
|
||||
cycle: "Cycles" = Relationship(back_populates="daily_accruals")
|
||||
|
||||
|
||||
class Exchanges(SQLModel, table=True):
|
||||
__tablename__ = "exchanges" # type: ignore[attr-defined]
|
||||
__table_args__ = (UniqueConstraint("user_id", "name", name="uq_exchanges_user_name"),)
|
||||
__table_args__ = (
|
||||
UniqueConstraint("user_id", "name", name="uq_exchanges_user_name"),
|
||||
)
|
||||
id: int | None = Field(default=None, primary_key=True)
|
||||
user_id: int = Field(foreign_key="users.id", nullable=False, index=True)
|
||||
name: str = Field(sa_column=Column(Text, nullable=False))
|
||||
@@ -190,10 +259,18 @@ class Sessions(SQLModel, table=True):
|
||||
id: int | None = Field(default=None, primary_key=True)
|
||||
user_id: int = Field(foreign_key="users.id", nullable=False, index=True)
|
||||
session_token_hash: str = Field(sa_column=Column(Text, nullable=False, unique=True))
|
||||
created_at: datetime = Field(sa_column=Column(DateTime(timezone=True), nullable=False))
|
||||
expires_at: datetime = Field(sa_column=Column(DateTime(timezone=True), nullable=False, index=True))
|
||||
last_seen_at: datetime | None = Field(sa_column=Column(DateTime(timezone=True), nullable=True))
|
||||
last_used_ip: str | None = Field(default=None, sa_column=Column(Text, nullable=True))
|
||||
created_at: datetime = Field(
|
||||
sa_column=Column(DateTime(timezone=True), nullable=False)
|
||||
)
|
||||
expires_at: datetime = Field(
|
||||
sa_column=Column(DateTime(timezone=True), nullable=False, index=True)
|
||||
)
|
||||
last_seen_at: datetime | None = Field(
|
||||
sa_column=Column(DateTime(timezone=True), nullable=True)
|
||||
)
|
||||
last_used_ip: str | None = Field(
|
||||
default=None, sa_column=Column(Text, nullable=True)
|
||||
)
|
||||
user_agent: str | None = Field(default=None, sa_column=Column(Text, nullable=True))
|
||||
device_name: str | None = Field(default=None, sa_column=Column(Text, nullable=True))
|
||||
user: "Users" = Relationship(back_populates="sessions")
|
||||
|
||||
@@ -69,33 +69,51 @@ class FundingSource(str, Enum):
|
||||
|
||||
class Trades(SQLModel, table=True):
|
||||
__tablename__ = "trades" # type: ignore[attr-defined]
|
||||
__table_args__ = (UniqueConstraint("user_id", "friendly_name", name="uq_trades_user_friendly_name"),)
|
||||
__table_args__ = (
|
||||
UniqueConstraint(
|
||||
"user_id", "friendly_name", name="uq_trades_user_friendly_name"
|
||||
),
|
||||
)
|
||||
|
||||
id: int | None = Field(default=None, primary_key=True)
|
||||
user_id: int = Field(foreign_key="users.id", nullable=False, index=True)
|
||||
# allow null while user may omit friendly_name; uniqueness enforced per-user by constraint
|
||||
friendly_name: str | None = Field(default=None, sa_column=Column(Text, nullable=True))
|
||||
friendly_name: str | None = Field(
|
||||
default=None, sa_column=Column(Text, nullable=True)
|
||||
)
|
||||
symbol: str = Field(sa_column=Column(Text, nullable=False))
|
||||
exchange_id: int = Field(foreign_key="exchanges.id", nullable=False, index=True)
|
||||
exchange: "Exchanges" = Relationship(back_populates="trades")
|
||||
underlying_currency: UnderlyingCurrency = 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))
|
||||
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 = Field(sa_column=Column(Integer, nullable=False))
|
||||
quantity_multiplier: int = Field(sa_column=Column(Integer, nullable=False), default=1)
|
||||
quantity_multiplier: int = Field(
|
||||
sa_column=Column(Integer, nullable=False), default=1
|
||||
)
|
||||
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))
|
||||
is_invalidated: bool = Field(default=False, nullable=False)
|
||||
invalidated_at: datetime | None = Field(default=None, sa_column=Column(DateTime(timezone=True), nullable=True))
|
||||
replaced_by_trade_id: int | None = Field(default=None, foreign_key="trades.id", nullable=True)
|
||||
invalidated_at: datetime | None = Field(
|
||||
default=None, sa_column=Column(DateTime(timezone=True), nullable=True)
|
||||
)
|
||||
replaced_by_trade_id: int | None = Field(
|
||||
default=None, foreign_key="trades.id", nullable=True
|
||||
)
|
||||
notes: str | None = Field(default=None, sa_column=Column(Text, nullable=True))
|
||||
cycle_id: int | None = Field(default=None, foreign_key="cycles.id", nullable=True, index=True)
|
||||
cycle_id: int | None = Field(
|
||||
default=None, foreign_key="cycles.id", nullable=True, index=True
|
||||
)
|
||||
|
||||
cycle: "Cycles" = Relationship(back_populates="trades")
|
||||
|
||||
@@ -107,15 +125,23 @@ class Trades(SQLModel, table=True):
|
||||
|
||||
class Cycles(SQLModel, table=True):
|
||||
__tablename__ = "cycles" # type: ignore[attr-defined]
|
||||
__table_args__ = (UniqueConstraint("user_id", "friendly_name", name="uq_cycles_user_friendly_name"),)
|
||||
__table_args__ = (
|
||||
UniqueConstraint(
|
||||
"user_id", "friendly_name", name="uq_cycles_user_friendly_name"
|
||||
),
|
||||
)
|
||||
|
||||
id: int | None = Field(default=None, primary_key=True)
|
||||
user_id: int = Field(foreign_key="users.id", nullable=False, index=True)
|
||||
friendly_name: str | None = Field(default=None, sa_column=Column(Text, nullable=True))
|
||||
friendly_name: str | None = Field(
|
||||
default=None, sa_column=Column(Text, nullable=True)
|
||||
)
|
||||
symbol: str = Field(sa_column=Column(Text, nullable=False))
|
||||
exchange_id: int = Field(foreign_key="exchanges.id", nullable=False, index=True)
|
||||
exchange: "Exchanges" = Relationship(back_populates="cycles")
|
||||
underlying_currency: UnderlyingCurrency = 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)
|
||||
@@ -127,23 +153,51 @@ class Cycles(SQLModel, table=True):
|
||||
loan_amount_cents: int | None = Field(default=None, nullable=True)
|
||||
loan_interest_rate_tenth_bps: int | None = Field(default=None, nullable=True)
|
||||
|
||||
latest_interest_accrued_date: date | None = Field(default=None, sa_column=Column(Date, nullable=True))
|
||||
total_accrued_amount_cents: int = Field(default=0, sa_column=Column(Integer, nullable=False))
|
||||
latest_interest_accrued_date: date | None = Field(
|
||||
default=None, sa_column=Column(Date, nullable=True)
|
||||
)
|
||||
total_accrued_amount_cents: int = Field(
|
||||
default=0, sa_column=Column(Integer, nullable=False)
|
||||
)
|
||||
|
||||
loan_change_events: list["CycleLoanChangeEvents"] = Relationship(back_populates="cycle")
|
||||
loan_change_events: list["CycleLoanChangeEvents"] = Relationship(
|
||||
back_populates="cycle"
|
||||
)
|
||||
daily_accruals: list["CycleDailyAccrual"] = Relationship(back_populates="cycle")
|
||||
|
||||
|
||||
class CycleLoanChangeEvents(SQLModel, table=True):
|
||||
__tablename__ = "cycle_loan_change_events" # type: ignore[attr-defined]
|
||||
__table_args__ = (
|
||||
UniqueConstraint(
|
||||
"cycle_id", "effective_date", name="uq_cycle_loan_change_cycle_date"
|
||||
),
|
||||
)
|
||||
|
||||
id: int | None = Field(default=None, primary_key=True)
|
||||
cycle_id: int = Field(sa_column=Column(Integer, ForeignKey("cycles.id", ondelete="CASCADE"), nullable=False, index=True))
|
||||
cycle_id: int = Field(
|
||||
sa_column=Column(
|
||||
Integer,
|
||||
ForeignKey("cycles.id", ondelete="CASCADE"),
|
||||
nullable=False,
|
||||
index=True,
|
||||
)
|
||||
)
|
||||
effective_date: date = Field(sa_column=Column(Date, nullable=False))
|
||||
loan_amount_cents: int | None = Field(default=None, sa_column=Column(Integer, nullable=True))
|
||||
loan_interest_rate_tenth_bps: int | None = Field(default=None, sa_column=Column(Integer, nullable=True))
|
||||
related_trade_id: int | None = Field(default=None, sa_column=Column(Integer, ForeignKey("trades.id"), nullable=True, unique=True))
|
||||
loan_amount_cents: int | None = Field(
|
||||
default=None, sa_column=Column(Integer, nullable=True)
|
||||
)
|
||||
loan_interest_rate_tenth_bps: int | None = Field(
|
||||
default=None, sa_column=Column(Integer, nullable=True)
|
||||
)
|
||||
related_trade_id: int | None = Field(
|
||||
default=None,
|
||||
sa_column=Column(Integer, ForeignKey("trades.id"), nullable=True, unique=True),
|
||||
) # Not used for now.
|
||||
notes: str | None = Field(default=None, sa_column=Column(Text, nullable=True))
|
||||
created_at: datetime = Field(sa_column=Column(DateTime(timezone=True), nullable=False))
|
||||
created_at: datetime = Field(
|
||||
sa_column=Column(DateTime(timezone=True), nullable=False)
|
||||
)
|
||||
|
||||
cycle: "Cycles" = Relationship(back_populates="loan_change_events")
|
||||
trade: Optional["Trades"] = Relationship(back_populates="related_loan_change_event")
|
||||
@@ -151,20 +205,35 @@ class CycleLoanChangeEvents(SQLModel, table=True):
|
||||
|
||||
class CycleDailyAccrual(SQLModel, table=True):
|
||||
__tablename__ = "cycle_daily_accrual" # type: ignore[attr-defined]
|
||||
__table_args__ = (UniqueConstraint("cycle_id", "accrual_date", name="uq_cycle_daily_accruals_cycle_date"),)
|
||||
__table_args__ = (
|
||||
UniqueConstraint(
|
||||
"cycle_id", "accrual_date", name="uq_cycle_daily_accruals_cycle_date"
|
||||
),
|
||||
)
|
||||
|
||||
id: int | None = Field(default=None, primary_key=True)
|
||||
cycle_id: int = Field(sa_column=Column(Integer, ForeignKey("cycles.id", ondelete="CASCADE"), nullable=False, index=True))
|
||||
cycle_id: int = Field(
|
||||
sa_column=Column(
|
||||
Integer,
|
||||
ForeignKey("cycles.id", ondelete="CASCADE"),
|
||||
nullable=False,
|
||||
index=True,
|
||||
)
|
||||
)
|
||||
accrual_date: date = Field(sa_column=Column(Date, nullable=False))
|
||||
accrual_amount_cents: int = Field(sa_column=Column(Integer, nullable=False))
|
||||
created_at: datetime = Field(sa_column=Column(DateTime(timezone=True), nullable=False))
|
||||
created_at: datetime = Field(
|
||||
sa_column=Column(DateTime(timezone=True), nullable=False)
|
||||
)
|
||||
|
||||
cycle: "Cycles" = Relationship(back_populates="daily_accruals")
|
||||
|
||||
|
||||
class Exchanges(SQLModel, table=True):
|
||||
__tablename__ = "exchanges" # type: ignore[attr-defined]
|
||||
__table_args__ = (UniqueConstraint("user_id", "name", name="uq_exchanges_user_name"),)
|
||||
__table_args__ = (
|
||||
UniqueConstraint("user_id", "name", name="uq_exchanges_user_name"),
|
||||
)
|
||||
id: int | None = Field(default=None, primary_key=True)
|
||||
user_id: int = Field(foreign_key="users.id", nullable=False, index=True)
|
||||
name: str = Field(sa_column=Column(Text, nullable=False))
|
||||
@@ -190,10 +259,18 @@ class Sessions(SQLModel, table=True):
|
||||
id: int | None = Field(default=None, primary_key=True)
|
||||
user_id: int = Field(foreign_key="users.id", nullable=False, index=True)
|
||||
session_token_hash: str = Field(sa_column=Column(Text, nullable=False, unique=True))
|
||||
created_at: datetime = Field(sa_column=Column(DateTime(timezone=True), nullable=False))
|
||||
expires_at: datetime = Field(sa_column=Column(DateTime(timezone=True), nullable=False, index=True))
|
||||
last_seen_at: datetime | None = Field(sa_column=Column(DateTime(timezone=True), nullable=True))
|
||||
last_used_ip: str | None = Field(default=None, sa_column=Column(Text, nullable=True))
|
||||
created_at: datetime = Field(
|
||||
sa_column=Column(DateTime(timezone=True), nullable=False)
|
||||
)
|
||||
expires_at: datetime = Field(
|
||||
sa_column=Column(DateTime(timezone=True), nullable=False, index=True)
|
||||
)
|
||||
last_seen_at: datetime | None = Field(
|
||||
sa_column=Column(DateTime(timezone=True), nullable=True)
|
||||
)
|
||||
last_used_ip: str | None = Field(
|
||||
default=None, sa_column=Column(Text, nullable=True)
|
||||
)
|
||||
user_agent: str | None = Field(default=None, sa_column=Column(Text, nullable=True))
|
||||
device_name: str | None = Field(default=None, sa_column=Column(Text, nullable=True))
|
||||
user: "Users" = Relationship(back_populates="sessions")
|
||||
|
||||
@@ -13,6 +13,7 @@ from trading_journal import crud, security
|
||||
from trading_journal.dto import (
|
||||
CycleBase,
|
||||
CycleCreate,
|
||||
CycleLoanChangeEventBase,
|
||||
CycleRead,
|
||||
CycleUpdate,
|
||||
ExchangesBase,
|
||||
@@ -26,6 +27,17 @@ from trading_journal.dto import (
|
||||
UserLogin,
|
||||
UserRead,
|
||||
)
|
||||
from trading_journal.service_error import (
|
||||
CycleLoanEventExistsError,
|
||||
CycleNotFoundError,
|
||||
ExchangeAlreadyExistsError,
|
||||
ExchangeNotFoundError,
|
||||
InvalidCycleDataError,
|
||||
InvalidTradeDataError,
|
||||
ServiceError,
|
||||
TradeNotFoundError,
|
||||
UserAlreadyExistsError,
|
||||
)
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from sqlmodel import Session
|
||||
@@ -44,7 +56,9 @@ logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class AuthMiddleWare(BaseHTTPMiddleware):
|
||||
async def dispatch(self, request: Request, call_next: RequestResponseEndpoint) -> Response: # noqa: PLR0911
|
||||
async def dispatch(
|
||||
self, request: Request, call_next: RequestResponseEndpoint
|
||||
) -> Response: # noqa: PLR0911
|
||||
if request.url.path in EXCEPT_PATHS:
|
||||
return await call_next(request)
|
||||
|
||||
@@ -62,22 +76,44 @@ class AuthMiddleWare(BaseHTTPMiddleware):
|
||||
|
||||
db_factory: Database | None = getattr(request.app.state, "db_factory", None)
|
||||
if db_factory is None:
|
||||
return JSONResponse(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, content={"detail": "db factory not configured"})
|
||||
return JSONResponse(
|
||||
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
||||
content={"detail": "db factory not configured"},
|
||||
)
|
||||
try:
|
||||
with db_factory.get_session_ctx_manager() as request_session:
|
||||
hashed_token = security.hash_session_token_sha256(token)
|
||||
request.state.db_session = request_session
|
||||
login_session: Sessions | None = crud.get_login_session_by_token_hash(request_session, hashed_token)
|
||||
login_session: Sessions | None = crud.get_login_session_by_token_hash(
|
||||
request_session, hashed_token
|
||||
)
|
||||
if not login_session:
|
||||
return JSONResponse(status_code=status.HTTP_401_UNAUTHORIZED, content={"detail": "Unauthorized"})
|
||||
session_expires_utc = login_session.expires_at.replace(tzinfo=timezone.utc)
|
||||
return JSONResponse(
|
||||
status_code=status.HTTP_401_UNAUTHORIZED,
|
||||
content={"detail": "Unauthorized"},
|
||||
)
|
||||
session_expires_utc = login_session.expires_at.replace(
|
||||
tzinfo=timezone.utc
|
||||
)
|
||||
if session_expires_utc < datetime.now(timezone.utc):
|
||||
crud.delete_login_session(request_session, login_session.session_token_hash)
|
||||
return JSONResponse(status_code=status.HTTP_401_UNAUTHORIZED, content={"detail": "Unauthorized"})
|
||||
crud.delete_login_session(
|
||||
request_session, login_session.session_token_hash
|
||||
)
|
||||
return JSONResponse(
|
||||
status_code=status.HTTP_401_UNAUTHORIZED,
|
||||
content={"detail": "Unauthorized"},
|
||||
)
|
||||
if login_session.user.is_active is False:
|
||||
return JSONResponse(status_code=status.HTTP_401_UNAUTHORIZED, content={"detail": "Unauthorized"})
|
||||
if session_expires_utc - datetime.now(timezone.utc) < timedelta(seconds=3600):
|
||||
updated_expiry = datetime.now(timezone.utc) + timedelta(seconds=settings.settings.session_expiry_seconds)
|
||||
return JSONResponse(
|
||||
status_code=status.HTTP_401_UNAUTHORIZED,
|
||||
content={"detail": "Unauthorized"},
|
||||
)
|
||||
if session_expires_utc - datetime.now(timezone.utc) < timedelta(
|
||||
seconds=3600
|
||||
):
|
||||
updated_expiry = datetime.now(timezone.utc) + timedelta(
|
||||
seconds=settings.settings.session_expiry_seconds
|
||||
)
|
||||
else:
|
||||
updated_expiry = session_expires_utc
|
||||
updated_session: SessionsUpdate = SessionsUpdate(
|
||||
@@ -88,46 +124,19 @@ class AuthMiddleWare(BaseHTTPMiddleware):
|
||||
)
|
||||
user_id = login_session.user_id
|
||||
request.state.user_id = user_id
|
||||
crud.update_login_session(request_session, hashed_token, update_session=updated_session)
|
||||
crud.update_login_session(
|
||||
request_session, hashed_token, update_session=updated_session
|
||||
)
|
||||
except Exception:
|
||||
logger.exception("Failed to authenticate user: \n")
|
||||
return JSONResponse(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, content={"detail": "Internal server error"})
|
||||
return JSONResponse(
|
||||
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
||||
content={"detail": "Internal server error"},
|
||||
)
|
||||
|
||||
return await call_next(request)
|
||||
|
||||
|
||||
class ServiceError(Exception):
|
||||
pass
|
||||
|
||||
|
||||
class UserAlreadyExistsError(ServiceError):
|
||||
pass
|
||||
|
||||
|
||||
class ExchangeAlreadyExistsError(ServiceError):
|
||||
pass
|
||||
|
||||
|
||||
class ExchangeNotFoundError(ServiceError):
|
||||
pass
|
||||
|
||||
|
||||
class CycleNotFoundError(ServiceError):
|
||||
pass
|
||||
|
||||
|
||||
class TradeNotFoundError(ServiceError):
|
||||
pass
|
||||
|
||||
|
||||
class InvalidTradeDataError(ServiceError):
|
||||
pass
|
||||
|
||||
|
||||
class InvalidCycleDataError(ServiceError):
|
||||
pass
|
||||
|
||||
|
||||
# User service
|
||||
def register_user_service(db_session: Session, user_in: UserCreate) -> UserRead:
|
||||
if crud.get_user_by_username(db_session, user_in.username):
|
||||
@@ -151,7 +160,9 @@ def register_user_service(db_session: Session, user_in: UserCreate) -> UserRead:
|
||||
return user
|
||||
|
||||
|
||||
def authenticate_user_service(db_session: Session, user_in: UserLogin) -> tuple[SessionsCreate, str] | None:
|
||||
def authenticate_user_service(
|
||||
db_session: Session, user_in: UserLogin
|
||||
) -> tuple[SessionsCreate, str] | None:
|
||||
user = crud.get_user_by_username(db_session, user_in.username)
|
||||
if not user:
|
||||
return None
|
||||
@@ -176,10 +187,14 @@ def authenticate_user_service(db_session: Session, user_in: UserLogin) -> tuple[
|
||||
|
||||
|
||||
# Exchanges service
|
||||
def create_exchange_service(db_session: Session, user_id: int, name: str, notes: str | None) -> ExchangesCreate:
|
||||
def create_exchange_service(
|
||||
db_session: Session, user_id: int, name: str, notes: str | None
|
||||
) -> ExchangesCreate:
|
||||
existing_exchange = crud.get_exchange_by_name_and_user_id(db_session, name, user_id)
|
||||
if existing_exchange:
|
||||
raise ExchangeAlreadyExistsError("Exchange with the same name already exists for this user")
|
||||
raise ExchangeAlreadyExistsError(
|
||||
"Exchange with the same name already exists for this user"
|
||||
)
|
||||
exchange_data = ExchangesCreate(
|
||||
user_id=user_id,
|
||||
name=name,
|
||||
@@ -198,12 +213,20 @@ def create_exchange_service(db_session: Session, user_id: int, name: str, notes:
|
||||
return exchange_dto
|
||||
|
||||
|
||||
def get_exchanges_by_user_service(db_session: Session, user_id: int) -> list[ExchangesRead]:
|
||||
def get_exchanges_by_user_service(
|
||||
db_session: Session, user_id: int
|
||||
) -> list[ExchangesRead]:
|
||||
exchanges = crud.get_all_exchanges_by_user_id(db_session, user_id)
|
||||
return [ExchangesRead.model_validate(exchange) for exchange in exchanges]
|
||||
|
||||
|
||||
def update_exchanges_service(db_session: Session, user_id: int, exchange_id: int, name: str | None, notes: str | None) -> ExchangesBase:
|
||||
def update_exchanges_service(
|
||||
db_session: Session,
|
||||
user_id: int,
|
||||
exchange_id: int,
|
||||
name: str | None,
|
||||
notes: str | None,
|
||||
) -> ExchangesBase:
|
||||
existing_exchange = crud.get_exchange_by_id(db_session, exchange_id)
|
||||
if not existing_exchange:
|
||||
raise ExchangeNotFoundError("Exchange not found")
|
||||
@@ -211,16 +234,22 @@ def update_exchanges_service(db_session: Session, user_id: int, exchange_id: int
|
||||
raise ExchangeNotFoundError("Exchange not found")
|
||||
|
||||
if name:
|
||||
other_exchange = crud.get_exchange_by_name_and_user_id(db_session, name, user_id)
|
||||
other_exchange = crud.get_exchange_by_name_and_user_id(
|
||||
db_session, name, user_id
|
||||
)
|
||||
if other_exchange and other_exchange.id != existing_exchange.id:
|
||||
raise ExchangeAlreadyExistsError("Another exchange with the same name already exists for this user")
|
||||
raise ExchangeAlreadyExistsError(
|
||||
"Another exchange with the same name already exists for this user"
|
||||
)
|
||||
|
||||
exchange_data = ExchangesBase(
|
||||
name=name or existing_exchange.name,
|
||||
notes=notes or existing_exchange.notes,
|
||||
)
|
||||
try:
|
||||
exchange = crud.update_exchange(db_session, cast("int", existing_exchange.id), update_data=exchange_data)
|
||||
exchange = crud.update_exchange(
|
||||
db_session, cast("int", existing_exchange.id), update_data=exchange_data
|
||||
)
|
||||
except Exception as e:
|
||||
logger.exception("Failed to update exchange: \n")
|
||||
raise ServiceError("Failed to update exchange") from e
|
||||
@@ -228,7 +257,9 @@ def update_exchanges_service(db_session: Session, user_id: int, exchange_id: int
|
||||
|
||||
|
||||
# Cycle Service
|
||||
def create_cycle_service(db_session: Session, user_id: int, cycle_data: CycleBase) -> CycleRead:
|
||||
def create_cycle_service(
|
||||
db_session: Session, user_id: int, cycle_data: CycleBase
|
||||
) -> CycleRead:
|
||||
raise NotImplementedError("Cycle creation not implemented")
|
||||
cycle_data_dict = cycle_data.model_dump()
|
||||
cycle_data_dict["user_id"] = user_id
|
||||
@@ -237,7 +268,9 @@ def create_cycle_service(db_session: Session, user_id: int, cycle_data: CycleBas
|
||||
return CycleRead.model_validate(created_cycle)
|
||||
|
||||
|
||||
def get_cycle_by_id_service(db_session: Session, user_id: int, cycle_id: int) -> CycleRead:
|
||||
def get_cycle_by_id_service(
|
||||
db_session: Session, user_id: int, cycle_id: int
|
||||
) -> CycleRead:
|
||||
cycle = crud.get_cycle_by_id(db_session, cycle_id)
|
||||
if not cycle:
|
||||
raise CycleNotFoundError("Cycle not found")
|
||||
@@ -256,22 +289,65 @@ def _validate_cycle_update_data(cycle_data: CycleUpdate) -> tuple[bool, str]: #
|
||||
return False, "end_date is required when status is CLOSED"
|
||||
if cycle_data.status == "OPEN" and cycle_data.end_date is not None:
|
||||
return False, "end_date must be empty when status is OPEN"
|
||||
if cycle_data.capital_exposure_cents is not None and cycle_data.capital_exposure_cents < 0:
|
||||
if (
|
||||
cycle_data.capital_exposure_cents is not None
|
||||
and cycle_data.capital_exposure_cents < 0
|
||||
):
|
||||
return False, "capital_exposure_cents must be non-negative"
|
||||
if (
|
||||
cycle_data.funding_source is not None
|
||||
and cycle_data.funding_source != "CASH"
|
||||
and (cycle_data.loan_amount_cents is None or cycle_data.loan_interest_rate_tenth_bps is None)
|
||||
and (
|
||||
cycle_data.loan_amount_cents is None
|
||||
or cycle_data.loan_interest_rate_tenth_bps is None
|
||||
)
|
||||
):
|
||||
return False, "loan_amount_cents and loan_interest_rate_tenth_bps are required when funding_source is not CASH"
|
||||
return (
|
||||
False,
|
||||
"loan_amount_cents and loan_interest_rate_tenth_bps are required when funding_source is not CASH",
|
||||
)
|
||||
if cycle_data.loan_amount_cents is not None and cycle_data.loan_amount_cents < 0:
|
||||
return False, "loan_amount_cents must be non-negative"
|
||||
if cycle_data.loan_interest_rate_tenth_bps is not None and cycle_data.loan_interest_rate_tenth_bps < 0:
|
||||
if (
|
||||
cycle_data.loan_interest_rate_tenth_bps is not None
|
||||
and cycle_data.loan_interest_rate_tenth_bps < 0
|
||||
):
|
||||
return False, "loan_interest_rate_tenth_bps must be non-negative"
|
||||
return True, ""
|
||||
|
||||
|
||||
def update_cycle_service(db_session: Session, user_id: int, cycle_data: CycleUpdate) -> CycleRead:
|
||||
def _create_cycle_loan_event(
|
||||
db_session: Session,
|
||||
cycle_id: int,
|
||||
loan_amount_cents: int | None,
|
||||
loan_interest_rate_tenth_bps: int | None,
|
||||
) -> None:
|
||||
now = datetime.now(timezone.utc)
|
||||
today = now.date()
|
||||
existing_loan_event = crud.get_loan_event_by_cycle_id_and_effective_date(
|
||||
db_session, cycle_id, today
|
||||
)
|
||||
if existing_loan_event:
|
||||
raise CycleLoanEventExistsError(
|
||||
"A loan event with the same effective_date already exists for this cycle."
|
||||
)
|
||||
loan_event_data = CycleLoanChangeEventBase(
|
||||
cycle_id=cycle_id,
|
||||
effective_date=today,
|
||||
loan_amount_cents=loan_amount_cents,
|
||||
loan_interest_rate_tenth_bps=loan_interest_rate_tenth_bps,
|
||||
created_at=now,
|
||||
)
|
||||
try:
|
||||
crud.create_cycle_loan_event(db_session, loan_event_data)
|
||||
except Exception as e:
|
||||
logger.exception("Failed to create cycle loan event: \n")
|
||||
raise ServiceError("Failed to create cycle loan event") from e
|
||||
|
||||
|
||||
def update_cycle_service(
|
||||
db_session: Session, user_id: int, cycle_data: CycleUpdate
|
||||
) -> CycleRead:
|
||||
is_valid, err_msg = _validate_cycle_update_data(cycle_data)
|
||||
if not is_valid:
|
||||
raise InvalidCycleDataError(err_msg)
|
||||
@@ -281,22 +357,68 @@ def update_cycle_service(db_session: Session, user_id: int, cycle_data: CycleUpd
|
||||
raise CycleNotFoundError("Cycle not found")
|
||||
if existing_cycle.user_id != user_id:
|
||||
raise CycleNotFoundError("Cycle not found")
|
||||
if (
|
||||
cycle_data.loan_amount_cents is not None
|
||||
or cycle_data.loan_interest_rate_tenth_bps is not None
|
||||
):
|
||||
_create_cycle_loan_event(
|
||||
db_session,
|
||||
cycle_id,
|
||||
cycle_data.loan_amount_cents,
|
||||
cycle_data.loan_interest_rate_tenth_bps,
|
||||
)
|
||||
|
||||
provided_data_dict = cycle_data.model_dump(exclude_unset=True)
|
||||
cycle_data_with_user_id: CycleBase = CycleBase.model_validate(provided_data_dict)
|
||||
|
||||
try:
|
||||
updated_cycle = crud.update_cycle(db_session, cycle_id, update_data=cycle_data_with_user_id)
|
||||
updated_cycle = crud.update_cycle(
|
||||
db_session, cycle_id, update_data=cycle_data_with_user_id
|
||||
)
|
||||
except Exception as e:
|
||||
logger.exception("Failed to update cycle: \n")
|
||||
raise ServiceError("Failed to update cycle") from e
|
||||
return CycleRead.model_validate(updated_cycle)
|
||||
|
||||
|
||||
def accural_interest_service(db_session: Session, cycle_id: int) -> None:
|
||||
cycle = crud.get_cycle_by_id(db_session, cycle_id)
|
||||
if not cycle:
|
||||
logger.exception("Cycle not found for interest accrual")
|
||||
raise CycleNotFoundError("Cycle not found")
|
||||
if cycle.loan_amount_cents is None or cycle.loan_interest_rate_tenth_bps is None:
|
||||
logger.info("Cycle has no loan, skipping interest accrual")
|
||||
return
|
||||
today = datetime.now(timezone.utc).date()
|
||||
amount_cents = round(
|
||||
cycle.loan_amount_cents * cycle.loan_interest_rate_tenth_bps / 100000 / 365
|
||||
)
|
||||
try:
|
||||
crud.create_cycle_daily_accrual(
|
||||
db_session,
|
||||
cycle_id=cycle_id,
|
||||
accrual_date=today,
|
||||
accrual_amount_cents=amount_cents,
|
||||
)
|
||||
except Exception as e:
|
||||
logger.exception("Failed to create cycle interest accrual: \n")
|
||||
raise ServiceError("Failed to create cycle interest accrual") from e
|
||||
|
||||
|
||||
def flush_interest_accruals_service(db_session: Session) -> None:
|
||||
pass
|
||||
|
||||
|
||||
# Trades service
|
||||
def _append_cashflows(trade_data: TradeCreate) -> TradeCreate:
|
||||
sign_multipler: int
|
||||
if trade_data.trade_type in ("SELL_PUT", "SELL_CALL", "EXERCISE_CALL", "CLOSE_LONG_SPOT", "SHORT_SPOT"):
|
||||
if trade_data.trade_type in (
|
||||
"SELL_PUT",
|
||||
"SELL_CALL",
|
||||
"EXERCISE_CALL",
|
||||
"CLOSE_LONG_SPOT",
|
||||
"SHORT_SPOT",
|
||||
):
|
||||
sign_multipler = 1
|
||||
else:
|
||||
sign_multipler = -1
|
||||
@@ -310,13 +432,18 @@ def _append_cashflows(trade_data: TradeCreate) -> TradeCreate:
|
||||
|
||||
def _validate_trade_data(trade_data: TradeCreate) -> bool:
|
||||
return not (
|
||||
trade_data.trade_type in ("SELL_PUT", "SELL_CALL") and (trade_data.expiry_date is None or trade_data.strike_price_cents is None)
|
||||
trade_data.trade_type in ("SELL_PUT", "SELL_CALL")
|
||||
and (trade_data.expiry_date is None or trade_data.strike_price_cents is None)
|
||||
)
|
||||
|
||||
|
||||
def create_trade_service(db_session: Session, user_id: int, trade_data: TradeCreate) -> TradeRead:
|
||||
def create_trade_service(
|
||||
db_session: Session, user_id: int, trade_data: TradeCreate
|
||||
) -> TradeRead:
|
||||
if not _validate_trade_data(trade_data):
|
||||
raise InvalidTradeDataError("Invalid trade data: expiry_date and strike_price_cents are required for SELL_PUT and SELL_CALL trades")
|
||||
raise InvalidTradeDataError(
|
||||
"Invalid trade data: expiry_date and strike_price_cents are required for SELL_PUT and SELL_CALL trades"
|
||||
)
|
||||
trade_data_dict = trade_data.model_dump()
|
||||
trade_data_dict["user_id"] = user_id
|
||||
trade_data_with_user_id: TradeCreate = TradeCreate.model_validate(trade_data_dict)
|
||||
@@ -325,7 +452,9 @@ def create_trade_service(db_session: Session, user_id: int, trade_data: TradeCre
|
||||
return TradeRead.model_validate(created_trade)
|
||||
|
||||
|
||||
def get_trade_by_id_service(db_session: Session, user_id: int, trade_id: int) -> TradeRead:
|
||||
def get_trade_by_id_service(
|
||||
db_session: Session, user_id: int, trade_id: int
|
||||
) -> TradeRead:
|
||||
trade = crud.get_trade_by_id(db_session, trade_id)
|
||||
if not trade:
|
||||
raise TradeNotFoundError("Trade not found")
|
||||
@@ -334,21 +463,27 @@ def get_trade_by_id_service(db_session: Session, user_id: int, trade_id: int) ->
|
||||
return TradeRead.model_validate(trade)
|
||||
|
||||
|
||||
def update_trade_friendly_name_service(db_session: Session, user_id: int, trade_id: int, friendly_name: str) -> TradeRead:
|
||||
def update_trade_friendly_name_service(
|
||||
db_session: Session, user_id: int, trade_id: int, friendly_name: str
|
||||
) -> TradeRead:
|
||||
existing_trade = crud.get_trade_by_id(db_session, trade_id)
|
||||
if not existing_trade:
|
||||
raise TradeNotFoundError("Trade not found")
|
||||
if existing_trade.user_id != user_id:
|
||||
raise TradeNotFoundError("Trade not found")
|
||||
try:
|
||||
updated_trade = crud.update_trade_friendly_name(db_session, trade_id, friendly_name)
|
||||
updated_trade = crud.update_trade_friendly_name(
|
||||
db_session, trade_id, friendly_name
|
||||
)
|
||||
except Exception as e:
|
||||
logger.exception("Failed to update trade friendly name: \n")
|
||||
raise ServiceError("Failed to update trade friendly name") from e
|
||||
return TradeRead.model_validate(updated_trade)
|
||||
|
||||
|
||||
def update_trade_note_service(db_session: Session, user_id: int, trade_id: int, note: str | None) -> TradeRead:
|
||||
def update_trade_note_service(
|
||||
db_session: Session, user_id: int, trade_id: int, note: str | None
|
||||
) -> TradeRead:
|
||||
existing_trade = crud.get_trade_by_id(db_session, trade_id)
|
||||
if not existing_trade:
|
||||
raise TradeNotFoundError("Trade not found")
|
||||
|
||||
34
backend/trading_journal/service_error.py
Normal file
34
backend/trading_journal/service_error.py
Normal file
@@ -0,0 +1,34 @@
|
||||
class ServiceError(Exception):
|
||||
pass
|
||||
|
||||
|
||||
class UserAlreadyExistsError(ServiceError):
|
||||
pass
|
||||
|
||||
|
||||
class ExchangeAlreadyExistsError(ServiceError):
|
||||
pass
|
||||
|
||||
|
||||
class ExchangeNotFoundError(ServiceError):
|
||||
pass
|
||||
|
||||
|
||||
class CycleNotFoundError(ServiceError):
|
||||
pass
|
||||
|
||||
|
||||
class TradeNotFoundError(ServiceError):
|
||||
pass
|
||||
|
||||
|
||||
class InvalidTradeDataError(ServiceError):
|
||||
pass
|
||||
|
||||
|
||||
class InvalidCycleDataError(ServiceError):
|
||||
pass
|
||||
|
||||
|
||||
class CycleLoanEventExistsError(ServiceError):
|
||||
pass
|
||||
Reference in New Issue
Block a user