"""remove_owners_from_collaborators

Revision ID: a38b21bdc1ca
Revises: 4a52a3db8fc7
Create Date: 2021-08-05 10:46:32.495868

"""
from alembic import op
from operator import and_
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision = 'a38b21bdc1ca'
down_revision = 'ace39aa56831'
branch_labels = None
depends_on = None


def upgrade():
    bind = op.get_bind()
    meta = sa.MetaData(bind=bind)
    meta.reflect()

    user_project = meta.tables["UserProject"]

    projects = bind.execute(''' SELECT id, owner_id FROM "Project" ''').fetchall()

    for project_id, owner_id in projects:
        if owner_id:
            op.execute(
                user_project
                .delete()
                .where(and_(user_project.c.project_id == project_id, user_project.c.user_id == owner_id))
            )


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    pass
    # ### end Alembic commands ###