diff --git a/alembic/versions/0660432b9840_create_classifications_table.py b/alembic/versions/0660432b9840_create_classifications_table.py new file mode 100644 index 0000000..bc4da99 --- /dev/null +++ b/alembic/versions/0660432b9840_create_classifications_table.py @@ -0,0 +1,63 @@ +"""Create classifications table + +Revision ID: 0660432b9840 +Revises: +Create Date: 2021-07-23 20:08:42.016711 + +""" +import sqlalchemy as sa +from sqlalchemy.dialects.postgresql import JSON +from alembic import op + +# revision identifiers, used by Alembic. +revision = "0660432b9840" +down_revision = None +branch_labels = None +depends_on = None + +CLASSIFICATION_TYPES = [ + "unknown", +] + +TRACE_TYPES = [ + "call", + "create", + "delegateCall", + "reward", + "suicide", +] + +PROTOCOL_TYPES = [ + "uniswap_v2", +] + + +def upgrade(): + op.create_table( + "classifications", + sa.Column("classified_at", sa.TIMESTAMP, server_default=sa.func.now()), + sa.Column("transaction_hash", sa.String(66), nullable=False), + sa.Column("block_number", sa.Integer, nullable=False), + sa.Column( + "classification_type", + sa.Enum(*CLASSIFICATION_TYPES, name="classification_type"), + nullable=False, + ), + sa.Column( + "trace_type", sa.Enum(*TRACE_TYPES, name="trace_type"), nullable=False + ), + sa.Column("trace_address", sa.String(256), nullable=False), + sa.Column("contract_address", sa.String(256), nullable=True), + sa.Column("protocol", sa.Enum(*PROTOCOL_TYPES, name="protocol"), nullable=True), + sa.Column("function_name", sa.String(2048), nullable=True), + sa.Column("function_signature", sa.String(2048), nullable=True), + sa.Column("input", JSON, nullable=True), + sa.Column("from_address", sa.String(256), nullable=True), + sa.Column("to_address", sa.String(256), nullable=True), + sa.Column("value", sa.Integer, nullable=True), + sa.PrimaryKeyConstraint("transaction_hash"), + ) + + +def downgrade(): + op.drop_table("classifications")