44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
"""Add user metadata fields
|
|
|
|
Revision ID: 491a530df230
|
|
Revises: 6c7014e84a5e
|
|
Create Date: 2016-11-04 18:03:05.237408
|
|
|
|
"""
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = '491a530df230'
|
|
down_revision = '6c7014e84a5e'
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from util.migrate import UTF8CharField
|
|
|
|
def upgrade(tables):
|
|
### commands auto generated by Alembic - please adjust! ###
|
|
op.add_column('user', sa.Column('company', UTF8CharField(length=255), nullable=True))
|
|
op.add_column('user', sa.Column('name', UTF8CharField(length=255), nullable=True))
|
|
### end Alembic commands ###
|
|
|
|
op.bulk_insert(tables.userpromptkind,
|
|
[
|
|
{'name':'enter_name'},
|
|
{'name':'enter_company'},
|
|
])
|
|
|
|
|
|
def downgrade(tables):
|
|
### commands auto generated by Alembic - please adjust! ###
|
|
op.drop_column('user', 'name')
|
|
op.drop_column('user', 'company')
|
|
### end Alembic commands ###
|
|
|
|
op.execute(
|
|
(tables.userpromptkind.delete()
|
|
.where(tables.userpromptkind.c.name == op.inline_literal('enter_name')))
|
|
)
|
|
|
|
op.execute(
|
|
(tables.userpromptkind.delete()
|
|
.where(tables.userpromptkind.c.name == op.inline_literal('enter_company')))
|
|
)
|