python类JSONB的实例源码

def upgrade():
    """Upgrade the database to a newer revision."""
    # commands auto generated by Alembic - please adjust! ###
    op.create_table('analyses',
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('ecosystem', sa.Integer(), nullable=True),
                    sa.Column('package', sa.String(length=255), nullable=True),
                    sa.Column('version', sa.String(length=255), nullable=True),
                    sa.Column('access_count', sa.Integer(), nullable=True),
                    sa.Column('started_at', sa.DateTime(), nullable=True),
                    sa.Column('finished_at', sa.DateTime(), nullable=True),
                    sa.Column('analyses', postgresql.JSONB(), nullable=True),
                    sa.Column('subtasks', postgresql.JSONB(), nullable=True),
                    sa.Column('release', sa.String(length=255), nullable=True),
                    sa.Column('audit', postgresql.JSONB(), nullable=True),
                    sa.PrimaryKeyConstraint('id'))
    op.create_table('worker_results',
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('worker', sa.String(length=255), nullable=True),
                    sa.Column('analysis_id', sa.Integer(), nullable=True),
                    sa.Column('task_result', postgresql.JSONB(), nullable=True),
                    sa.ForeignKeyConstraint(['analysis_id'], ['analyses.id'], ),
                    sa.PrimaryKeyConstraint('id'))
    # end Alembic commands ###

def upgrade():
    """Upgrade the database to a newer revision."""
    # commands auto generated by Alembic - please adjust! ###
    op.create_table('batch',
                    sa.Column('token', sa.Integer(), nullable=False),
                    sa.Column('started_at', sa.DateTime(), nullable=True),
                    sa.Column('epvs', postgresql.JSONB(), nullable=True),
                    sa.PrimaryKeyConstraint('token'))
    op.create_table('review',
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('approved', sa.Boolean(), nullable=True),
                    sa.Column('user', sa.Integer(), nullable=True),
                    sa.Column('timestamp', sa.DateTime(), nullable=True),
                    sa.Column('comment', sa.Text(), nullable=True),
                    sa.Column('epv', sa.String(length=255), nullable=True),
                    sa.ForeignKeyConstraint(['user'], ['user.id'], ),
                    sa.PrimaryKeyConstraint('id'))
    # end Alembic commands ###

def upgrade():
    """Upgrade the database to a newer revision."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('stacks',
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('is_ref_stack', sa.Boolean(), nullable=False),
                    sa.Column('stack_json', postgresql.JSONB(), nullable=False),
                    sa.PrimaryKeyConstraint('id'))
    op.add_column('similar_stacks', sa.Column('analysis', postgresql.JSONB()))
    op.add_column('similar_stacks', sa.Column('similar_stack_id', sa.Integer(), nullable=False))
    op.add_column('similar_stacks', sa.Column('similarity_value', sa.Float(), nullable=False))
    op.add_column('similar_stacks', sa.Column('stack_id', sa.Integer(), nullable=False))
    op.create_unique_constraint('sim_unique', 'similar_stacks', ['stack_id', 'similar_stack_id'])
    op.drop_constraint('similar_stacks_appstack_id_fkey', 'similar_stacks', type_='foreignkey')
    op.create_foreign_key(None, 'similar_stacks', 'stacks', ['stack_id'], ['id'])
    op.create_foreign_key(None, 'similar_stacks', 'stacks', ['similar_stack_id'], ['id'])
    op.drop_column('similar_stacks', 'dependency_list')
    op.drop_column('similar_stacks', 'appstack_id')
    op.drop_table('reference_stacks')
    op.drop_table('app_stacks')
    # ### end Alembic commands ###

def upgrade():
    """Upgrade the database to a newer revision."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('batch',
                    sa.Column('token', sa.Integer(), nullable=False),
                    sa.Column('started_at', sa.DateTime(), nullable=True),
                    sa.Column('epvs', postgresql.JSONB(), nullable=True),
                    sa.PrimaryKeyConstraint('token'))
    op.create_table('review',
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('approved', sa.Boolean(), nullable=True),
                    sa.Column('user', sa.Integer(), nullable=True),
                    sa.Column('timestamp', sa.DateTime(), nullable=True),
                    sa.Column('comment', sa.Text(), nullable=True),
                    sa.Column('epv', sa.String(length=255), nullable=True),
                    sa.ForeignKeyConstraint(['user'], ['user.id'], ),
                    sa.PrimaryKeyConstraint('id'))
    # ### end Alembic commands ###

