schemas.py

from datetime import datetime
from itertools import count
from typing import Literal, Optional

from pydantic import BaseModel, Field


class DepositRequest(BaseModel):
    account: str = Field(..., json_schema_extra={"example": "DE000000000000000000000"})
    amount: float = Field(..., json_schema_extra={"example": 50.0})


class DepositResponse(BaseModel):
    account: str
    new_balance: float


class WithdrawRequest(BaseModel):
    account: str = Field(..., json_schema_extra={"example": "DE000000000000000000000"})
    amount: float = Field(..., json_schema_extra={"example": 50.0})


class WithdrawResponse(BaseModel):
    account: str
    new_balance: float


class TransferRequest(BaseModel):
    src_account: str = Field(..., json_schema_extra={"example": "DE000000000000000000000"})
    dest_account: str = Field(..., json_schema_extra={"example": "DE000000000000000000001"})
    amount: float = Field(..., json_schema_extra={"example": 50.0})


class TransferResponse(BaseModel):
    src_account: str
    dest_account: str
    src_new_balance: float
    dest_new_balance: float


class Transaction(BaseModel):
    id: int = Field(default_factory=lambda: next(transaction_id_counter), example=1)
    src_account: str = Field(..., example="A1")
    type: Literal["deposit", "withdraw", "transfer"] = Field(..., example="deposit")
    amount: float = Field(..., example=100.0)
    dest_account: Optional[str] = Field(None, example="B1")  # Nullable, relevant only for transfers
    timestamp: datetime = Field(default_factory=datetime.utcnow, example="2024-01-01T12:00:00Z")
    balance: float = Field(..., example=500.0)  # Add a balance field


class AccountCreateRequest(BaseModel):
    account_number: str


# Initialize a global counter for auto-incrementing the ID
transaction_id_counter = count(1)