diff --git a/alembic/versions/3fca121b7554_added_tp_sl_and_pnl.py b/alembic/versions/3fca121b7554_added_tp_sl_and_pnl.py new file mode 100644 index 0000000..8b0d671 --- /dev/null +++ b/alembic/versions/3fca121b7554_added_tp_sl_and_pnl.py @@ -0,0 +1,44 @@ +"""Added TP_SL and PNL + +Revision ID: 3fca121b7554 +Revises: adf3d2991896 +Create Date: 2025-10-29 11:07:45.350771 + +""" +from typing import Sequence, Union + +from alembic import op +import sqlalchemy as sa +from sqlalchemy import inspect + + +# revision identifiers, used by Alembic. +revision: str = '3fca121b7554' +down_revision: Union[str, Sequence[str], None] = 'adf3d2991896' +branch_labels: Union[str, Sequence[str], None] = None +depends_on: Union[str, Sequence[str], None] = None + + +def upgrade() -> None: + """Upgrade schema.""" + conn = op.get_bind() + inspector = inspect(conn) + + columns = [col['name'] for col in inspector.get_columns('user_deals')] + + if 'pnl_series' not in columns: + op.add_column('user_deals', sa.Column('pnl_series', sa.Float(), nullable=True)) + if 'take_profit' not in columns: + op.add_column('user_deals', sa.Column('take_profit', sa.Boolean(), nullable=False, server_default=sa.false())) + if 'stop_loss' not in columns: + op.add_column('user_deals', sa.Column('stop_loss', sa.Boolean(), nullable=False, server_default=sa.false())) + # ### end Alembic commands ### + + +def downgrade() -> None: + """Downgrade schema.""" + # ### commands auto generated by Alembic - please adjust! ### + op.drop_column('user_deals', 'stop_loss') + op.drop_column('user_deals', 'take_profit') + op.drop_column('user_deals', 'pnl_series') + # ### end Alembic commands ### diff --git a/alembic/versions/8329c0994b26_fixed_tp_sl_type.py b/alembic/versions/8329c0994b26_fixed_tp_sl_type.py new file mode 100644 index 0000000..8e51ded --- /dev/null +++ b/alembic/versions/8329c0994b26_fixed_tp_sl_type.py @@ -0,0 +1,49 @@ +"""Fixed TP_SL type + +Revision ID: 8329c0994b26 +Revises: 3fca121b7554 +Create Date: 2025-10-29 13:07:52.161139 + +""" +from typing import Sequence, Union + +from alembic import op +import sqlalchemy as sa + + +# revision identifiers, used by Alembic. +revision: str = '8329c0994b26' +down_revision: Union[str, Sequence[str], None] = '3fca121b7554' +branch_labels: Union[str, Sequence[str], None] = None +depends_on: Union[str, Sequence[str], None] = None + + +def upgrade(): + with op.batch_alter_table('user_deals', recreate='always') as batch_op: + # Добавляем новую колонку с нужным типом + batch_op.add_column(sa.Column('take_profit_new', sa.Float(), nullable=False, server_default='0')) + + # После закрытия batch создается и переименовывается таблица. + # Теперь мы можем обновить данные. + op.execute( + "UPDATE user_deals SET take_profit_new = CAST(take_profit AS FLOAT)" + ) + + with op.batch_alter_table('user_deals', recreate='always') as batch_op: + # Удаляем старую колонку + batch_op.drop_column('take_profit') + # Меняем имя новой колонки на старое + batch_op.alter_column('take_profit_new', new_column_name='take_profit') + +def downgrade(): + # Аналогично, но в обратном порядке и типе + with op.batch_alter_table('user_deals', recreate='always') as batch_op: + batch_op.add_column(sa.Column('take_profit_old', sa.Boolean(), nullable=False, server_default='0')) + + op.execute( + "UPDATE user_deals SET take_profit_old = CAST(take_profit AS BOOLEAN)" + ) + + with op.batch_alter_table('user_deals', recreate='always') as batch_op: + batch_op.drop_column('take_profit') + batch_op.alter_column('take_profit_old', new_column_name='take_profit') \ No newline at end of file