Tulir Asokan
49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
"""Add edit index to messages
|
|
|
|
Revision ID: 9e9c89b0b877
|
|
Revises: 17574c57f3f8
|
|
Create Date: 2019-05-29 15:28:23.128377
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = '9e9c89b0b877'
|
|
down_revision = '17574c57f3f8'
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade():
|
|
op.create_table('_message_temp',
|
|
sa.Column('mxid', sa.String),
|
|
sa.Column('mx_room', sa.String),
|
|
sa.Column('tgid', sa.Integer),
|
|
sa.Column('tg_space', sa.Integer),
|
|
sa.Column('edit_index', sa.Integer),
|
|
sa.PrimaryKeyConstraint('tgid', 'tg_space', 'edit_index'),
|
|
sa.UniqueConstraint("mxid", "mx_room", "tg_space", name="_mx_id_room_2"))
|
|
c = op.get_bind()
|
|
c.execute("INSERT INTO _message_temp (mxid, mx_room, tgid, tg_space, edit_index) "
|
|
"SELECT message.mxid, message.mx_room, message.tgid, message.tg_space, 0 "
|
|
"FROM message")
|
|
c.execute("DROP TABLE message")
|
|
c.execute("ALTER TABLE _message_temp RENAME TO message")
|
|
|
|
|
|
def downgrade():
|
|
op.create_table('_message_temp',
|
|
sa.Column('mxid', sa.String),
|
|
sa.Column('mx_room', sa.String),
|
|
sa.Column('tgid', sa.Integer),
|
|
sa.Column('tg_space', sa.Integer),
|
|
sa.PrimaryKeyConstraint('tgid', 'tg_space'),
|
|
sa.UniqueConstraint("mxid", "mx_room", "tg_space", name="_mx_id_room"))
|
|
c = op.get_bind()
|
|
c.execute("INSERT INTO _message_temp (mxid, mx_room, tgid, tg_space) "
|
|
"SELECT message.mxid, message.mx_room, message.tgid, message.tg_space "
|
|
"FROM message")
|
|
c.execute("DROP TABLE message")
|
|
c.execute("ALTER TABLE _message_temp RENAME TO message")
|