atst/alembic/versions/994a80ee92c9_add_invitations.py
2018-10-31 10:21:48 -04:00

51 lines
2.4 KiB
Python

"""add invitations
Revision ID: 994a80ee92c9
Revises: 9c24c609878a
Create Date: 2018-10-30 16:49:53.688621
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
# revision identifiers, used by Alembic.
revision = '994a80ee92c9'
down_revision = '9c24c609878a'
branch_labels = None
depends_on = None
def upgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.create_table('invitations',
sa.Column('time_created', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
sa.Column('time_updated', sa.TIMESTAMP(timezone=True), server_default=sa.text('now()'), nullable=False),
sa.Column('id', postgresql.UUID(as_uuid=True), server_default=sa.text('uuid_generate_v4()'), nullable=False),
sa.Column('user_id', postgresql.UUID(as_uuid=True), nullable=True),
sa.Column('workspace_role_id', postgresql.UUID(as_uuid=True), nullable=True),
sa.Column('inviter_id', postgresql.UUID(as_uuid=True), nullable=True),
sa.Column('status', sa.Enum('ACCEPTED', 'REVOKED', 'PENDING', 'REJECTED', name='status', native_enum=False), nullable=True),
sa.Column('expiration_time', sa.TIMESTAMP(timezone=True), nullable=True),
sa.Column('token', sa.String(), nullable=True),
sa.ForeignKeyConstraint(['inviter_id'], ['users.id'], ),
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
sa.ForeignKeyConstraint(['workspace_role_id'], ['workspace_roles.id'], ),
sa.PrimaryKeyConstraint('id')
)
op.create_index(op.f('ix_invitations_inviter_id'), 'invitations', ['inviter_id'], unique=False)
op.create_index(op.f('ix_invitations_token'), 'invitations', ['token'], unique=False)
op.create_index(op.f('ix_invitations_user_id'), 'invitations', ['user_id'], unique=False)
op.create_index(op.f('ix_invitations_workspace_role_id'), 'invitations', ['workspace_role_id'], unique=False)
# ### end Alembic commands ###
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.drop_index(op.f('ix_invitations_workspace_role_id'), table_name='invitations')
op.drop_index(op.f('ix_invitations_user_id'), table_name='invitations')
op.drop_index(op.f('ix_invitations_token'), table_name='invitations')
op.drop_index(op.f('ix_invitations_inviter_id'), table_name='invitations')
op.drop_table('invitations')
# ### end Alembic commands ###