"""Territory Worldwide

Revision ID: 0a8fb6ce1c0f
Revises: b10a4297c1ee
Create Date: 2019-06-05 18:32:55.690531

"""
from alembic import op
import sqlalchemy as sa

from models.territory_types import TerritoryType
from models.territories import Territory


# revision identifiers, used by Alembic.
revision = '0a8fb6ce1c0f'
down_revision = 'b10a4297c1ee'
branch_labels = None
depends_on = None


def upgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    bind = op.get_bind()
    session = sa.orm.Session(bind=bind)

    territory_type = TerritoryType()
    territory_type.id = 0
    territory_type.name = 'Worldwide'
    session.add(territory_type)

    territory = Territory()
    territory.id = 0
    territory.name = 'Worldwide'
    territory.territory_type_id = 0
    session.add(territory)

    (
        session.query(Territory)
        .filter(Territory.id <= 246, Territory.id > 0)
        .update({Territory.parent_id: 0}, synchronize_session=False)
    )

    session.commit()
    session.close()
    # ### end Alembic commands ###


def downgrade():
    # ### commands auto generated by Alembic - please adjust! ###
    bind = op.get_bind()
    session = sa.orm.Session(bind=bind)

    (
        session.query(Territory)
        .filter(Territory.id <= 246, Territory.id > 0)
        .update({Territory.parent_id: None}, synchronize_session=False)
    )

    (
        session.query(Territory)
        .filter(Territory.id == 0)
        .delete(synchronize_session=False)
    )

    (
        session.query(TerritoryType)
        .filter(TerritoryType.id == 0)
        .delete(synchronize_session=False)
    )

    session.commit()
    session.close()
    # ### end Alembic commands ###