mirror of
https://github.com/Zproger/Aurum.git
synced 2026-10-13 11:06:40 +07:00
feat: портфели в разделе «Криптовалюта»
Криптовалюту теперь можно разделять на отдельные портфели (например, долгосрочные инвестиции, накопления на недвижимость, спекулятивные монеты) вместо одного общего списка — по тому же принципу, что уже есть у счетов (Account): у портфеля есть имя, цветная метка и архивация, каждая монета обязательно принадлежит одному портфелю, один и тот же токен можно держать в нескольких портфелях сразу.
This commit is contained in:
@@ -0,0 +1,55 @@
|
||||
"""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')
|
||||
Reference in New Issue
Block a user