# Copyright 2019 Tecnativa - Pedro M. Baeza
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
from openupgradelib import openupgrade
from psycopg2 import sql

_column_renames = {
    "account_asset_profile": [("parent_id", None)],
    "account_asset": [("parent_id", None)],
}


def move_view_assets(cr):
    """Copy view assets to other table for preserving them, but outside of the
    main table, so remove them from there.
    """
    temp_table = sql.Identifier(openupgrade.get_legacy_name("account_asset_view"))
    openupgrade.logged_query(
        cr,
        sql.SQL(
            """
        CREATE TABLE {} AS (
            SELECT * FROM account_asset
            WHERE type='view'
        )"""
        ).format(temp_table),
    )
    openupgrade.logged_query(cr, "DELETE FROM account_asset WHERE type='view'")


@openupgrade.migrate()
def migrate(env, version):
    if openupgrade.column_exists(env.cr, "account_asset", "parent_id"):
        # if migrating directly from v11 `account_asset` module, there are no
        # view assets nor parents
        openupgrade.rename_columns(env.cr, _column_renames)
        openupgrade.logged_query(
            env.cr,
            """
            ALTER TABLE account_asset
            DROP CONSTRAINT account_asset_parent_id_fkey""",
        )
        openupgrade.logged_query(
            env.cr,
            """
            ALTER TABLE account_asset_profile
            DROP CONSTRAINT account_asset_profile_parent_id_fkey""",
        )
        move_view_assets(env.cr)