###########################################################################
# Copyright (c), The AiiDA team. All rights reserved. #
# This file is part of the AiiDA code. #
# #
# The code is hosted on GitHub at https://github.com/aiidateam/aiida-core #
# For further information on the license, see the LICENSE.txt file #
# For further information please visit http://www.aiida.net #
###########################################################################
"""Initial main branch schema
This schema is mainly equivalent to the `main_0001` schema of the `psql_dos` backend.
The difference are:
1. Data types: the replacement of ``JSONB`` with ``JSON``, and ``UUID`` with ``CHAR(32)``.
2. Some more fields are nullable, to allow migrations from legacy to main.
The nullable fields are then filled with default values, and set to non-nullable, in subsequent migrations.
Revision ID: main_0000
Revises:
Create Date: 2021-02-02
"""
import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects.sqlite import JSON
revision = 'main_0000'
down_revision = None
branch_labels = ('main',)
depends_on = None
[docs]
def upgrade():
"""Migrations for the upgrade."""
op.create_table(
'db_dbcomputer',
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
sa.Column('uuid', sa.CHAR(32), nullable=False, unique=True),
sa.Column('label', sa.String(length=255), nullable=False, unique=True),
sa.Column('hostname', sa.String(length=255), nullable=False),
sa.Column('description', sa.Text(), nullable=False),
sa.Column('scheduler_type', sa.String(length=255), nullable=False),
sa.Column('transport_type', sa.String(length=255), nullable=False),
sa.Column('metadata', JSON(), nullable=False),
)
op.create_table(
'db_dbuser',
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
sa.Column('email', sa.String(length=254), nullable=False, unique=True),
sa.Column('first_name', sa.String(length=254), nullable=False),
sa.Column('last_name', sa.String(length=254), nullable=False),
sa.Column('institution', sa.String(length=254), nullable=False),
)
op.create_table(
'db_dbauthinfo',
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
sa.Column('aiidauser_id', sa.Integer(), nullable=False, index=True),
sa.Column('dbcomputer_id', sa.Integer(), nullable=False, index=True),
sa.Column('metadata', JSON(), nullable=False),
sa.Column('auth_params', JSON(), nullable=False),
sa.Column('enabled', sa.Boolean(), nullable=False),
sa.ForeignKeyConstraint(
['aiidauser_id'],
['db_dbuser.id'],
ondelete='CASCADE',
initially='DEFERRED',
deferrable=True,
),
sa.ForeignKeyConstraint(
['dbcomputer_id'],
['db_dbcomputer.id'],
ondelete='CASCADE',
initially='DEFERRED',
deferrable=True,
),
sa.UniqueConstraint('aiidauser_id', 'dbcomputer_id'),
)
op.create_table(
'db_dbgroup',
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
sa.Column('uuid', sa.CHAR(32), nullable=False, unique=True),
sa.Column('label', sa.String(length=255), nullable=False, index=True),
sa.Column('type_string', sa.String(length=255), nullable=False, index=True),
sa.Column('time', sa.DateTime(timezone=True), nullable=False),
sa.Column('description', sa.Text(), nullable=False),
sa.Column('extras', JSON(), nullable=False),
sa.Column('user_id', sa.Integer(), nullable=False, index=True),
sa.ForeignKeyConstraint(
['user_id'],
['db_dbuser.id'],
ondelete='CASCADE',
initially='DEFERRED',
deferrable=True,
),
sa.UniqueConstraint('label', 'type_string'),
)
op.create_table(
'db_dbnode',
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
sa.Column('uuid', sa.CHAR(32), nullable=False, unique=True),
sa.Column('node_type', sa.String(length=255), nullable=False, index=True),
sa.Column('process_type', sa.String(length=255), nullable=True, index=True),
sa.Column('label', sa.String(length=255), nullable=False, index=True),
sa.Column('description', sa.Text(), nullable=False),
sa.Column('ctime', sa.DateTime(timezone=True), nullable=False, index=True),
sa.Column('mtime', sa.DateTime(timezone=True), nullable=False, index=True),
sa.Column('attributes', JSON(), nullable=True),
sa.Column('extras', JSON(), nullable=True),
sa.Column('repository_metadata', JSON(), nullable=False),
sa.Column('dbcomputer_id', sa.Integer(), nullable=True, index=True),
sa.Column('user_id', sa.Integer(), nullable=False, index=True),
sa.ForeignKeyConstraint(
['dbcomputer_id'],
['db_dbcomputer.id'],
ondelete='RESTRICT',
initially='DEFERRED',
deferrable=True,
),
sa.ForeignKeyConstraint(
['user_id'],
['db_dbuser.id'],
ondelete='restrict',
initially='DEFERRED',
deferrable=True,
),
)
op.create_table(
'db_dbcomment',
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
sa.Column('uuid', sa.CHAR(32), nullable=False, unique=True),
sa.Column('dbnode_id', sa.Integer(), nullable=False, index=True),
sa.Column('ctime', sa.DateTime(timezone=True), nullable=False),
sa.Column('mtime', sa.DateTime(timezone=True), nullable=False),
sa.Column('user_id', sa.Integer(), nullable=False, index=True),
sa.Column('content', sa.Text(), nullable=False),
sa.ForeignKeyConstraint(
['dbnode_id'],
['db_dbnode.id'],
ondelete='CASCADE',
initially='DEFERRED',
deferrable=True,
),
sa.ForeignKeyConstraint(
['user_id'],
['db_dbuser.id'],
ondelete='CASCADE',
initially='DEFERRED',
deferrable=True,
),
)
op.create_table(
'db_dbgroup_dbnodes',
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
sa.Column('dbnode_id', sa.Integer(), nullable=False, index=True),
sa.Column('dbgroup_id', sa.Integer(), nullable=False, index=True),
sa.ForeignKeyConstraint(['dbgroup_id'], ['db_dbgroup.id'], initially='DEFERRED', deferrable=True),
sa.ForeignKeyConstraint(['dbnode_id'], ['db_dbnode.id'], initially='DEFERRED', deferrable=True),
sa.UniqueConstraint('dbgroup_id', 'dbnode_id'),
)
op.create_table(
'db_dblink',
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
sa.Column('input_id', sa.Integer(), nullable=False, index=True),
sa.Column('output_id', sa.Integer(), nullable=False, index=True),
sa.Column('label', sa.String(length=255), nullable=False, index=True),
sa.Column('type', sa.String(length=255), nullable=False, index=True),
sa.ForeignKeyConstraint(['input_id'], ['db_dbnode.id'], initially='DEFERRED', deferrable=True),
sa.ForeignKeyConstraint(
['output_id'],
['db_dbnode.id'],
ondelete='CASCADE',
initially='DEFERRED',
deferrable=True,
),
)
op.create_table(
'db_dblog',
sa.Column('id', sa.Integer(), nullable=False, primary_key=True),
sa.Column('uuid', sa.CHAR(32), nullable=False, unique=True),
sa.Column('time', sa.DateTime(timezone=True), nullable=False),
sa.Column('loggername', sa.String(length=255), nullable=False, index=True),
sa.Column('levelname', sa.String(length=50), nullable=False, index=True),
sa.Column('dbnode_id', sa.Integer(), nullable=False, index=True),
sa.Column('message', sa.Text(), nullable=False),
sa.Column('metadata', JSON(), nullable=False),
sa.ForeignKeyConstraint(
['dbnode_id'],
['db_dbnode.id'],
ondelete='CASCADE',
initially='DEFERRED',
deferrable=True,
),
)
[docs]
def downgrade():
"""Migrations for the downgrade."""
raise NotImplementedError('Downgrade of main_0000.')
