aboutsummaryrefslogtreecommitdiff
path: root/ihatemoney/migrations/versions/927ed575acbd_add_currencies.py
blob: 88b8a5b00c91884c4541853ef1141fab73470ae2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
"""Add currencies

Revision ID: 927ed575acbd
Revises: cb038f79982e
Create Date: 2020-04-25 14:49:41.136602

"""

# revision identifiers, used by Alembic.
revision = "927ed575acbd"
down_revision = "cb038f79982e"

from alembic import op
import sqlalchemy as sa
from ihatemoney.currency_convertor import CurrencyConverter


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column("bill", sa.Column("converted_amount", sa.Float(), nullable=True))
    op.add_column(
        "bill",
        sa.Column(
            "original_currency",
            sa.String(length=3),
            server_default=CurrencyConverter.no_currency,
            nullable=True,
        ),
    )
    op.add_column(
        "bill_version",
        sa.Column("converted_amount", sa.Float(), autoincrement=False, nullable=True),
    )
    op.add_column(
        "bill_version",
        sa.Column(
            "original_currency", sa.String(length=3), autoincrement=False, nullable=True
        ),
    )
    op.add_column(
        "project",
        sa.Column(
            "default_currency",
            sa.String(length=3),
            server_default=CurrencyConverter.no_currency,
            nullable=True,
        ),
    )
    op.add_column(
        "project_version",
        sa.Column(
            "default_currency", sa.String(length=3), autoincrement=False, nullable=True
        ),
    )
    # ### end Alembic commands ###
    op.execute(
        """
    UPDATE bill
    SET converted_amount = amount
    WHERE converted_amount IS NULL
    """
    )


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_column("project_version", "default_currency")
    op.drop_column("project", "default_currency")
    op.drop_column("bill_version", "original_currency")
    op.drop_column("bill_version", "converted_amount")
    op.drop_column("bill", "original_currency")
    op.drop_column("bill", "converted_amount")
    # ### end Alembic commands ###