"""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 import String
from sqlalchemy.dialects import postgresql
from sqlalchemy.sql import column, table

# 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 ###
    # ### Manual Changes
    # Create an ad-hoc table to use for the insert statement.
    headers_table = table(
        "headers", column("database", String), column("spreadsheet", String)
    )

    # generated by
    # header_list = [('', h) for h in <COPIED SPREADSHEET HEADERS>]
    # fill db names in the first slot, recreate header_list
    # print(*[f"""{{"database":"{pair[0]}", "spreadsheet":"{pair[1]}"}}""" for pair in header_list], sep=",\n")

    # special care should be given to headers with multi-line names, for now manual is sufficient
    op.bulk_insert(
        headers_table,
        [
            {"database": "id", "spreadsheet": "Уникальный номер образца"},
            {"database": "fadr", "spreadsheet": "кордината X"},
            {"database": "internal_id", "spreadsheet": "координата Y"},
            {
                "database": "x_coord",
                "spreadsheet": """Категория в ГИС
Geology - керн, каменный материал
Soil - почва, грунты, морские осадки
Material - искуственные материалы, минералы""",
            },
            {
                "database": "y_coord",
                "spreadsheet": "Категория (Археология, Биология, Минералы)",
            },
            {"database": "gis_category", "spreadsheet": "Бассейн"},
            {"database": "category", "spreadsheet": "Месторождение"},
            {"database": "basin", "spreadsheet": "№ скважины"},
            {"database": "deposit", "spreadsheet": "Глубина, интервал, привязка"},
            {"database": "well", "spreadsheet": "Свита\пласт"},
            {"database": "depth", "spreadsheet": "Хозяин (ФИО)"},
            {"database": "stratum", "spreadsheet": "Организация"},
            {"database": "owner", "spreadsheet": "Контакты хозяина"},
            {"database": "org", "spreadsheet": "перечень объектов/образцов"},
            {"database": "ownercontacts", "spreadsheet": "Описание объекта"},
            {"database": "samplelist", "spreadsheet": "Форма, размер"},
            {"database": "description", "spreadsheet": "Перечень данных"},
            {"database": "form_dimentions", "spreadsheet": "Разрешение, µm"},
            {"database": "datalist", "spreadsheet": "Дата съёмки"},
            {"database": "resolution", "spreadsheet": "Дополнительная информация"},
            {"database": "date", "spreadsheet": "Томограф"},
            {"database": "additional_info", "spreadsheet": "комментарий"},
        ],
    )

    # ### End Manual Changes


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 ###
    # ### No need to clean our Manual Changes since we drop the "headers" table whole