2021-04-04 14:40:59 +02:00
|
|
|
"""Lock
|
2021-02-01 18:12:51 +01:00
|
|
|
|
2021-04-04 14:40:59 +02:00
|
|
|
Revision ID: 75d4767b3031
|
|
|
|
Revises: 1f1b3b641d08
|
|
|
|
Create Date: 2021-04-02 18:41:20.864265
|
2021-02-01 18:12:51 +01:00
|
|
|
|
|
|
|
"""
|
2021-04-23 23:02:51 +02:00
|
|
|
import datetime
|
2021-02-01 18:12:51 +01:00
|
|
|
from alembic import op
|
|
|
|
import sqlalchemy as sa
|
2021-04-23 23:02:51 +02:00
|
|
|
from cic_eth.db.enum import LockEnum
|
2021-08-29 10:43:12 +02:00
|
|
|
from cic_eth.encode import ZERO_ADDRESS_NORMAL
|
2021-02-01 18:12:51 +01:00
|
|
|
|
|
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
2021-04-04 14:40:59 +02:00
|
|
|
revision = '75d4767b3031'
|
|
|
|
down_revision = '1f1b3b641d08'
|
2021-02-01 18:12:51 +01:00
|
|
|
branch_labels = None
|
|
|
|
depends_on = None
|
|
|
|
|
|
|
|
|
|
|
|
def upgrade():
|
|
|
|
op.create_table(
|
|
|
|
'lock',
|
|
|
|
sa.Column('id', sa.Integer, primary_key=True),
|
2021-10-28 21:45:47 +02:00
|
|
|
sa.Column("address", sa.String, nullable=True),
|
2021-02-01 18:12:51 +01:00
|
|
|
sa.Column('blockchain', sa.String),
|
|
|
|
sa.Column("flags", sa.BIGINT(), nullable=False, default=0),
|
2021-04-23 23:02:51 +02:00
|
|
|
sa.Column("date_created", sa.DateTime, nullable=False, default=datetime.datetime.utcnow),
|
2021-04-04 14:40:59 +02:00
|
|
|
sa.Column("otx_id", sa.Integer, sa.ForeignKey('otx.id'), nullable=True),
|
2021-02-01 18:12:51 +01:00
|
|
|
)
|
|
|
|
op.create_index('idx_chain_address', 'lock', ['blockchain', 'address'], unique=True)
|
2021-08-29 10:43:12 +02:00
|
|
|
op.execute("INSERT INTO lock (address, date_created, blockchain, flags) VALUES('{}', '{}', '::', {})".format(ZERO_ADDRESS_NORMAL, datetime.datetime.utcnow(), LockEnum.INIT | LockEnum.SEND | LockEnum.QUEUE))
|
2021-02-01 18:12:51 +01:00
|
|
|
|
2021-04-04 14:40:59 +02:00
|
|
|
|
2021-02-01 18:12:51 +01:00
|
|
|
def downgrade():
|
|
|
|
op.drop_index('idx_chain_address')
|
|
|
|
op.drop_table('lock')
|