def upgrade():
    op.create_table('fda_dap',

        # Meta

        sa.Column('meta_id', sa.Text, unique=True),
        sa.Column('meta_source', sa.Text),
        sa.Column('meta_created', sa.DateTime(timezone=True)),
        sa.Column('meta_updated', sa.DateTime(timezone=True)),

        # General

        sa.Column('id', sa.Text, unique=True),
        sa.Column('documents', JSONB),
        sa.Column('approval_type', sa.Text),
        sa.Column('supplement_number', sa.Integer),
        sa.Column('action_date', sa.Date),
        sa.Column('fda_application_num', sa.Text),
        sa.Column('notes', sa.Text),

    )

def upgrade():
    op.add_column(
        'dataset',
        sa.Column(
            'meta',
            postgresql.JSONB(),
            nullable=True
        )
    )
    op.execute('''
        UPDATE "dataset" SET
            meta = jsonb_set(coalesce(meta, '{}'), '{organization_id}', to_jsonb(organization_id))
        WHERE
            organization_id IS NOT NULL
    ''')
    op.drop_column('dataset', 'organization_id')

def upgrade():
    ### commands auto generated by Alembic - please adjust! ###
    op.add_column('rule', sa.Column('conditions', postgresql.JSONB(), nullable=True))
    op.add_column('rule', sa.Column('description', sa.String(), nullable=True))
    op.drop_column('rule', 'action')
    op.drop_column('rule', 'type')
    op.drop_column('rule', 'config')
    ### end Alembic commands ###

def downgrade():
    ### commands auto generated by Alembic - please adjust! ###
    op.add_column('rule', sa.Column('config', postgresql.JSONB(), autoincrement=False, nullable=True))
    op.add_column('rule', sa.Column('type', sa.VARCHAR(), autoincrement=False, nullable=False))
    op.add_column('rule', sa.Column('action', postgresql.ENUM(u'added', u'removed', u'both', name='rule_actions'), autoincrement=False, nullable=False))
    op.drop_column('rule', 'description')
    op.drop_column('rule', 'conditions')
    ### end Alembic commands ###

def upgrade():
    ### commands auto generated by Alembic - please adjust! ###
    op.create_table('rule',
    sa.Column('id', sa.Integer(), nullable=False),
    sa.Column('type', sa.String(), nullable=False),
    sa.Column('name', sa.String(), nullable=False),
    sa.Column('action', sa.Enum('added', 'removed', 'both', name='rule_actions'), nullable=False),
    sa.Column('alerters', postgresql.ARRAY(sa.String()), nullable=False),
    sa.Column('config', postgresql.JSONB(), nullable=True),
    sa.PrimaryKeyConstraint('id')
    )
    ### end Alembic commands ###

def upgrade():
    ### commands auto generated by Alembic - please adjust! ###
    op.add_column('node', sa.Column('node_info', postgresql.JSONB(), server_default='{}', nullable=False))
    op.drop_index('idx__rule__updated_at', table_name='rule')
    ### end Alembic commands ###

def downgrade():
    """Downgrade the database to an older revision."""
    # commands auto generated by Alembic - please adjust! ###
    op.add_column('batch', sa.Column('epvs', postgresql.JSONB(), autoincrement=False,
                  nullable=True))
    op.drop_table('batches_versions')
    # end Alembic commands ###

def upgrade():
    """Upgrade the database to a newer revision."""
    # commands auto generated by Alembic - please adjust! ###
    op.create_table('stack_analyses_request',
                    sa.Column('id', sa.String(length=64), nullable=False),
                    sa.Column('submitTime', sa.DateTime(), nullable=False),
                    sa.Column('startTime', sa.DateTime(), nullable=True),
                    sa.Column('endTime', sa.DateTime(), nullable=True),
                    sa.Column('requestJson', postgresql.JSONB(), nullable=False),
                    sa.PrimaryKeyConstraint('id'))
    # end Alembic commands ###

