diff --git a/ereuse_devicehub/migrations/versions/97bef94f7982_add_wbid_user_in_snapshoterrors.py b/ereuse_devicehub/migrations/versions/97bef94f7982_add_wbid_user_in_snapshoterrors.py new file mode 100644 index 00000000..d65a6b86 --- /dev/null +++ b/ereuse_devicehub/migrations/versions/97bef94f7982_add_wbid_user_in_snapshoterrors.py @@ -0,0 +1,57 @@ +"""add wbid user in snapshotErrors + +Revision ID: 97bef94f7982 +Revises: 23d9e7ebbd7d +Create Date: 2022-04-12 09:27:59.670911 + +""" +import citext +import sqlalchemy as sa +from alembic import context, op +from sqlalchemy.dialects import postgresql + +# revision identifiers, used by Alembic. +revision = '97bef94f7982' +down_revision = '23d9e7ebbd7d' +branch_labels = None +depends_on = None + + +def get_inv(): + INV = context.get_x_argument(as_dictionary=True).get('inventory') + if not INV: + raise ValueError("Inventory value is not specified") + return INV + + +def upgrade(): + op.add_column( + 'snapshot_errors', + sa.Column('wbid', citext.CIText(), nullable=True), + schema=f'{get_inv()}', + ) + op.add_column( + 'snapshot_errors', + sa.Column('owner_id', postgresql.UUID(), nullable=True), + schema=f'{get_inv()}', + ) + op.create_foreign_key( + "fk_snapshot_errors_owner_id_user_id", + "snapshot_errors", + "user", + ["owner_id"], + ["id"], + ondelete="SET NULL", + source_schema=f'{get_inv()}', + referent_schema='common', + ) + + +def downgrade(): + op.drop_constraint( + "fk_snapshot_errors_owner_id_user_id", + "snapshot_errors", + type_="foreignkey", + schema=f'{get_inv()}', + ) + op.drop_column('snapshot_errors', 'owner_id', schema=f'{get_inv()}')