# coding: utf-8
# Part of Odoo. See LICENSE file for full copyright and licensing details.

import logging
from os.path import join, dirname, realpath
from odoo import tools

_logger = logging.getLogger(__name__)


def post_init_hook(env):
    _load_unspsc_codes(env)
    _assign_codes_uom(env)
    demo_product = env.ref('product.consu_delivery_03', raise_if_not_found=False)
    if demo_product:
        _assign_codes_demo(env)

def uninstall_hook(env):
    env.cr.execute("DELETE FROM product_unspsc_code;")
    env.cr.execute("DELETE FROM ir_model_data WHERE model='product_unspsc_code';")

def _load_unspsc_codes(env):
    """Import CSV data as it is faster than xml and because we can't use
    noupdate anymore with csv
    Even with the faster CSVs, it would take +30 seconds to load it with
    the regular ORM methods, while here, it is under 3 seconds
    """
    csv_path = 'product_unspsc/data/product.unspsc.code.csv'
    with tools.misc.file_open(csv_path, 'rb') as csv_file:
        csv_file.readline() # Read the header, so we avoid copying it to the db
        env.cr.copy_expert(
            """COPY product_unspsc_code (code, name, applies_to, active)
               FROM STDIN WITH DELIMITER '|'""", csv_file)
    # Create xml_id, to allow make reference to this data
    env.cr.execute(
        """INSERT INTO ir_model_data
           (name, res_id, module, model, noupdate)
           SELECT concat('unspsc_code_', code), id, 'product_unspsc', 'product.unspsc.code', 't'
           FROM product_unspsc_code""")


def _insert_missing_unspsc_codes(cr):
    """Insert the CSV codes missing from the catalog, with their xml_id. The
    CSV is only imported on install, so upgrades rely on this."""
    cr.execute("""
        CREATE UNLOGGED TABLE _upgrade_product_unspsc_code (
            code varchar,
            name jsonb,
            applies_to varchar,
            active boolean
        )
    """)
    csv_path = 'product_unspsc/data/product.unspsc.code.csv'
    with tools.misc.file_open(csv_path, 'rb') as csv_file:
        csv_file.readline()  # Read the header, so we avoid copying it to the db
        cr.copy_from(csv_file, "_upgrade_product_unspsc_code", sep="|", columns=["code", "name", "applies_to", "active"])

    cr.execute("""
        WITH insert_cte AS (
            INSERT INTO product_unspsc_code (code, name, applies_to, active)
            SELECT code, name, applies_to, active
            FROM _upgrade_product_unspsc_code upc
            WHERE NOT EXISTS (
                    SELECT 1 FROM product_unspsc_code
                    WHERE code=upc.code
                    )
            RETURNING code, id
        )
        INSERT INTO ir_model_data(name, res_id, module, model, noupdate)
        SELECT 'unspsc_code_' || code, id,  'product_unspsc', 'product.unspsc.code', True
        FROM insert_cte
    """)

    cr.execute("DROP TABLE _upgrade_product_unspsc_code;")


def _assign_codes_uom(env):
    """Assign the codes in UoM of each data, this is here because the data is
    created in the last method"""
    tools.convert.convert_file(
        env, 'product_unspsc', 'data/product_data.xml', None, mode='init')

def _assign_codes_demo(env):
    """Assign the codes in the products used in demo invoices, this is here because the data is
    created in the last method"""
    tools.convert.convert_file(env, 'product_unspsc', 'demo/product_demo.xml', None, mode='init')