From 5f9bd3a274fdad2d02caa04405be06b2f7979abd Mon Sep 17 00:00:00 2001 From: Luke Van Seters Date: Fri, 19 Nov 2021 08:42:08 -0500 Subject: [PATCH] Change transfers trace address to ARRAY --- ...change_transfers_trace_address_to_array.py | 47 +++++++++++++++++++ 1 file changed, 47 insertions(+) create mode 100644 alembic/versions/5427d62a2cc0_change_transfers_trace_address_to_array.py diff --git a/alembic/versions/5427d62a2cc0_change_transfers_trace_address_to_array.py b/alembic/versions/5427d62a2cc0_change_transfers_trace_address_to_array.py new file mode 100644 index 0000000..7ec3d78 --- /dev/null +++ b/alembic/versions/5427d62a2cc0_change_transfers_trace_address_to_array.py @@ -0,0 +1,47 @@ +"""Change transfers trace address to ARRAY + +Revision ID: 5427d62a2cc0 +Revises: d540242ae368 +Create Date: 2021-11-19 13:25:11.252774 + +""" +import sqlalchemy as sa +from alembic import op + + +# revision identifiers, used by Alembic. +revision = "5427d62a2cc0" +down_revision = "d540242ae368" +branch_labels = None +depends_on = None + + +def upgrade(): + op.drop_constraint("transfers_pkey", "transfers") + op.alter_column( + "transfers", + "trace_address", + type_=sa.ARRAY(sa.Integer), + nullable=False, + postgresql_using="trace_address::int[]", + ) + op.create_primary_key( + "transfers_pkey", + "transfers", + ["block_number", "transaction_hash", "trace_address"], + ) + + +def downgrade(): + op.drop_constraint("transfers_pkey", "transfers") + op.alter_column( + "transfers", + "trace_address", + type_=sa.String(256), + nullable=False, + ) + op.create_primary_key( + "transfers_pkey", + "transfers", + ["block_number", "transaction_hash", "trace_address"], + )