mirror of
https://github.com/jupyterhub/jupyterhub.git
synced 2025-10-17 23:13:00 +00:00
5e60582ef3
apologies to anyone finding this commit via git blame or log
run the autoformatting by
pre-commit run --all-files
43 lines
1.0 KiB
Python
43 lines
1.0 KiB
Python
"""Add session_id to auth tokens
|
|
|
|
Revision ID: 1cebaf56856c
|
|
Revises: 3ec6993fe20c
|
|
Create Date: 2017-12-07 14:43:51.500740
|
|
|
|
"""
|
|
# revision identifiers, used by Alembic.
|
|
revision = '1cebaf56856c'
|
|
down_revision = '3ec6993fe20c'
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
import logging
|
|
|
|
logger = logging.getLogger('alembic')
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
tables = ('oauth_access_tokens', 'oauth_codes')
|
|
|
|
|
|
def add_column_if_table_exists(table, column):
|
|
engine = op.get_bind().engine
|
|
if table not in engine.table_names():
|
|
# table doesn't exist, no need to upgrade
|
|
# because jupyterhub will create it on launch
|
|
logger.warning("Skipping upgrade of absent table: %s", table)
|
|
return
|
|
op.add_column(table, column)
|
|
|
|
|
|
def upgrade():
|
|
for table in tables:
|
|
add_column_if_table_exists(table, sa.Column('session_id', sa.Unicode(255)))
|
|
|
|
|
|
def downgrade():
|
|
# sqlite cannot downgrade because of limited ALTER TABLE support (no DROP COLUMN)
|
|
for table in tables:
|
|
op.drop_column(table, 'session_id')
|