You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
135 lines
5.9 KiB
135 lines
5.9 KiB
"""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
|