212 lines
10 KiB
Python
212 lines
10 KiB
Python
"""Initial schema
|
|
|
|
Revision ID: 001
|
|
Revises:
|
|
Create Date: 2026-04-28
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
revision = "001"
|
|
down_revision = None
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.create_table(
|
|
"roles",
|
|
sa.Column("id", sa.Integer, primary_key=True),
|
|
sa.Column("name", sa.String(50), unique=True, nullable=False),
|
|
sa.Column("description", sa.Text),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
)
|
|
|
|
op.create_table(
|
|
"users",
|
|
sa.Column("id", sa.Integer, primary_key=True),
|
|
sa.Column("username", sa.String(100), unique=True, nullable=False),
|
|
sa.Column("email", sa.String(255), unique=True, nullable=False),
|
|
sa.Column("password_hash", sa.String(255), nullable=False),
|
|
sa.Column("role_id", sa.Integer, sa.ForeignKey("roles.id"), nullable=False),
|
|
sa.Column("is_active", sa.Boolean, default=True, nullable=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
)
|
|
op.create_index("ix_users_username", "users", ["username"])
|
|
op.create_index("ix_users_email", "users", ["email"])
|
|
|
|
op.create_table(
|
|
"currencies",
|
|
sa.Column("id", sa.Integer, primary_key=True),
|
|
sa.Column("code", sa.String(3), unique=True, nullable=False),
|
|
sa.Column("name", sa.String(100), nullable=False),
|
|
sa.Column("symbol", sa.String(10)),
|
|
sa.Column("is_active", sa.Boolean, default=True, nullable=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
)
|
|
|
|
op.create_table(
|
|
"exchange_rates",
|
|
sa.Column("id", sa.Integer, primary_key=True),
|
|
sa.Column("currency_from_id", sa.Integer, sa.ForeignKey("currencies.id"), nullable=False),
|
|
sa.Column("currency_to_id", sa.Integer, sa.ForeignKey("currencies.id"), nullable=False),
|
|
sa.Column("rate", sa.Numeric(20, 6), nullable=False),
|
|
sa.Column("rate_date", sa.Date, nullable=False),
|
|
sa.Column("source", sa.String(100)),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
sa.UniqueConstraint("currency_from_id", "currency_to_id", "rate_date", name="uq_exchange_rate_day"),
|
|
)
|
|
|
|
op.create_table(
|
|
"locations",
|
|
sa.Column("id", sa.Integer, primary_key=True),
|
|
sa.Column("user_id", sa.Integer, sa.ForeignKey("users.id"), nullable=True),
|
|
sa.Column("name", sa.String(200), nullable=False),
|
|
sa.Column("type", sa.String(50)),
|
|
sa.Column("address", sa.Text),
|
|
sa.Column("city", sa.String(100)),
|
|
sa.Column("country", sa.String(100)),
|
|
sa.Column("latitude", sa.Numeric(9, 6)),
|
|
sa.Column("longitude", sa.Numeric(9, 6)),
|
|
sa.Column("parent_id", sa.Integer, sa.ForeignKey("locations.id"), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
)
|
|
|
|
op.create_table(
|
|
"account_types",
|
|
sa.Column("id", sa.Integer, primary_key=True),
|
|
sa.Column("name", sa.String(100), unique=True, nullable=False),
|
|
sa.Column("icon", sa.String(50)),
|
|
sa.Column("description", sa.Text),
|
|
)
|
|
|
|
op.create_table(
|
|
"accounts",
|
|
sa.Column("id", sa.Integer, primary_key=True),
|
|
sa.Column("user_id", sa.Integer, sa.ForeignKey("users.id"), nullable=False),
|
|
sa.Column("name", sa.String(200), nullable=False),
|
|
sa.Column("account_type_id", sa.Integer, sa.ForeignKey("account_types.id"), nullable=False),
|
|
sa.Column("currency_id", sa.Integer, sa.ForeignKey("currencies.id"), nullable=False),
|
|
sa.Column("balance", sa.Numeric(20, 2), default=0, nullable=False),
|
|
sa.Column("initial_balance", sa.Numeric(20, 2), default=0, nullable=False),
|
|
sa.Column("is_active", sa.Boolean, default=True, nullable=False),
|
|
sa.Column("credit_limit", sa.Numeric(20, 2)),
|
|
sa.Column("description", sa.Text),
|
|
sa.Column("icon", sa.String(50)),
|
|
sa.Column("color", sa.String(7)),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
)
|
|
op.create_index("ix_accounts_user_id", "accounts", ["user_id"])
|
|
|
|
op.create_table(
|
|
"categories",
|
|
sa.Column("id", sa.Integer, primary_key=True),
|
|
sa.Column("user_id", sa.Integer, sa.ForeignKey("users.id"), nullable=True),
|
|
sa.Column("name", sa.String(200), nullable=False),
|
|
sa.Column("type", sa.String(20), nullable=False),
|
|
sa.Column("parent_id", sa.Integer, sa.ForeignKey("categories.id"), nullable=True),
|
|
sa.Column("icon", sa.String(50)),
|
|
sa.Column("color", sa.String(7)),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
)
|
|
op.create_index("ix_categories_user_id", "categories", ["user_id"])
|
|
|
|
op.create_table(
|
|
"budgets",
|
|
sa.Column("id", sa.Integer, primary_key=True),
|
|
sa.Column("user_id", sa.Integer, sa.ForeignKey("users.id"), nullable=False),
|
|
sa.Column("category_id", sa.Integer, sa.ForeignKey("categories.id"), nullable=False),
|
|
sa.Column("name", sa.String(200), nullable=False),
|
|
sa.Column("amount", sa.Numeric(20, 2), nullable=False),
|
|
sa.Column("currency_id", sa.Integer, sa.ForeignKey("currencies.id"), nullable=False),
|
|
sa.Column("period_type", sa.String(20), nullable=False),
|
|
sa.Column("start_date", sa.Date, nullable=False),
|
|
sa.Column("end_date", sa.Date),
|
|
sa.Column("is_active", sa.Boolean, default=True, nullable=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
)
|
|
op.create_index("ix_budgets_user_id", "budgets", ["user_id"])
|
|
|
|
op.create_table(
|
|
"operation_types",
|
|
sa.Column("id", sa.Integer, primary_key=True),
|
|
sa.Column("name", sa.String(50), unique=True, nullable=False),
|
|
sa.Column("description", sa.Text),
|
|
)
|
|
|
|
op.create_table(
|
|
"operations",
|
|
sa.Column("id", sa.Integer, primary_key=True),
|
|
sa.Column("user_id", sa.Integer, sa.ForeignKey("users.id"), nullable=False),
|
|
sa.Column("account_id", sa.Integer, sa.ForeignKey("accounts.id"), nullable=False),
|
|
sa.Column("category_id", sa.Integer, sa.ForeignKey("categories.id"), nullable=True),
|
|
sa.Column("operation_type_id", sa.Integer, sa.ForeignKey("operation_types.id"), nullable=False),
|
|
sa.Column("amount", sa.Numeric(20, 2), nullable=False),
|
|
sa.Column("currency_id", sa.Integer, sa.ForeignKey("currencies.id"), nullable=False),
|
|
sa.Column("amount_converted", sa.Numeric(20, 2)),
|
|
sa.Column("exchange_rate", sa.Numeric(20, 6)),
|
|
sa.Column("date", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column("description", sa.Text),
|
|
sa.Column("location_id", sa.Integer, sa.ForeignKey("locations.id"), nullable=True),
|
|
sa.Column("is_planned", sa.Boolean, default=False, nullable=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
)
|
|
op.create_index("ix_operations_user_id", "operations", ["user_id"])
|
|
op.create_index("ix_operations_account_id", "operations", ["account_id"])
|
|
op.create_index("ix_operations_date", "operations", ["date"])
|
|
|
|
op.create_table(
|
|
"operation_relations",
|
|
sa.Column("id", sa.Integer, primary_key=True),
|
|
sa.Column("operation_from_id", sa.Integer, sa.ForeignKey("operations.id"), nullable=False),
|
|
sa.Column("operation_to_id", sa.Integer, sa.ForeignKey("operations.id"), nullable=False),
|
|
sa.Column("relation_type", sa.String(50), nullable=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
)
|
|
|
|
op.create_table(
|
|
"calendar_events",
|
|
sa.Column("id", sa.Integer, primary_key=True),
|
|
sa.Column("user_id", sa.Integer, sa.ForeignKey("users.id"), nullable=False),
|
|
sa.Column("title", sa.String(200), nullable=False),
|
|
sa.Column("description", sa.Text),
|
|
sa.Column("event_date", sa.Date, nullable=False),
|
|
sa.Column("event_time", sa.Time),
|
|
sa.Column("is_recurring", sa.Boolean, default=False, nullable=False),
|
|
sa.Column("recurrence_pattern", sa.String(50)),
|
|
sa.Column("recurrence_end_date", sa.Date),
|
|
sa.Column("related_operation_id", sa.Integer, sa.ForeignKey("operations.id"), nullable=True),
|
|
sa.Column("category_id", sa.Integer, sa.ForeignKey("categories.id"), nullable=True),
|
|
sa.Column("amount", sa.Numeric(20, 2)),
|
|
sa.Column("currency_id", sa.Integer, sa.ForeignKey("currencies.id"), nullable=True),
|
|
sa.Column("reminder_days", sa.Integer),
|
|
sa.Column("is_completed", sa.Boolean, default=False, nullable=False),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
|
|
)
|
|
op.create_index("ix_calendar_events_user_id", "calendar_events", ["user_id"])
|
|
op.create_index("ix_calendar_events_event_date", "calendar_events", ["event_date"])
|
|
|
|
# Grant permissions to app user (runs only if the role exists)
|
|
op.execute("""
|
|
DO $$
|
|
BEGIN
|
|
IF EXISTS (SELECT FROM pg_roles WHERE rolname = 'finance_app_user') THEN
|
|
GRANT SELECT, INSERT, UPDATE, DELETE ON ALL TABLES IN SCHEMA public TO finance_app_user;
|
|
GRANT USAGE, SELECT ON ALL SEQUENCES IN SCHEMA public TO finance_app_user;
|
|
END IF;
|
|
END$$;
|
|
""")
|
|
|
|
|
|
def downgrade() -> None:
|
|
for table in [
|
|
"calendar_events", "operation_relations", "operations", "operation_types",
|
|
"budgets", "categories", "accounts", "account_types",
|
|
"locations", "exchange_rates", "currencies", "users", "roles",
|
|
]:
|
|
op.drop_table(table)
|