devicehub-teal/ereuse_devicehub/migrations/versions/51439cf24be8_change_trade_action.py

"""change trade action

Revision ID: 51439cf24be8
Revises: eca457d8b2a4
Create Date: 2021-03-15 17:40:34.410408

"""
import citext
import sqlalchemy as sa
from alembic import context, op
from sqlalchemy.dialects import postgresql

# revision identifiers, used by Alembic.
revision = '51439cf24be8'
down_revision = '21afd375a654'
branch_labels = None
depends_on = None


def get_inv():
    INV = context.get_x_argument(as_dictionary=True).get('inventory')
    if not INV:
        raise ValueError("Inventory value is not specified")
    return INV


def upgrade_data():
    con = op.get_bind()
    sql = "update common.user set active='t';"
    con.execute(sql)
    sql = "update common.user set phantom='f';"
    con.execute(sql)


def upgrade():
    # Trade
    currency = sa.Enum(
        'AFN',
        'ARS',
        'AWG',
        'AUD',
        'AZN',
        'BSD',
        'BBD',
        'BDT',
        'BYR',
        'BZD',
        'BMD',
        'BOB',
        'BAM',
        'BWP',
        'BGN',
        'BRL',
        'BND',
        'KHR',
        'CAD',
        'KYD',
        'CLP',
        'CNY',
        'COP',
        'CRC',
        'HRK',
        'CUP',
        'CZK',
        'DKK',
        'DOP',
        'XCD',
        'EGP',
        'SVC',
        'EEK',
        'EUR',
        'FKP',
        'FJD',
        'GHC',
        'GIP',
        'GTQ',
        'GGP',
        'GYD',
        'HNL',
        'HKD',
        'HUF',
        'ISK',
        'INR',
        'IDR',
        'IRR',
        'IMP',
        'ILS',
        'JMD',
        'JPY',
        'JEP',
        'KZT',
        'KPW',
        'KRW',
        'KGS',
        'LAK',
        'LVL',
        'LBP',
        'LRD',
        'LTL',
        'MKD',
        'MYR',
        'MUR',
        'MXN',
        'MNT',
        'MZN',
        'NAD',
        'NPR',
        'ANG',
        'NZD',
        'NIO',
        'NGN',
        'NOK',
        'OMR',
        'PKR',
        'PAB',
        'PYG',
        'PEN',
        'PHP',
        'PLN',
        'QAR',
        'RON',
        'RUB',
        'SHP',
        'SAR',
        'RSD',
        'SCR',
        'SGD',
        'SBD',
        'SOS',
        'ZAR',
        'LKR',
        'SEK',
        'CHF',
        'SRD',
        'SYP',
        'TWD',
        'THB',
        'TTD',
        'TRY',
        'TRL',
        'TVD',
        'UAH',
        'GBP',
        'USD',
        'UYU',
        'UZS',
        'VEF',
        name='currency',
        create_type=False,
        checkfirst=True,
    )

    op.drop_table('trade', schema=f'{get_inv()}')
    op.create_table(
        'trade',
        sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column('price', sa.Float(decimal_return_scale=4), nullable=True),
        sa.Column('lot_id', postgresql.UUID(as_uuid=True), nullable=True),
        sa.Column('date', sa.TIMESTAMP(timezone=True), nullable=True),
        sa.Column('user_from_id', postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column('user_to_id', postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column('document_id', citext.CIText(), nullable=True),
        sa.Column('confirm', sa.Boolean(), nullable=True),
        sa.Column(
            'code',
            citext.CIText(),
            default='',
            nullable=True,
            comment="This code is used for traceability",
        ),
        sa.ForeignKeyConstraint(
            ['id'],
            [f'{get_inv()}.action.id'],
        ),
        sa.ForeignKeyConstraint(
            ['user_from_id'],
            ['common.user.id'],
        ),
        sa.ForeignKeyConstraint(
            ['user_to_id'],
            ['common.user.id'],
        ),
        sa.ForeignKeyConstraint(
            ['lot_id'],
            [f'{get_inv()}.lot.id'],
        ),
        sa.PrimaryKeyConstraint('id'),
        schema=f'{get_inv()}',
    )

    op.add_column(
        "trade", sa.Column("currency", currency, nullable=False), schema=f'{get_inv()}'
    )

    op.create_table(
        'confirm',
        sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column('user_id', postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column('action_id', postgresql.UUID(as_uuid=True), nullable=False),
        sa.ForeignKeyConstraint(
            ['id'],
            [f'{get_inv()}.action.id'],
        ),
        sa.ForeignKeyConstraint(
            ['action_id'],
            [f'{get_inv()}.action.id'],
        ),
        sa.ForeignKeyConstraint(
            ['user_id'],
            ['common.user.id'],
        ),
        sa.PrimaryKeyConstraint('id'),
        schema=f'{get_inv()}',
    )

    # User
    op.add_column(
        'user',
        sa.Column('active', sa.Boolean(), default=True, nullable=True),
        schema='common',
    )
    op.add_column(
        'user',
        sa.Column('phantom', sa.Boolean(), default=False, nullable=True),
        schema='common',
    )

    upgrade_data()

    op.alter_column('user', 'active', nullable=False, schema='common')
    op.alter_column('user', 'phantom', nullable=False, schema='common')


def downgrade():
    op.drop_table('confirm', schema=f'{get_inv()}')
    op.drop_table('trade', schema=f'{get_inv()}')
    op.create_table(
        'trade',
        sa.Column(
            'shipping_date',
            sa.TIMESTAMP(timezone=True),
            nullable=True,
            comment='When are the devices going to be ready \n    \
                                      for shipping?\n    ',
        ),
        sa.Column(
            'invoice_number',
            citext.CIText(),
            nullable=True,
            comment='The id of the invoice so they can be linked.',
        ),
        sa.Column(
            'price_id',
            postgresql.UUID(as_uuid=True),
            nullable=True,
            comment='The price set for this trade.            \n    \
                                      If no price is set it is supposed that the trade was\n   \
                                      not payed, usual in donations.\n        ',
        ),
        sa.Column('to_id', postgresql.UUID(as_uuid=True), nullable=False),
        sa.Column(
            'confirms_id',
            postgresql.UUID(as_uuid=True),
            nullable=True,
            comment='An organize action that this association confirms. \
                                      \n    \n    For example, a ``Sell`` or ``Rent``\n   \
                                      can confirm a ``Reserve`` action.\n    ',
        ),
        sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
        sa.ForeignKeyConstraint(
            ['confirms_id'],
            [f'{get_inv()}.organize.id'],
        ),
        sa.ForeignKeyConstraint(
            ['id'],
            [f'{get_inv()}.action.id'],
        ),
        sa.ForeignKeyConstraint(
            ['price_id'],
            [f'{get_inv()}.price.id'],
        ),
        sa.ForeignKeyConstraint(
            ['to_id'],
            [f'{get_inv()}.agent.id'],
        ),
        sa.PrimaryKeyConstraint('id'),
        schema=f'{get_inv()}',
    )
    op.drop_column('user', 'active', schema='common')
    op.drop_column('user', 'phantom', schema='common')