diff options
Diffstat (limited to 'indoteknik_api/controllers/api_v1')
| -rw-r--r-- | indoteknik_api/controllers/api_v1/category.py | 91 |
1 files changed, 90 insertions, 1 deletions
diff --git a/indoteknik_api/controllers/api_v1/category.py b/indoteknik_api/controllers/api_v1/category.py index 82347196..a673e42f 100644 --- a/indoteknik_api/controllers/api_v1/category.py +++ b/indoteknik_api/controllers/api_v1/category.py @@ -32,7 +32,96 @@ class Category(controller.Controller): 'product.public.category', 'image_1920', category['id'] ) return self.response(categories) - + + @http.route(prefix + 'category/numFound', auth='public', methods=['GET', 'OPTIONS']) + @controller.Controller.must_authorized() + def get_category_numfound(self, **kw): + params = self.get_request_params(kw, { + 'parent_id': ['number', 'default:0'] + }) + if not params['valid']: + return self.response(code=400, description=params) + if params['value']['parent_id'] == 0: + params['value']['parent_id'] = False + + parent_id = params['value'].get('parent_id', 0) + if parent_id == 0: + parent_id = False + + def get_category_level(category_id): + """Return the level of the category based on its parent hierarchy.""" + category = request.env['product.public.category'].browse(category_id) + level = 0 + while category.parent_frontend_id: + category = category.parent_frontend_id + level += 1 + return level + + def compute_num_found(category, depth, start_depth): + """Recursively compute the total number of product templates in the category and its children starting from the given depth.""" + if depth > 2: + return 0 + child_categories = request.env['product.public.category'].search([('parent_frontend_id', '=', category.id)]) + num_found = len(category.product_tmpl_ids) + for child in child_categories: + num_found += compute_num_found(child, depth + 1, start_depth) + return num_found + + # Fetch the category and determine its level + category = request.env['product.public.category'].browse(parent_id) + category_level = get_category_level(category.id) if category else -1 + + if category: + start_depth = category_level + category_data = { + 'id': category.id, + 'name': category.name, + } + + if category_level == 0: + # Level 1 category, fetch child level 2 and their children level 3 + category_data['numFound'] = compute_num_found(category, 0, start_depth) + child_categories_level_2 = request.env['product.public.category'].search_read( + [('parent_frontend_id', '=', category.id)], + ['id', 'name'] + ) + for child in child_categories_level_2: + child_record = request.env['product.public.category'].browse(child['id']) + child['numFound'] = compute_num_found(child_record, 1, start_depth) + child['children'] = request.env['product.public.category'].search_read( + [('parent_frontend_id', '=', child['id'])], + ['id', 'name'] + ) + for grandchild in child['children']: + grandchild_record = request.env['product.public.category'].browse(grandchild['id']) + grandchild['numFound'] = compute_num_found(grandchild_record, 2, start_depth) + + category_data['children'] = child_categories_level_2 + + elif category_level == 1: + # Level 2 category, fetch child level 3 + category_data['numFound'] = compute_num_found(category, 1, start_depth) + child_categories_level_3 = request.env['product.public.category'].search_read( + [('parent_frontend_id', '=', category.id)], + ['id', 'name'] + ) + for child in child_categories_level_3: + child_record = request.env['product.public.category'].browse(child['id']) + child['numFound'] = compute_num_found(child_record, 2, start_depth) + + category_data['children'] = child_categories_level_3 + + else: + # Level 3 category or deeper, no children + category_data['numFound'] = compute_num_found(category, 2, start_depth) + category_data['children'] = [] + + response_data = category_data + else: + response_data = {} + + return self.response(response_data) + @http.route(prefix + 'category/tree', auth='public', methods=['GET', 'OPTIONS']) @controller.Controller.must_authorized() def get_category_tree(self): |