2c6298ca4b
- Set up project structure with FastAPI - Implement user and account management - Add send and receive money functionality - Set up transaction processing system - Add JWT authentication - Configure SQLAlchemy with SQLite - Set up Alembic for database migrations - Create comprehensive API documentation
50 lines
1.9 KiB
Python
50 lines
1.9 KiB
Python
from sqlalchemy import Column, Integer, String, Float, ForeignKey, DateTime, Enum
|
|
from sqlalchemy.orm import relationship
|
|
from sqlalchemy.sql import func
|
|
import enum
|
|
|
|
from app.db.base import Base
|
|
|
|
|
|
class TransactionType(str, enum.Enum):
|
|
DEPOSIT = "deposit"
|
|
WITHDRAWAL = "withdrawal"
|
|
TRANSFER = "transfer"
|
|
|
|
|
|
class TransactionStatus(str, enum.Enum):
|
|
PENDING = "pending"
|
|
COMPLETED = "completed"
|
|
FAILED = "failed"
|
|
CANCELLED = "cancelled"
|
|
|
|
|
|
class Transaction(Base):
|
|
__tablename__ = "transactions"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
transaction_reference = Column(String, unique=True, index=True, nullable=False)
|
|
|
|
sender_id = Column(Integer, ForeignKey("users.id"), nullable=True)
|
|
sender_account_id = Column(Integer, ForeignKey("accounts.id"), nullable=True)
|
|
|
|
receiver_id = Column(Integer, ForeignKey("users.id"), nullable=True)
|
|
receiver_account_id = Column(Integer, ForeignKey("accounts.id"), nullable=True)
|
|
|
|
amount = Column(Float, nullable=False)
|
|
currency = Column(String, default="USD", nullable=False)
|
|
|
|
transaction_type = Column(Enum(TransactionType), nullable=False)
|
|
status = Column(Enum(TransactionStatus), default=TransactionStatus.PENDING, nullable=False)
|
|
|
|
description = Column(String, nullable=True)
|
|
created_at = Column(DateTime(timezone=True), server_default=func.now())
|
|
updated_at = Column(DateTime(timezone=True), onupdate=func.now())
|
|
completed_at = Column(DateTime(timezone=True), nullable=True)
|
|
|
|
# Relationships
|
|
sender = relationship("User", foreign_keys=[sender_id], back_populates="sent_transactions")
|
|
receiver = relationship("User", foreign_keys=[receiver_id], back_populates="received_transactions")
|
|
|
|
sender_account = relationship("Account", foreign_keys=[sender_account_id], back_populates="sent_transactions")
|
|
receiver_account = relationship("Account", foreign_keys=[receiver_account_id], back_populates="received_transactions") |