def upgrade():
    """Upgrade the database to a newer revision."""
    # commands auto generated by Alembic - please adjust! ###
    op.create_table('stacks',
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('is_ref_stack', sa.Boolean(), nullable=False),
                    sa.Column('stack_json', postgresql.JSONB(), nullable=False),
                    sa.PrimaryKeyConstraint('id'))
    op.add_column('similar_stacks', sa.Column('analysis', postgresql.JSONB()))
    op.add_column('similar_stacks', sa.Column('similar_stack_id', sa.Integer(), nullable=False))
    op.add_column('similar_stacks', sa.Column('similarity_value', sa.Float(), nullable=False))
    op.add_column('similar_stacks', sa.Column('stack_id', sa.Integer(), nullable=False))
    op.create_unique_constraint('sim_unique', 'similar_stacks', ['stack_id', 'similar_stack_id'])
    op.drop_constraint('similar_stacks_appstack_id_fkey', 'similar_stacks', type_='foreignkey')
    op.create_foreign_key(None, 'similar_stacks', 'stacks', ['stack_id'], ['id'])
    op.create_foreign_key(None, 'similar_stacks', 'stacks', ['similar_stack_id'], ['id'])
    op.drop_column('similar_stacks', 'dependency_list')
    op.drop_column('similar_stacks', 'appstack_id')
    op.drop_table('reference_stacks')
    op.drop_table('app_stacks')
    # end Alembic commands ###

def downgrade():
    """Downgrade the database to an older revision."""
    # commands auto generated by Alembic - please adjust! ###
    op.add_column('similar_stacks', sa.Column('appstack_id', sa.INTEGER(), autoincrement=False,
                  nullable=True))
    op.add_column('similar_stacks', sa.Column('dependency_list', postgresql.JSONB(),
                  autoincrement=False, nullable=False))
    op.drop_constraint(None, 'similar_stacks', type_='foreignkey')
    op.drop_constraint(None, 'similar_stacks', type_='foreignkey')
    op.create_foreign_key('similar_stacks_appstack_id_fkey', 'similar_stacks', 'app_stacks',
                          ['appstack_id'], ['id'])
    op.drop_constraint('sim_unique', 'similar_stacks', type_='unique')
    op.drop_column('similar_stacks', 'stack_id')
    op.drop_column('similar_stacks', 'similarity_value')
    op.drop_column('similar_stacks', 'similar_stack_id')
    op.drop_column('similar_stacks', 'analysis')
    op.create_table('app_stacks',
                    sa.Column('id', sa.INTEGER(), nullable=False),
                    sa.Column('stack_json', postgresql.JSONB(), autoincrement=False,
                              nullable=False),
                    sa.PrimaryKeyConstraint('id', name='app_stacks_pkey'))
    op.create_table('reference_stacks',
                    sa.Column('id', sa.INTEGER(), nullable=False),
                    sa.Column('name', sa.VARCHAR(length=255), autoincrement=False, nullable=False),
                    sa.Column('version', sa.VARCHAR(length=255), autoincrement=False,
                              nullable=False),
                    sa.Column('description', sa.TEXT(), autoincrement=False, nullable=False),
                    sa.Column('dependencies', postgresql.JSONB(), autoincrement=False,
                              nullable=False),
                    sa.PrimaryKeyConstraint('id', name='reference_stacks_pkey'),
                    sa.UniqueConstraint('name', 'version', name='stack_unique'))
    op.drop_table('stacks')
    # end Alembic commands ###

def upgrade():
    """Upgrade the database to a newer revision."""
    # commands auto generated by Alembic - please adjust! ###
    op.create_table('reference_stacks',
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('name', sa.String(length=255), nullable=False),
                    sa.Column('version', sa.String(length=255), nullable=False),
                    sa.Column('description', sa.Text(), nullable=False),
                    sa.Column('dependencies', postgresql.JSONB(), nullable=False),
                    sa.PrimaryKeyConstraint('id'),
                    sa.UniqueConstraint('name', 'version', name='stack_unique'))
    # end Alembic commands ###

def downgrade():
    """Downgrade the database to an older revision."""
    # commands auto generated by Alembic - please adjust! ###
    op.add_column('analyses', sa.Column('analyses', postgresql.JSONB(), autoincrement=False,
                  nullable=True))
    # end Alembic commands ###

