galaxis-po/backend/alembic/versions/6c09aa4368e5_add_signals_table.py
zephyrdark a64636f6ff feat: add Signal model for KJB trading signals
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-02-19 15:05:17 +09:00

48 lines
2.0 KiB
Python

"""add signals table
Revision ID: 6c09aa4368e5
Revises: 882512221354
Create Date: 2026-02-19 12:00:00.000000
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision: str = '6c09aa4368e5'
down_revision: Union[str, None] = '882512221354'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
op.create_table('signals',
sa.Column('id', sa.Integer(), nullable=False),
sa.Column('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('signal_type', sa.Enum('BUY', 'SELL', 'PARTIAL_SELL', name='signaltype'), nullable=False),
sa.Column('entry_price', sa.Numeric(precision=12, scale=2), nullable=True),
sa.Column('target_price', sa.Numeric(precision=12, scale=2), nullable=True),
sa.Column('stop_loss_price', sa.Numeric(precision=12, scale=2), nullable=True),
sa.Column('reason', sa.String(length=200), nullable=True),
sa.Column('status', sa.Enum('ACTIVE', 'EXECUTED', 'EXPIRED', name='signalstatus'), nullable=True),
sa.Column('created_at', sa.DateTime(), nullable=True),
sa.PrimaryKeyConstraint('id'),
)
op.create_index(op.f('ix_signals_date'), 'signals', ['date'], unique=False)
op.create_index(op.f('ix_signals_id'), 'signals', ['id'], unique=False)
op.create_index(op.f('ix_signals_ticker'), 'signals', ['ticker'], unique=False)
def downgrade() -> None:
op.drop_index(op.f('ix_signals_ticker'), table_name='signals')
op.drop_index(op.f('ix_signals_id'), table_name='signals')
op.drop_index(op.f('ix_signals_date'), table_name='signals')
op.drop_table('signals')
sa.Enum('BUY', 'SELL', 'PARTIAL_SELL', name='signaltype').drop(op.get_bind(), checkfirst=True)
sa.Enum('ACTIVE', 'EXECUTED', 'EXPIRED', name='signalstatus').drop(op.get_bind(), checkfirst=True)