external/aurweb
0
0
Fork 0
mirror of https://gitlab.archlinux.org/archlinux/aurweb.git synced 2025-02-03 10:43:03 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Conditionally apply SSOAccountId migration to support misaligned databases

Browse source 
Closes: #34

Signed-off-by: Leonidas Spyropoulos <artafinde@gmail.com>
This commit is contained in:
Leonidas Spyropoulos 2021-05-18 13:04:20 +01:00 • committed by Kevin Morris
parent 82f6d2ce75
commit 0d68b914bf
1 changed files with 15 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
migrations/versions/ef39fcd6e1cd_add_sso_account_id_in_table_users.py
View file

@ -5,9 +5,9 @@ Revises: f47cad5d6d03
Create Date: 2020-06-08 10:04:13.898617 Create Date: 2020-06-08 10:04:13.898617
""" """
from alembic import op
import sqlalchemy as sa import sqlalchemy as sa
from alembic import op
from sqlalchemy.engine.reflection import Inspector
# revision identifiers, used by Alembic. # revision identifiers, used by Alembic.
revision = 'ef39fcd6e1cd' revision = 'ef39fcd6e1cd'
@ -16,15 +16,20 @@ branch_labels = None
depends_on = None depends_on = None
def table_has_column(table, column_name):
for element in Inspector.from_engine(op.get_bind()).get_columns(table):
if element.get('name') == column_name:
return True
return False
def upgrade(): def upgrade():
# ### commands auto generated by Alembic - please adjust! ### if not table_has_column('Users', 'SSOAccountID'):
op.add_column('Users', sa.Column('SSOAccountID', sa.String(length=255), nullable=True)) op.add_column('Users', sa.Column('SSOAccountID', sa.String(length=255), nullable=True))
op.create_unique_constraint(None, 'Users', ['SSOAccountID']) op.create_unique_constraint(None, 'Users', ['SSOAccountID'])
# ### end Alembic commands ###
def downgrade(): def downgrade():
# ### commands auto generated by Alembic - please adjust! ### if table_has_column('Users', 'SSOAccountID'):
op.drop_constraint('SSOAccountID', 'Users', type_='unique') op.drop_constraint('SSOAccountID', 'Users', type_='unique')
op.drop_column('Users', 'SSOAccountID') op.drop_column('Users', 'SSOAccountID')
# ### end Alembic commands ###