diff --git a/product_harmonized_system/models/product_category.py b/product_harmonized_system/models/product_category.py index 6f6233e..2da1e33 100644 --- a/product_harmonized_system/models/product_category.py +++ b/product_harmonized_system/models/product_category.py @@ -26,5 +26,5 @@ class ProductCategory(models.Model): elif self.parent_id: res = self.parent_id.get_hs_code_recursively() else: - res = None + res = self.env['hs.code'] return res diff --git a/product_harmonized_system/models/product_template.py b/product_harmonized_system/models/product_template.py index 8893f39..e803194 100644 --- a/product_harmonized_system/models/product_template.py +++ b/product_harmonized_system/models/product_template.py @@ -25,11 +25,13 @@ class ProductTemplate(models.Model): @api.multi def get_hs_code_recursively(self): - self.ensure_one() - if self.hs_code_id: - res = self.hs_code_id - elif self.categ_id: - res = self.categ_id.get_hs_code_recursively() + res = self.env['hs.code'] + if not self: + return res else: - res = None - return res + self.ensure_one() + if self.hs_code_id: + res = self.hs_code_id + elif self.categ_id: + res = self.categ_id.get_hs_code_recursively() + return res