moving alembic upgrade and downgrade functions. correcting error when I created join table migration before main table

This commit is contained in:
elicb 2022-05-05 10:12:39 -07:00 committed by Eli Barbieri
parent b9c628ce60
commit 9e617f5a7f
2 changed files with 36 additions and 34 deletions

View File

@ -15,29 +15,24 @@ branch_labels = None
depends_on = None depends_on = None
# This revision is switched with add_swap_jit_liquidity_table becasue I made them in the wrong order
def upgrade(): def upgrade():
op.create_table( op.create_table(
"jit_liquidity", "jit_liquidity_swaps",
sa.Column("id", sa.String, primary_key=True), sa.Column("created_at", sa.TIMESTAMP, server_default=sa.func.now()),
sa.Column("block_number", sa.Numeric(), nullable=False), sa.Column("jit_liquidity_id", sa.String(1024), primary_key=True),
sa.Column("bot_address", sa.String(42), nullable=True), sa.Column("swap_transaction_hash", sa.String(66), primary_key=True),
sa.Column("pool_address", sa.String(42), nullable=False), sa.Column("swap_trace_address", sa.ARRAY(sa.Integer), primary_key=True),
sa.Column("token0_address", sa.String(42), nullable=True), sa.ForeignKeyConstraint(
sa.Column("token1_address", sa.String(42), nullable=True), ["jit_liquidity_id"], ["jit_liquidity.id"], ondelete="CASCADE"
sa.Column("mint_transaction_hash", sa.String(66), nullable=False), ),
sa.Column("mint_transaction_trace", sa.ARRAY(sa.Integer)), sa.ForeignKeyConstraint(
sa.Column("burn_transaction_hash", sa.String(66), nullable=False), ["swap_transaction_hash", "swap_trace_address"],
sa.Column("burn_transaction_trace", sa.ARRAY(sa.Integer)), ["swaps.transaction_hash", "swaps.trace_address"],
sa.Column("mint_token0_amount", sa.Numeric), ondelete="CASCADE",
sa.Column("mint_token1_amount", sa.Numeric), ),
sa.Column("burn_token0_amount", sa.Numeric),
sa.Column("burn_token1_amount", sa.Numeric),
sa.Column("token0_swap_volume", sa.Numeric),
sa.Column("token1_swap_volume", sa.Numeric),
) )
op.create_index("ix_jit_liquidity_block_number", "jit_liquidity", ["block_number"])
def downgrade(): def downgrade():
op.drop_index("ix_jit_liquidity_block_number") op.drop_table("jit_liquidity_swaps")
op.drop_table("jit_liquidity")

View File

@ -15,23 +15,30 @@ branch_labels = None
depends_on = None depends_on = None
# This revision was swapped with adding_jit_liquidity_table because I created revisions in wrong order
def upgrade(): def upgrade():
op.create_table( op.create_table(
"jit_liquidity_swaps", "jit_liquidity",
sa.Column("created_at", sa.TIMESTAMP, server_default=sa.func.now()), sa.Column("id", sa.String, primary_key=True),
sa.Column("jit_liquidity_id", sa.String(1024), primary_key=True), sa.Column("block_number", sa.Numeric(), nullable=False),
sa.Column("swap_transaction_hash", sa.String(66), primary_key=True), sa.Column("bot_address", sa.String(42), nullable=True),
sa.Column("swap_trace_address", sa.ARRAY(sa.Integer), primary_key=True), sa.Column("pool_address", sa.String(42), nullable=False),
sa.ForeignKeyConstraint( sa.Column("token0_address", sa.String(42), nullable=True),
["jit_liquidity_id"], ["jit_liquidity.id"], ondelete="CASCADE" sa.Column("token1_address", sa.String(42), nullable=True),
), sa.Column("mint_transaction_hash", sa.String(66), nullable=False),
sa.ForeignKeyConstraint( sa.Column("mint_transaction_trace", sa.ARRAY(sa.Integer)),
["swap_transaction_hash", "swap_trace_address"], sa.Column("burn_transaction_hash", sa.String(66), nullable=False),
["swaps.transaction_hash", "swaps.trace_address"], sa.Column("burn_transaction_trace", sa.ARRAY(sa.Integer)),
ondelete="CASCADE", sa.Column("mint_token0_amount", sa.Numeric),
), sa.Column("mint_token1_amount", sa.Numeric),
sa.Column("burn_token0_amount", sa.Numeric),
sa.Column("burn_token1_amount", sa.Numeric),
sa.Column("token0_swap_volume", sa.Numeric),
sa.Column("token1_swap_volume", sa.Numeric),
) )
op.create_index("ix_jit_liquidity_block_number", "jit_liquidity", ["block_number"])
def downgrade(): def downgrade():
op.drop_table("jit_liquidity_swaps") op.drop_index("ix_jit_liquidity_block_number")
op.drop_table("jit_liquidity")