133 lines
6.1 KiB
Python

"""Initial schema
Revision ID: 001
Revises:
Create Date: 2025-05-14
"""
from alembic import op
import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision = '001'
down_revision = None
branch_labels = None
depends_on = None
def upgrade() -> None:
# Create assets table
op.create_table(
'assets',
sa.Column('id', sa.String(), nullable=False),
sa.Column('rank', sa.Integer(), nullable=True),
sa.Column('symbol', sa.String(), nullable=True),
sa.Column('name', sa.String(), nullable=True),
sa.Column('supply', sa.Float(), nullable=True),
sa.Column('max_supply', sa.Float(), nullable=True),
sa.Column('market_cap_usd', sa.Float(), nullable=True),
sa.Column('volume_usd_24hr', sa.Float(), nullable=True),
sa.Column('price_usd', sa.Float(), nullable=True),
sa.Column('change_percent_24hr', sa.Float(), nullable=True),
sa.Column('vwap_24hr', sa.Float(), nullable=True),
sa.Column('explorer', sa.String(), nullable=True),
sa.Column('last_updated', sa.DateTime(timezone=True), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=True),
sa.PrimaryKeyConstraint('id')
)
op.create_index(op.f('ix_assets_name'), 'assets', ['name'], unique=False)
op.create_index(op.f('ix_assets_symbol'), 'assets', ['symbol'], unique=False)
# Create asset_price_history table
op.create_table(
'asset_price_history',
sa.Column('id', sa.Integer(), nullable=False),
sa.Column('asset_id', sa.String(), nullable=True),
sa.Column('price_usd', sa.Float(), nullable=True),
sa.Column('timestamp', sa.DateTime(timezone=True), nullable=True),
sa.Column('interval', sa.String(), nullable=True),
sa.ForeignKeyConstraint(['asset_id'], ['assets.id'], ondelete='CASCADE'),
sa.PrimaryKeyConstraint('id')
)
op.create_index(op.f('ix_asset_price_history_asset_id'), 'asset_price_history', ['asset_id'], unique=False)
op.create_index(op.f('ix_asset_price_history_timestamp'), 'asset_price_history', ['timestamp'], unique=False)
# Create exchanges table
op.create_table(
'exchanges',
sa.Column('id', sa.String(), nullable=False),
sa.Column('name', sa.String(), nullable=True),
sa.Column('rank', sa.Integer(), nullable=True),
sa.Column('percent_total_volume', sa.Float(), nullable=True),
sa.Column('volume_usd', sa.Float(), nullable=True),
sa.Column('trading_pairs', sa.Integer(), nullable=True),
sa.Column('socket', sa.Boolean(), nullable=True),
sa.Column('exchange_url', sa.String(), nullable=True),
sa.Column('updated_timestamp', sa.Integer(), nullable=True),
sa.Column('last_updated', sa.DateTime(timezone=True), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=True),
sa.PrimaryKeyConstraint('id')
)
op.create_index(op.f('ix_exchanges_name'), 'exchanges', ['name'], unique=False)
op.create_index(op.f('ix_exchanges_rank'), 'exchanges', ['rank'], unique=False)
# Create markets table
op.create_table(
'markets',
sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
sa.Column('exchange_id', sa.String(), nullable=True),
sa.Column('rank', sa.Integer(), nullable=True),
sa.Column('base_id', sa.String(), nullable=True),
sa.Column('quote_id', sa.String(), nullable=True),
sa.Column('base_symbol', sa.String(), nullable=True),
sa.Column('quote_symbol', sa.String(), nullable=True),
sa.Column('price_quote', sa.Float(), nullable=True),
sa.Column('price_usd', sa.Float(), nullable=True),
sa.Column('volume_usd_24hr', sa.Float(), nullable=True),
sa.Column('percent_exchange_volume', sa.Float(), nullable=True),
sa.Column('trades_count_24hr', sa.Integer(), nullable=True),
sa.Column('updated_timestamp', sa.Integer(), nullable=True),
sa.Column('last_updated', sa.DateTime(timezone=True), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=True),
sa.PrimaryKeyConstraint('id')
)
op.create_index(op.f('ix_markets_base_id'), 'markets', ['base_id'], unique=False)
op.create_index(op.f('ix_markets_base_symbol'), 'markets', ['base_symbol'], unique=False)
op.create_index(op.f('ix_markets_exchange_id'), 'markets', ['exchange_id'], unique=False)
op.create_index(op.f('ix_markets_quote_id'), 'markets', ['quote_id'], unique=False)
op.create_index(op.f('ix_markets_quote_symbol'), 'markets', ['quote_symbol'], unique=False)
# Create rates table
op.create_table(
'rates',
sa.Column('id', sa.String(), nullable=False),
sa.Column('symbol', sa.String(), nullable=True),
sa.Column('currency_symbol', sa.String(), nullable=True),
sa.Column('type', sa.String(), nullable=True),
sa.Column('rate_usd', sa.Float(), nullable=True),
sa.Column('last_updated', sa.DateTime(timezone=True), server_default=sa.text('CURRENT_TIMESTAMP'), nullable=True),
sa.PrimaryKeyConstraint('id')
)
op.create_index(op.f('ix_rates_symbol'), 'rates', ['symbol'], unique=False)
def downgrade() -> None:
op.drop_index(op.f('ix_rates_symbol'), table_name='rates')
op.drop_table('rates')
op.drop_index(op.f('ix_markets_quote_symbol'), table_name='markets')
op.drop_index(op.f('ix_markets_quote_id'), table_name='markets')
op.drop_index(op.f('ix_markets_exchange_id'), table_name='markets')
op.drop_index(op.f('ix_markets_base_symbol'), table_name='markets')
op.drop_index(op.f('ix_markets_base_id'), table_name='markets')
op.drop_table('markets')
op.drop_index(op.f('ix_exchanges_rank'), table_name='exchanges')
op.drop_index(op.f('ix_exchanges_name'), table_name='exchanges')
op.drop_table('exchanges')
op.drop_index(op.f('ix_asset_price_history_timestamp'), table_name='asset_price_history')
op.drop_index(op.f('ix_asset_price_history_asset_id'), table_name='asset_price_history')
op.drop_table('asset_price_history')
op.drop_index(op.f('ix_assets_symbol'), table_name='assets')
op.drop_index(op.f('ix_assets_name'), table_name='assets')
op.drop_table('assets')