def upgrade():
    """Upgrade the database to a newer revision."""
    # commands auto generated by Alembic - please adjust! ###
    op.alter_column('stack_analyses_request', 'requestJson',
                    existing_type=postgresql.JSONB(),
                    type_=sa.String(length=4096),
                    existing_nullable=False)
    # end Alembic commands ###

def downgrade():
    """Downgrade the database to an older revision."""
    # commands auto generated by Alembic - please adjust! ###
    op.alter_column('stack_analyses_request', 'requestJson',
                    existing_type=sa.String(length=4096),
                    type_=postgresql.JSONB(),
                    existing_nullable=False)
    # end Alembic commands ###

def load_dialect_impl(self, dialect):
        if dialect.name == 'postgresql':
            # Use the native JSON type.
            return dialect.type_descriptor(_JSONB())
        else:
            return dialect.type_descriptor(self.impl)

def downgrade():
    """Downgrade the database to an older revision."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('batch', sa.Column('epvs', postgresql.JSONB(), autoincrement=False,
                  nullable=True))
    op.drop_table('batches_versions')
    # ### end Alembic commands ###

def upgrade():
    """Upgrade the database to a newer revision."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('stack_analyses_request',
                    sa.Column('id', sa.String(length=64), nullable=False),
                    sa.Column('submitTime', sa.DateTime(), nullable=False),
                    sa.Column('startTime', sa.DateTime(), nullable=True),
                    sa.Column('endTime', sa.DateTime(), nullable=True),
                    sa.Column('requestJson', postgresql.JSONB(), nullable=False),
                    sa.PrimaryKeyConstraint('id'))
    # ### end Alembic commands ###

def downgrade():
    """Downgrade the database to an older revision."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('similar_stacks', sa.Column('appstack_id', sa.INTEGER(), autoincrement=False,
                  nullable=True))
    op.add_column('similar_stacks', sa.Column('dependency_list', postgresql.JSONB(),
                  autoincrement=False, nullable=False))
    op.drop_constraint(None, 'similar_stacks', type_='foreignkey')
    op.drop_constraint(None, 'similar_stacks', type_='foreignkey')
    op.create_foreign_key('similar_stacks_appstack_id_fkey', 'similar_stacks', 'app_stacks',
                          ['appstack_id'], ['id'])
    op.drop_constraint('sim_unique', 'similar_stacks', type_='unique')
    op.drop_column('similar_stacks', 'stack_id')
    op.drop_column('similar_stacks', 'similarity_value')
    op.drop_column('similar_stacks', 'similar_stack_id')
    op.drop_column('similar_stacks', 'analysis')
    op.create_table('app_stacks',
                    sa.Column('id', sa.INTEGER(), nullable=False),
                    sa.Column('stack_json', postgresql.JSONB(), autoincrement=False,
                              nullable=False),
                    sa.PrimaryKeyConstraint('id', name='app_stacks_pkey'))
    op.create_table('reference_stacks',
                    sa.Column('id', sa.INTEGER(), nullable=False),
                    sa.Column('name', sa.VARCHAR(length=255), autoincrement=False,
                              nullable=False),
                    sa.Column('version', sa.VARCHAR(length=255), autoincrement=False,
                              nullable=False),
                    sa.Column('description', sa.TEXT(), autoincrement=False, nullable=False),
                    sa.Column('dependencies', postgresql.JSONB(), autoincrement=False,
                              nullable=False),
                    sa.PrimaryKeyConstraint('id', name='reference_stacks_pkey'),
                    sa.UniqueConstraint('name', 'version', name='stack_unique'))
    op.drop_table('stacks')
    # ### end Alembic commands ###

def upgrade():
    """Upgrade the database to a newer revision."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('monitored_upstreams',
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('package_id', sa.Integer(), nullable=True),
                    sa.Column('url', sa.String(length=255), nullable=False),
                    sa.Column('updated_at', sa.DateTime(), nullable=True),
                    sa.Column('added_at', sa.DateTime(), nullable=False),
                    sa.Column('deactivated_at', sa.DateTime(), nullable=True),
                    sa.Column('active', sa.Boolean(), nullable=False),
                    sa.ForeignKeyConstraint(['package_id'], ['packages.id'], ),
                    sa.PrimaryKeyConstraint('id'))
    op.create_index(op.f('ix_monitored_upstreams_package_id'), 'monitored_upstreams',
                    ['package_id'], unique=False)
    op.create_table('package_analyses',
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('package_id', sa.Integer(), nullable=True),
                    sa.Column('started_at', sa.DateTime(), nullable=True),
                    sa.Column('finished_at', sa.DateTime(), nullable=True),
                    sa.ForeignKeyConstraint(['package_id'], ['packages.id'], ),
                    sa.PrimaryKeyConstraint('id'))
    op.create_index(op.f('ix_package_analyses_package_id'), 'package_analyses', ['package_id'],
                    unique=False)
    op.create_table('package_worker_results',
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('package_analysis_id', sa.Integer(), nullable=True),
                    sa.Column('worker', sa.String(length=255), nullable=True),
                    sa.Column('worker_id', sa.String(length=64), nullable=True),
                    sa.Column('external_request_id', sa.String(length=64), nullable=True),
                    sa.Column('task_result', postgresql.JSONB(astext_type=sa.Text()),
                              nullable=True),
                    sa.Column('error', sa.Boolean(), nullable=False),
                    sa.ForeignKeyConstraint(['package_analysis_id'], ['package_analyses.id'], ),
                    sa.PrimaryKeyConstraint('id'),
                    sa.UniqueConstraint('worker_id'))
    op.create_index(op.f('ix_package_worker_results_package_analysis_id'),
                    'package_worker_results', ['package_analysis_id'], unique=False)
    op.create_index(op.f('ix_package_worker_results_worker'), 'package_worker_results',
                    ['worker'], unique=False)
    # ### end Alembic commands ###

