feat: backend core — models, auth, CRUD, tests
This commit is contained in:
@@ -0,0 +1,16 @@
|
||||
# Import all models so Alembic can discover them for autogenerate
|
||||
from app.models.budget import Budget
|
||||
from app.models.category import Category, CategoryType
|
||||
from app.models.refresh_token import RefreshToken
|
||||
from app.models.transaction import Transaction, TransactionType
|
||||
from app.models.user import User
|
||||
|
||||
__all__ = [
|
||||
"User",
|
||||
"Category",
|
||||
"CategoryType",
|
||||
"Transaction",
|
||||
"TransactionType",
|
||||
"Budget",
|
||||
"RefreshToken",
|
||||
]
|
||||
|
||||
@@ -0,0 +1,47 @@
|
||||
import uuid
|
||||
from datetime import datetime
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
from sqlalchemy import (
|
||||
CheckConstraint,
|
||||
DateTime,
|
||||
ForeignKey,
|
||||
Integer,
|
||||
String,
|
||||
UniqueConstraint,
|
||||
Uuid,
|
||||
func,
|
||||
)
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.database import Base
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.models.category import Category
|
||||
|
||||
|
||||
class Budget(Base):
|
||||
__tablename__ = "budgets"
|
||||
__table_args__ = (
|
||||
UniqueConstraint("user_id", "category_id", "month", name="uq_budgets_user_category_month"),
|
||||
CheckConstraint("limit_cents > 0", name="ck_budgets_limit_positive"),
|
||||
)
|
||||
|
||||
id: Mapped[uuid.UUID] = mapped_column(Uuid, primary_key=True, default=uuid.uuid4)
|
||||
user_id: Mapped[uuid.UUID] = mapped_column(
|
||||
Uuid, ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True
|
||||
)
|
||||
category_id: Mapped[uuid.UUID] = mapped_column(
|
||||
Uuid, ForeignKey("categories.id"), nullable=False
|
||||
)
|
||||
# Format YYYY-MM
|
||||
month: Mapped[str] = mapped_column(String(7), nullable=False)
|
||||
limit_cents: Mapped[int] = mapped_column(Integer, nullable=False)
|
||||
created_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=False), nullable=False, server_default=func.now()
|
||||
)
|
||||
updated_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=False), nullable=False, server_default=func.now()
|
||||
)
|
||||
|
||||
category: Mapped["Category"] = relationship("Category", lazy="raise")
|
||||
@@ -0,0 +1,36 @@
|
||||
import enum
|
||||
import uuid
|
||||
from datetime import datetime
|
||||
|
||||
from sqlalchemy import Boolean, DateTime, Enum, ForeignKey, String, Uuid, func
|
||||
from sqlalchemy.orm import Mapped, mapped_column
|
||||
|
||||
from app.database import Base
|
||||
|
||||
|
||||
class CategoryType(str, enum.Enum):
|
||||
income = "income"
|
||||
expense = "expense"
|
||||
|
||||
|
||||
class Category(Base):
|
||||
__tablename__ = "categories"
|
||||
|
||||
id: Mapped[uuid.UUID] = mapped_column(Uuid, primary_key=True, default=uuid.uuid4)
|
||||
user_id: Mapped[uuid.UUID] = mapped_column(
|
||||
Uuid, ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True
|
||||
)
|
||||
name: Mapped[str] = mapped_column(String(50), nullable=False)
|
||||
type: Mapped[CategoryType] = mapped_column(
|
||||
Enum(CategoryType, name="categorytype", native_enum=False, length=10),
|
||||
nullable=False,
|
||||
)
|
||||
color: Mapped[str | None] = mapped_column(String(7), nullable=True)
|
||||
icon: Mapped[str | None] = mapped_column(String(50), nullable=True)
|
||||
is_default: Mapped[bool] = mapped_column(
|
||||
Boolean, nullable=False, default=False, server_default="false"
|
||||
)
|
||||
deleted_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=False), nullable=True)
|
||||
created_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=False), nullable=False, server_default=func.now()
|
||||
)
|
||||
@@ -0,0 +1,23 @@
|
||||
import uuid
|
||||
from datetime import datetime
|
||||
|
||||
from sqlalchemy import DateTime, ForeignKey, String, Uuid, func
|
||||
from sqlalchemy.orm import Mapped, mapped_column
|
||||
|
||||
from app.database import Base
|
||||
|
||||
|
||||
class RefreshToken(Base):
|
||||
__tablename__ = "refresh_tokens"
|
||||
|
||||
id: Mapped[uuid.UUID] = mapped_column(Uuid, primary_key=True, default=uuid.uuid4)
|
||||
user_id: Mapped[uuid.UUID] = mapped_column(
|
||||
Uuid, ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True
|
||||
)
|
||||
# SHA-256 hash of the raw token (never store raw tokens)
|
||||
token_hash: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
||||
expires_at: Mapped[datetime] = mapped_column(DateTime(timezone=False), nullable=False)
|
||||
revoked_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=False), nullable=True)
|
||||
created_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=False), nullable=False, server_default=func.now()
|
||||
)
|
||||
@@ -0,0 +1,62 @@
|
||||
import enum
|
||||
import uuid
|
||||
from datetime import date, datetime
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
from sqlalchemy import (
|
||||
CheckConstraint,
|
||||
Date,
|
||||
DateTime,
|
||||
Enum,
|
||||
ForeignKey,
|
||||
Index,
|
||||
Integer,
|
||||
String,
|
||||
Uuid,
|
||||
func,
|
||||
)
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.database import Base
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.models.category import Category
|
||||
|
||||
|
||||
class TransactionType(str, enum.Enum):
|
||||
income = "income"
|
||||
expense = "expense"
|
||||
|
||||
|
||||
class Transaction(Base):
|
||||
__tablename__ = "transactions"
|
||||
__table_args__ = (
|
||||
CheckConstraint("amount_cents > 0", name="ck_transactions_amount_positive"),
|
||||
Index("ix_transactions_user_date", "user_id", "transaction_date", "deleted_at"),
|
||||
Index("ix_transactions_user_category", "user_id", "category_id", "deleted_at"),
|
||||
)
|
||||
|
||||
id: Mapped[uuid.UUID] = mapped_column(Uuid, primary_key=True, default=uuid.uuid4)
|
||||
user_id: Mapped[uuid.UUID] = mapped_column(
|
||||
Uuid, ForeignKey("users.id", ondelete="CASCADE"), nullable=False
|
||||
)
|
||||
category_id: Mapped[uuid.UUID] = mapped_column(
|
||||
Uuid, ForeignKey("categories.id"), nullable=False
|
||||
)
|
||||
amount_cents: Mapped[int] = mapped_column(Integer, nullable=False)
|
||||
type: Mapped[TransactionType] = mapped_column(
|
||||
Enum(TransactionType, name="transactiontype", native_enum=False, length=10),
|
||||
nullable=False,
|
||||
)
|
||||
description: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
||||
transaction_date: Mapped[date] = mapped_column(Date, nullable=False)
|
||||
deleted_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=False), nullable=True)
|
||||
created_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=False), nullable=False, server_default=func.now()
|
||||
)
|
||||
updated_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=False), nullable=False, server_default=func.now()
|
||||
)
|
||||
|
||||
# Relationships — use selectinload() explicitly; lazy="raise" prevents accidental N+1
|
||||
category: Mapped["Category"] = relationship("Category", lazy="raise")
|
||||
@@ -0,0 +1,23 @@
|
||||
import uuid
|
||||
from datetime import datetime
|
||||
|
||||
from sqlalchemy import Boolean, DateTime, String, Uuid, func
|
||||
from sqlalchemy.orm import Mapped, mapped_column
|
||||
|
||||
from app.database import Base
|
||||
|
||||
|
||||
class User(Base):
|
||||
__tablename__ = "users"
|
||||
|
||||
id: Mapped[uuid.UUID] = mapped_column(Uuid, primary_key=True, default=uuid.uuid4)
|
||||
email: Mapped[str] = mapped_column(String(255), unique=True, nullable=False, index=True)
|
||||
hashed_password: Mapped[str] = mapped_column(String(255), nullable=False)
|
||||
full_name: Mapped[str] = mapped_column(String(100), nullable=False)
|
||||
is_active: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True, server_default="true")
|
||||
created_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=False), nullable=False, server_default=func.now()
|
||||
)
|
||||
updated_at: Mapped[datetime] = mapped_column(
|
||||
DateTime(timezone=False), nullable=False, server_default=func.now()
|
||||
)
|
||||
Reference in New Issue
Block a user