mirror of
https://github.com/GSA/notifications-api.git
synced 2026-09-06 00:08:26 -04:00
reformat
This commit is contained in:
@@ -9,45 +9,67 @@ from alembic import op
|
||||
import sqlalchemy as sa
|
||||
from sqlalchemy.dialects import postgresql
|
||||
|
||||
revision = '0166_add_org_user_stuff'
|
||||
down_revision = '0164_add_organisation_to_service'
|
||||
revision = "0166_add_org_user_stuff"
|
||||
down_revision = "0164_add_organisation_to_service"
|
||||
|
||||
|
||||
def upgrade():
|
||||
op.create_table('invite_status_type',
|
||||
sa.Column('name', sa.String(), nullable=False),
|
||||
sa.PrimaryKeyConstraint('name')
|
||||
op.create_table(
|
||||
"invite_status_type",
|
||||
sa.Column("name", sa.String(), nullable=False),
|
||||
sa.PrimaryKeyConstraint("name"),
|
||||
)
|
||||
|
||||
op.execute("insert into invite_status_type values ('pending'), ('accepted'), ('cancelled')")
|
||||
|
||||
op.create_table('invited_organisation_users',
|
||||
sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
|
||||
sa.Column('email_address', sa.String(length=255), nullable=False),
|
||||
sa.Column('invited_by_id', postgresql.UUID(as_uuid=True), nullable=False),
|
||||
sa.Column('organisation_id', postgresql.UUID(as_uuid=True), nullable=False),
|
||||
sa.Column('created_at', sa.DateTime(), nullable=False),
|
||||
sa.Column('status', sa.String(), nullable=False),
|
||||
|
||||
sa.ForeignKeyConstraint(['invited_by_id'], ['users.id'], ),
|
||||
sa.ForeignKeyConstraint(['organisation_id'], ['organisation.id'], ),
|
||||
sa.ForeignKeyConstraint(['status'], ['invite_status_type.name'], ),
|
||||
sa.PrimaryKeyConstraint('id')
|
||||
op.execute(
|
||||
"insert into invite_status_type values ('pending'), ('accepted'), ('cancelled')"
|
||||
)
|
||||
|
||||
op.create_table('user_to_organisation',
|
||||
sa.Column('user_id', postgresql.UUID(as_uuid=True), nullable=True),
|
||||
sa.Column('organisation_id', postgresql.UUID(as_uuid=True), nullable=True),
|
||||
|
||||
sa.ForeignKeyConstraint(['organisation_id'], ['organisation.id'], ),
|
||||
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ),
|
||||
sa.UniqueConstraint('user_id', 'organisation_id', name='uix_user_to_organisation')
|
||||
op.create_table(
|
||||
"invited_organisation_users",
|
||||
sa.Column("id", postgresql.UUID(as_uuid=True), nullable=False),
|
||||
sa.Column("email_address", sa.String(length=255), nullable=False),
|
||||
sa.Column("invited_by_id", postgresql.UUID(as_uuid=True), nullable=False),
|
||||
sa.Column("organisation_id", postgresql.UUID(as_uuid=True), nullable=False),
|
||||
sa.Column("created_at", sa.DateTime(), nullable=False),
|
||||
sa.Column("status", sa.String(), nullable=False),
|
||||
sa.ForeignKeyConstraint(
|
||||
["invited_by_id"],
|
||||
["users.id"],
|
||||
),
|
||||
sa.ForeignKeyConstraint(
|
||||
["organisation_id"],
|
||||
["organisation.id"],
|
||||
),
|
||||
sa.ForeignKeyConstraint(
|
||||
["status"],
|
||||
["invite_status_type.name"],
|
||||
),
|
||||
sa.PrimaryKeyConstraint("id"),
|
||||
)
|
||||
|
||||
op.create_unique_constraint(None, 'organisation_to_service', columns=['service_id', 'organisation_id'])
|
||||
op.create_table(
|
||||
"user_to_organisation",
|
||||
sa.Column("user_id", postgresql.UUID(as_uuid=True), nullable=True),
|
||||
sa.Column("organisation_id", postgresql.UUID(as_uuid=True), nullable=True),
|
||||
sa.ForeignKeyConstraint(
|
||||
["organisation_id"],
|
||||
["organisation.id"],
|
||||
),
|
||||
sa.ForeignKeyConstraint(
|
||||
["user_id"],
|
||||
["users.id"],
|
||||
),
|
||||
sa.UniqueConstraint(
|
||||
"user_id", "organisation_id", name="uix_user_to_organisation"
|
||||
),
|
||||
)
|
||||
|
||||
op.create_unique_constraint(
|
||||
None, "organisation_to_service", columns=["service_id", "organisation_id"]
|
||||
)
|
||||
|
||||
|
||||
def downgrade():
|
||||
op.drop_table('user_to_organisation')
|
||||
op.drop_table('invited_organisation_users')
|
||||
op.drop_table('invite_status_type')
|
||||
op.drop_table("user_to_organisation")
|
||||
op.drop_table("invited_organisation_users")
|
||||
op.drop_table("invite_status_type")
|
||||
|
||||
Reference in New Issue
Block a user