"""Create item model

Revision ID: b76b7f791b3b
Revises: 
Create Date: 2022-07-08 20:32:20.039600

"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql

# revision identifiers, used by Alembic.
revision = "b76b7f791b3b"
down_revision = None
branch_labels = None
depends_on = None


def upgrade() -> None:
    # ### commands auto generated by Alembic ###
    op.create_table(
        "geodata",
        sa.Column("id", sa.Integer(), nullable=False),
        sa.Column("fadr", sa.String(), nullable=True),
        sa.Column("internal_id", sa.String(), nullable=True),
        sa.Column("x_coord", sa.String(), nullable=True),
        sa.Column("y_coord", sa.String(), nullable=True),
        sa.Column("gis_category", sa.String(), nullable=True),
        sa.Column("category", sa.String(), nullable=True),
        sa.Column("basin", sa.String(), nullable=True),
        sa.Column("deposit", sa.String(), nullable=True),
        sa.Column("well", sa.String(), nullable=True),
        sa.Column("depth", sa.String(), nullable=True),
        sa.Column("stratum", sa.String(), nullable=True),
        sa.Column("owner", sa.String(), nullable=True),
        sa.Column("org", sa.String(), nullable=True),
        sa.Column("ownercontacts", sa.String(), nullable=True),
        sa.Column("samplelist", sa.String(), nullable=True),
        sa.Column("description", sa.String(), nullable=True),
        sa.Column("form_dimentions", sa.String(), nullable=True),
        sa.Column("datalist", sa.String(), nullable=True),
        sa.Column("resolution", sa.String(), nullable=True),
        sa.Column("date", sa.String(), nullable=True),
        sa.Column("additional_info", sa.String(), nullable=True),
        sa.Column("scanner", sa.String(), nullable=True),
        sa.Column("comment", sa.String(), nullable=True),
        sa.Column(
            "geodata_search_ts",
            postgresql.TSVECTOR(),
            sa.Computed(
                "to_tsvector('russian', coalesce(geodata.description, '') || ' ' || coalesce(geodata.additional_info, '') || ' ' || coalesce(geodata.comment, ''))",
                persisted=True,
            ),
            nullable=True,
        ),
        sa.PrimaryKeyConstraint("id"),
    )
    op.create_index(
        "geodata_search_idx",
        "geodata",
        ["geodata_search_ts"],
        unique=False,
        postgresql_using="gin",
    )
    op.create_index(op.f("ix_geodata_id"), "geodata", ["id"], unique=False)
    op.create_table(
        "headers",
        sa.Column("database", sa.String(), nullable=False),
        sa.Column("spreadsheet", sa.String(), nullable=True),
        sa.PrimaryKeyConstraint("database"),
    )
    # ### end Alembic commands ###


def downgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_table("headers")
    op.drop_index(op.f("ix_geodata_id"), table_name="geodata")
    op.drop_index("geodata_search_idx", table_name="geodata", postgresql_using="gin")
    op.drop_table("geodata")
    # ### end Alembic commands ###