mirror of
https://github.com/Zproger/Aurum.git
synced 2026-10-12 18:46:40 +07:00
aefd6cc035
Криптовалюту теперь можно разделять на отдельные портфели (например, долгосрочные инвестиции, накопления на недвижимость, спекулятивные монеты) вместо одного общего списка — по тому же принципу, что уже есть у счетов (Account): у портфеля есть имя, цветная метка и архивация, каждая монета обязательно принадлежит одному портфелю, один и тот же токен можно держать в нескольких портфелях сразу.
56 lines
2.1 KiB
Python
56 lines
2.1 KiB
Python
"""add crypto portfolios
|
|
|
|
Revision ID: b7d4e9f2a610
|
|
Revises: 7b535ef9e189
|
|
Create Date: 2026-09-04 12:00:00.000000
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = 'b7d4e9f2a610'
|
|
down_revision: Union[str, None] = '7b535ef9e189'
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.create_table(
|
|
'crypto_portfolios',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('name', sa.String(length=100), nullable=False),
|
|
sa.Column('color', sa.String(length=7), nullable=True),
|
|
sa.Column('is_archived', sa.Boolean(), nullable=False, server_default=sa.text('false')),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
)
|
|
# Every crypto holding needs a portfolio_id (see below) — seed one
|
|
# default portfolio up front so existing holdings have somewhere to
|
|
# land, same "row exists immediately after migrating" reasoning as
|
|
# 9f3a2d7c5e11's app_settings seed.
|
|
op.execute("INSERT INTO crypto_portfolios (id, name, color, is_archived) VALUES (1, 'Main Portfolio', '#2a78d6', false)")
|
|
op.execute("SELECT setval(pg_get_serial_sequence('crypto_portfolios', 'id'), 1)")
|
|
|
|
op.add_column('crypto_holdings', sa.Column('portfolio_id', sa.Integer(), nullable=True))
|
|
op.execute("UPDATE crypto_holdings SET portfolio_id = 1")
|
|
op.alter_column('crypto_holdings', 'portfolio_id', nullable=False)
|
|
op.create_foreign_key(
|
|
'fk_crypto_holdings_portfolio_id',
|
|
'crypto_holdings',
|
|
'crypto_portfolios',
|
|
['portfolio_id'],
|
|
['id'],
|
|
ondelete='RESTRICT',
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_constraint('fk_crypto_holdings_portfolio_id', 'crypto_holdings', type_='foreignkey')
|
|
op.drop_column('crypto_holdings', 'portfolio_id')
|
|
op.drop_table('crypto_portfolios')
|