87 lines
3.3 KiB
Python
87 lines
3.3 KiB
Python
# 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')
|