def upgrade():
    """Upgrade the database to a newer revision."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('stack_analyses_request', sa.Column('result', postgresql.JSONB(), nullable=True))
    op.add_column('stack_analyses_request', sa.Column('team', sa.String(length=64), nullable=True))
    # ### end Alembic commands ###

def upgrade():
    """Upgrade the database to a newer revision."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('reference_stacks',
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('name', sa.String(length=255), nullable=False),
                    sa.Column('version', sa.String(length=255), nullable=False),
                    sa.Column('description', sa.Text(), nullable=False),
                    sa.Column('dependencies', postgresql.JSONB(), nullable=False),
                    sa.PrimaryKeyConstraint('id'),
                    sa.UniqueConstraint('name', 'version', name='stack_unique'))
    # ### end Alembic commands ###

def downgrade():
    """Downgrade the database to an older revision."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.add_column('analyses', sa.Column('analyses', postgresql.JSONB(), autoincrement=False,
                  nullable=True))
    # ### end Alembic commands ###

def downgrade():
    """Downgrade the database to an older revision."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.alter_column('stack_analyses_request', 'requestJson',
                    existing_type=sa.String(length=4096),
                    type_=postgresql.JSONB(),
                    existing_nullable=False)
    # ### end Alembic commands ###

def upgrade():
    """Upgrade the database to a newer revision."""
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('app_stacks',
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('stack_json', postgresql.JSONB(), nullable=False),
                    sa.PrimaryKeyConstraint('id'))
    op.create_table('similar_stacks',
                    sa.Column('id', sa.Integer(), nullable=False),
                    sa.Column('appstack_id', sa.Integer(), nullable=True),
                    sa.Column('dependency_list', postgresql.JSONB(), nullable=False),
                    sa.ForeignKeyConstraint(['appstack_id'], ['app_stacks.id'], ),
                    sa.PrimaryKeyConstraint('id'))
    # ### end Alembic commands ###

def upgrade():
    op.create_table('document',
        sa.Column('id', sa.Integer(), nullable=False),
        sa.Column('data', postgresql.JSONB(), nullable=False),
        sa.PrimaryKeyConstraint('id')
    )

def load_dialect_impl(self, dialect):
        if dialect.name == 'postgresql':
            from sqlalchemy.dialects.postgresql import JSONB
            return dialect.type_descriptor(JSONB())
        else:
            return dialect.type_descriptor(VARCHAR())