271 lines
14 KiB
Python
271 lines
14 KiB
Python
"""initial
|
|
|
|
Revision ID: 882512221354
|
|
Revises:
|
|
Create Date: 2026-02-06 22:48:52.480626
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = '882512221354'
|
|
down_revision: Union[str, None] = None
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.create_table('etf_prices',
|
|
sa.Column('ticker', sa.String(length=20), nullable=False),
|
|
sa.Column('date', sa.Date(), nullable=False),
|
|
sa.Column('close', sa.Numeric(precision=12, scale=2), nullable=False),
|
|
sa.Column('nav', sa.Numeric(precision=12, scale=2), nullable=True),
|
|
sa.Column('volume', sa.BigInteger(), nullable=True),
|
|
sa.PrimaryKeyConstraint('ticker', 'date')
|
|
)
|
|
op.create_table('etfs',
|
|
sa.Column('ticker', sa.String(length=20), nullable=False),
|
|
sa.Column('name', sa.String(length=100), nullable=False),
|
|
sa.Column('asset_class', sa.Enum('EQUITY', 'BOND', 'GOLD', 'MIXED', name='assetclass'), nullable=False),
|
|
sa.Column('market', sa.String(length=20), nullable=False),
|
|
sa.Column('expense_ratio', sa.Numeric(precision=5, scale=4), nullable=True),
|
|
sa.PrimaryKeyConstraint('ticker')
|
|
)
|
|
op.create_table('financials',
|
|
sa.Column('ticker', sa.String(length=20), nullable=False),
|
|
sa.Column('base_date', sa.Date(), nullable=False),
|
|
sa.Column('report_type', sa.Enum('ANNUAL', 'QUARTERLY', name='reporttype'), nullable=False),
|
|
sa.Column('account', sa.String(length=50), nullable=False),
|
|
sa.Column('value', sa.Numeric(precision=20, scale=2), nullable=True),
|
|
sa.PrimaryKeyConstraint('ticker', 'base_date', 'report_type', 'account')
|
|
)
|
|
op.create_table('job_logs',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('job_name', sa.String(length=50), nullable=False),
|
|
sa.Column('status', sa.String(length=20), nullable=False),
|
|
sa.Column('started_at', sa.DateTime(), nullable=True),
|
|
sa.Column('finished_at', sa.DateTime(), nullable=True),
|
|
sa.Column('records_count', sa.Integer(), nullable=True),
|
|
sa.Column('error_msg', sa.Text(), nullable=True),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_job_logs_id'), 'job_logs', ['id'], unique=False)
|
|
op.create_table('prices',
|
|
sa.Column('ticker', sa.String(length=20), nullable=False),
|
|
sa.Column('date', sa.Date(), nullable=False),
|
|
sa.Column('open', sa.Numeric(precision=12, scale=2), nullable=False),
|
|
sa.Column('high', sa.Numeric(precision=12, scale=2), nullable=False),
|
|
sa.Column('low', sa.Numeric(precision=12, scale=2), nullable=False),
|
|
sa.Column('close', sa.Numeric(precision=12, scale=2), nullable=False),
|
|
sa.Column('volume', sa.BigInteger(), nullable=False),
|
|
sa.PrimaryKeyConstraint('ticker', 'date')
|
|
)
|
|
op.create_table('sectors',
|
|
sa.Column('ticker', sa.String(length=20), nullable=False),
|
|
sa.Column('sector_code', sa.String(length=10), nullable=False),
|
|
sa.Column('company_name', sa.String(length=100), nullable=False),
|
|
sa.Column('sector_name', sa.String(length=50), nullable=False),
|
|
sa.Column('base_date', sa.Date(), nullable=False),
|
|
sa.PrimaryKeyConstraint('ticker')
|
|
)
|
|
op.create_table('stocks',
|
|
sa.Column('ticker', sa.String(length=20), nullable=False),
|
|
sa.Column('name', sa.String(length=100), nullable=False),
|
|
sa.Column('market', sa.String(length=20), nullable=False),
|
|
sa.Column('close_price', sa.Numeric(precision=12, scale=2), nullable=True),
|
|
sa.Column('market_cap', sa.BigInteger(), nullable=True),
|
|
sa.Column('eps', sa.Numeric(precision=12, scale=2), nullable=True),
|
|
sa.Column('forward_eps', sa.Numeric(precision=12, scale=2), nullable=True),
|
|
sa.Column('bps', sa.Numeric(precision=12, scale=2), nullable=True),
|
|
sa.Column('dividend_per_share', sa.Numeric(precision=12, scale=2), nullable=True),
|
|
sa.Column('stock_type', sa.Enum('COMMON', 'SPAC', 'PREFERRED', 'REIT', 'OTHER', name='stocktype'), nullable=True),
|
|
sa.Column('base_date', sa.Date(), nullable=False),
|
|
sa.PrimaryKeyConstraint('ticker')
|
|
)
|
|
op.create_table('users',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('username', sa.String(length=50), nullable=False),
|
|
sa.Column('email', sa.String(length=100), nullable=False),
|
|
sa.Column('hashed_password', sa.String(length=255), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), nullable=True),
|
|
sa.Column('updated_at', sa.DateTime(), nullable=True),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_users_email'), 'users', ['email'], unique=True)
|
|
op.create_index(op.f('ix_users_id'), 'users', ['id'], unique=False)
|
|
op.create_index(op.f('ix_users_username'), 'users', ['username'], unique=True)
|
|
op.create_table('valuations',
|
|
sa.Column('ticker', sa.String(length=20), nullable=False),
|
|
sa.Column('base_date', sa.Date(), nullable=False),
|
|
sa.Column('per', sa.Numeric(precision=10, scale=2), nullable=True),
|
|
sa.Column('pbr', sa.Numeric(precision=10, scale=2), nullable=True),
|
|
sa.Column('psr', sa.Numeric(precision=10, scale=2), nullable=True),
|
|
sa.Column('pcr', sa.Numeric(precision=10, scale=2), nullable=True),
|
|
sa.Column('dividend_yield', sa.Numeric(precision=6, scale=2), nullable=True),
|
|
sa.PrimaryKeyConstraint('ticker', 'base_date')
|
|
)
|
|
op.create_table('backtests',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('user_id', sa.Integer(), nullable=False),
|
|
sa.Column('strategy_type', sa.String(length=50), nullable=False),
|
|
sa.Column('strategy_params', sa.JSON(), nullable=True),
|
|
sa.Column('start_date', sa.Date(), nullable=False),
|
|
sa.Column('end_date', sa.Date(), nullable=False),
|
|
sa.Column('rebalance_period', sa.Enum('MONTHLY', 'QUARTERLY', 'SEMI_ANNUAL', 'ANNUAL', name='rebalanceperiod'), nullable=True),
|
|
sa.Column('initial_capital', sa.Numeric(precision=20, scale=2), nullable=False),
|
|
sa.Column('commission_rate', sa.Numeric(precision=10, scale=6), nullable=True),
|
|
sa.Column('slippage_rate', sa.Numeric(precision=10, scale=6), nullable=True),
|
|
sa.Column('benchmark', sa.String(length=20), nullable=True),
|
|
sa.Column('top_n', sa.Integer(), nullable=True),
|
|
sa.Column('status', sa.Enum('PENDING', 'RUNNING', 'COMPLETED', 'FAILED', name='backteststatus'), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(), nullable=True),
|
|
sa.Column('completed_at', sa.DateTime(), nullable=True),
|
|
sa.Column('error_message', sa.Text(), nullable=True),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_backtests_id'), 'backtests', ['id'], unique=False)
|
|
op.create_table('portfolios',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('user_id', sa.Integer(), nullable=False),
|
|
sa.Column('name', sa.String(length=100), nullable=False),
|
|
sa.Column('portfolio_type', sa.Enum('PENSION', 'GENERAL', name='portfoliotype'), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(), nullable=True),
|
|
sa.Column('updated_at', sa.DateTime(), nullable=True),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_portfolios_id'), 'portfolios', ['id'], unique=False)
|
|
op.create_table('backtest_equity_curve',
|
|
sa.Column('backtest_id', sa.Integer(), nullable=False),
|
|
sa.Column('date', sa.Date(), nullable=False),
|
|
sa.Column('portfolio_value', sa.Numeric(precision=20, scale=2), nullable=True),
|
|
sa.Column('benchmark_value', sa.Numeric(precision=20, scale=2), nullable=True),
|
|
sa.Column('drawdown', sa.Numeric(precision=10, scale=4), nullable=True),
|
|
sa.ForeignKeyConstraint(['backtest_id'], ['backtests.id'], ),
|
|
sa.PrimaryKeyConstraint('backtest_id', 'date')
|
|
)
|
|
op.create_table('backtest_holdings',
|
|
sa.Column('backtest_id', sa.Integer(), nullable=False),
|
|
sa.Column('rebalance_date', sa.Date(), nullable=False),
|
|
sa.Column('ticker', sa.String(length=20), nullable=False),
|
|
sa.Column('name', sa.String(length=100), nullable=True),
|
|
sa.Column('weight', sa.Numeric(precision=10, scale=4), nullable=True),
|
|
sa.Column('shares', sa.Integer(), nullable=True),
|
|
sa.Column('price', sa.Numeric(precision=12, scale=2), nullable=True),
|
|
sa.ForeignKeyConstraint(['backtest_id'], ['backtests.id'], ),
|
|
sa.PrimaryKeyConstraint('backtest_id', 'rebalance_date', 'ticker')
|
|
)
|
|
op.create_table('backtest_results',
|
|
sa.Column('backtest_id', sa.Integer(), nullable=False),
|
|
sa.Column('total_return', sa.Numeric(precision=10, scale=4), nullable=True),
|
|
sa.Column('cagr', sa.Numeric(precision=10, scale=4), nullable=True),
|
|
sa.Column('mdd', sa.Numeric(precision=10, scale=4), nullable=True),
|
|
sa.Column('sharpe_ratio', sa.Numeric(precision=10, scale=4), nullable=True),
|
|
sa.Column('volatility', sa.Numeric(precision=10, scale=4), nullable=True),
|
|
sa.Column('benchmark_return', sa.Numeric(precision=10, scale=4), nullable=True),
|
|
sa.Column('excess_return', sa.Numeric(precision=10, scale=4), nullable=True),
|
|
sa.ForeignKeyConstraint(['backtest_id'], ['backtests.id'], ),
|
|
sa.PrimaryKeyConstraint('backtest_id')
|
|
)
|
|
op.create_table('backtest_transactions',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('backtest_id', sa.Integer(), nullable=False),
|
|
sa.Column('date', sa.Date(), nullable=False),
|
|
sa.Column('ticker', sa.String(length=20), nullable=False),
|
|
sa.Column('action', sa.String(length=10), nullable=False),
|
|
sa.Column('shares', sa.Integer(), nullable=False),
|
|
sa.Column('price', sa.Numeric(precision=12, scale=2), nullable=False),
|
|
sa.Column('commission', sa.Numeric(precision=12, scale=2), nullable=False),
|
|
sa.ForeignKeyConstraint(['backtest_id'], ['backtests.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_backtest_transactions_id'), 'backtest_transactions', ['id'], unique=False)
|
|
op.create_table('holdings',
|
|
sa.Column('portfolio_id', sa.Integer(), nullable=False),
|
|
sa.Column('ticker', sa.String(length=20), nullable=False),
|
|
sa.Column('quantity', sa.Integer(), nullable=False),
|
|
sa.Column('avg_price', sa.Numeric(precision=12, scale=2), nullable=False),
|
|
sa.ForeignKeyConstraint(['portfolio_id'], ['portfolios.id'], ),
|
|
sa.PrimaryKeyConstraint('portfolio_id', 'ticker')
|
|
)
|
|
op.create_table('portfolio_snapshots',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('portfolio_id', sa.Integer(), nullable=False),
|
|
sa.Column('total_value', sa.Numeric(precision=15, scale=2), nullable=False),
|
|
sa.Column('snapshot_date', sa.Date(), nullable=False),
|
|
sa.ForeignKeyConstraint(['portfolio_id'], ['portfolios.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_portfolio_snapshots_id'), 'portfolio_snapshots', ['id'], unique=False)
|
|
op.create_table('targets',
|
|
sa.Column('portfolio_id', sa.Integer(), nullable=False),
|
|
sa.Column('ticker', sa.String(length=20), nullable=False),
|
|
sa.Column('target_ratio', sa.Numeric(precision=5, scale=2), nullable=False),
|
|
sa.ForeignKeyConstraint(['portfolio_id'], ['portfolios.id'], ),
|
|
sa.PrimaryKeyConstraint('portfolio_id', 'ticker')
|
|
)
|
|
op.create_table('transactions',
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('portfolio_id', sa.Integer(), nullable=False),
|
|
sa.Column('ticker', sa.String(length=20), nullable=False),
|
|
sa.Column('tx_type', sa.Enum('BUY', 'SELL', name='transactiontype'), nullable=False),
|
|
sa.Column('quantity', sa.Integer(), nullable=False),
|
|
sa.Column('price', sa.Numeric(precision=12, scale=2), nullable=False),
|
|
sa.Column('executed_at', sa.DateTime(), nullable=False),
|
|
sa.Column('memo', sa.Text(), nullable=True),
|
|
sa.ForeignKeyConstraint(['portfolio_id'], ['portfolios.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index(op.f('ix_transactions_id'), 'transactions', ['id'], unique=False)
|
|
op.create_table('snapshot_holdings',
|
|
sa.Column('snapshot_id', sa.Integer(), nullable=False),
|
|
sa.Column('ticker', sa.String(length=20), nullable=False),
|
|
sa.Column('quantity', sa.Integer(), nullable=False),
|
|
sa.Column('price', sa.Numeric(precision=12, scale=2), nullable=False),
|
|
sa.Column('value', sa.Numeric(precision=15, scale=2), nullable=False),
|
|
sa.Column('current_ratio', sa.Numeric(precision=5, scale=2), nullable=False),
|
|
sa.ForeignKeyConstraint(['snapshot_id'], ['portfolio_snapshots.id'], ),
|
|
sa.PrimaryKeyConstraint('snapshot_id', 'ticker')
|
|
)
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade() -> None:
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.drop_table('snapshot_holdings')
|
|
op.drop_index(op.f('ix_transactions_id'), table_name='transactions')
|
|
op.drop_table('transactions')
|
|
op.drop_table('targets')
|
|
op.drop_index(op.f('ix_portfolio_snapshots_id'), table_name='portfolio_snapshots')
|
|
op.drop_table('portfolio_snapshots')
|
|
op.drop_table('holdings')
|
|
op.drop_index(op.f('ix_backtest_transactions_id'), table_name='backtest_transactions')
|
|
op.drop_table('backtest_transactions')
|
|
op.drop_table('backtest_results')
|
|
op.drop_table('backtest_holdings')
|
|
op.drop_table('backtest_equity_curve')
|
|
op.drop_index(op.f('ix_portfolios_id'), table_name='portfolios')
|
|
op.drop_table('portfolios')
|
|
op.drop_index(op.f('ix_backtests_id'), table_name='backtests')
|
|
op.drop_table('backtests')
|
|
op.drop_table('valuations')
|
|
op.drop_index(op.f('ix_users_username'), table_name='users')
|
|
op.drop_index(op.f('ix_users_id'), table_name='users')
|
|
op.drop_index(op.f('ix_users_email'), table_name='users')
|
|
op.drop_table('users')
|
|
op.drop_table('stocks')
|
|
op.drop_table('sectors')
|
|
op.drop_table('prices')
|
|
op.drop_index(op.f('ix_job_logs_id'), table_name='job_logs')
|
|
op.drop_table('job_logs')
|
|
op.drop_table('financials')
|
|
op.drop_table('etfs')
|
|
op.drop_table('etf_prices')
|
|
# ### end Alembic commands ###
|