Merge branch 'production' of https://bitbucket.org/altafixco/indoteknik-addons into production

4 files changed, 123 insertions, 3 deletions

diff --git a/indoteknik_custom/models/solr/__init__.py b/indoteknik_custom/models/solr/__init__.py
index 606c0035..dafd5a1e 100644
--- a/indoteknik_custom/models/solr/__init__.py
+++ b/indoteknik_custom/models/solr/__init__.py
@@ -10,4 +10,5 @@ from . import x_banner_banner
 from . import product_public_category
 from . import x_banner_category
 from . import promotion_program
-from . import promotion_program_line
\ No newline at end of file
+from . import promotion_program_line
+from . import website_categories_management
\ No newline at end of file
diff --git a/indoteknik_custom/models/solr/product_template.py b/indoteknik_custom/models/solr/product_template.py
index d8dec47c..1eb6f31b 100644
--- a/indoteknik_custom/models/solr/product_template.py
+++ b/indoteknik_custom/models/solr/product_template.py
@@ -112,8 +112,8 @@ class ProductTemplate(models.Model):
                 "description_clean_t": cleaned_desc or '',
                 'has_product_info_b': True,
                 'publish_b': not template.unpublished,
-                'sni_b': template.unpublished,
-                'tkdn_b': template.unpublished,
+                'sni_b': template.sni,
+                'tkdn_b': template.tkdn,
                 "qty_sold_f": template.qty_sold,
                 "is_in_bu_b": is_in_bu,
                 "voucher_min_purchase_f" : voucher.min_purchase_amount or 0,
diff --git a/indoteknik_custom/models/solr/promotion_program_line.py b/indoteknik_custom/models/solr/promotion_program_line.py
index 3e3a2a28..64ad4209 100644
--- a/indoteknik_custom/models/solr/promotion_program_line.py
+++ b/indoteknik_custom/models/solr/promotion_program_line.py
@@ -53,6 +53,11 @@ class PromotionProgramLine(models.Model):
                 'package_limit_user_i': rec.package_limit_user,
                 'package_limit_trx_i': rec.package_limit_trx,
                 'price_f': rec.price,
+                'price_tier_1_f': rec.price_tier_1,
+                'price_tier_2_f': rec.price_tier_2,
+                'price_tier_3_f': rec.price_tier_3,
+                'price_tier_4_f': rec.price_tier_4,
+                'price_tier_5_f': rec.price_tier_5,
                 'sequence_i': sequence_value,
                 'product_ids': [x.product_id.id for x in rec.product_ids],
                 'products_s': json.dumps(products),
diff --git a/indoteknik_custom/models/solr/website_categories_management.py b/indoteknik_custom/models/solr/website_categories_management.py
new file mode 100644
index 00000000..0a40a356
--- /dev/null
+++ b/indoteknik_custom/models/solr/website_categories_management.py
@@ -0,0 +1,114 @@
+from odoo import models, fields, api
+from datetime import datetime
+import json
+
+class WebsiteCategoriesHomepage(models.Model):
+    _inherit = 'website.categories.management'
+
+    last_update_solr = fields.Datetime('Last Update Solr')
+
+    def solr(self):
+        """Returns the Solr connection object."""
+        return self.env['apache.solr'].connect('category_management')
+
+    def update_last_update_solr(self):
+        """Updates the last sync time for the record."""
+        self.last_update_solr = datetime.utcnow()
+
+    def _create_solr_queue(self, function_name):
+        """Creates unique Solr queue for each record."""
+        for rec in self:
+            self.env['apache.solr.queue'].create_unique({
+                'res_model': self._name,
+                'res_id': rec.id,
+                'function_name': function_name
+            })
+
+    @api.constrains('status')
+    def _create_solr_queue_sync_status(self):
+        """Triggers Solr sync when the status changes."""
+        self._create_solr_queue('_sync_status_category_homepage_solr')
+
+    @api.constrains('category_id', 'category_id2', 'sequence')
+    def _create_solr_queue_sync_category_homepage(self):
+        """Triggers Solr sync when categories or sequence change."""
+        self._create_solr_queue('_sync_category_management_to_solr')
+
+    def action_sync_to_solr(self):
+        """Manual action to sync selected categories to Solr."""
+        category_ids = self.env.context.get('active_ids', [])
+        categories = self.search([('id', 'in', category_ids)])
+        categories._create_solr_queue('_sync_category_management_to_solr')
+
+    def unlink(self):
+        """Overrides unlink method to remove records from Solr."""
+        for rec in self:
+            self.solr().delete(rec.id)
+        self.solr().optimize()
+        self.solr().commit()
+        return super(WebsiteCategoriesHomepage, self).unlink()
+
+    def _sync_status_category_homepage_solr(self):
+        """Syncs status to Solr or deletes if not active."""
+        for rec in self:
+            if rec.status == 'tayang':
+                rec._sync_category_management_to_solr()
+            else:
+                rec.unlink()
+
+    def _sync_category_management_to_solr(self):
+        """Syncs categories (Level 1, 2, and 3) to Solr."""
+        solr_model = self.env['apache.solr']
+
+        for category in self:
+            if category.status != 'tayang':
+                continue
+
+            # Prepare Level 1 document
+            document = {
+                'id': category.id,
+                'sequence_i': category.sequence or '',
+                'category_id_i': category.category_id.id,
+                'name_s': category.category_id.name,
+                'numFound_i': len(category.category_id.product_tmpl_ids),
+                'image_s': self.env['ir.attachment'].api_image(
+                    'product.public.category', 'image_1920', category.category_id.id
+                ),
+                'categories': []
+            }
+
+            # Prepare Level 2 documents
+            for category_level_2 in category.line_ids.mapped('category_id2'):
+                level_2_doc = {
+                    'id_level_2': category_level_2.id,
+                    'name': category_level_2.name,
+                    'numFound': len(category_level_2.product_tmpl_ids),
+                    'image': self.env['ir.attachment'].api_image(
+                        'product.public.category', 'image_1920', category_level_2.id
+                    ),
+                    'child_frontend_id_i': []
+                }
+
+                # Prepare Level 3 documents
+                for category_level_3 in category_level_2.child_frontend_id2:
+                    level_3_doc = {
+                        'id_level_3': category_level_3.id,
+                        'name': category_level_3.name,
+                        'numFound': len(category_level_3.product_tmpl_ids),
+                        'image': self.env['ir.attachment'].api_image(
+                            'product.public.category', 'image_1920', category_level_3.id
+                        ),
+                    }
+                    level_2_doc['child_frontend_id_i'].append(json.dumps(level_3_doc))
+
+                # Add Level 2 document to Level 1
+                document['categories'].append(json.dumps(level_2_doc))
+
+            # Sync document with Solr
+            self.solr().add([document])
+            category.update_last_update_solr()
+
+        # Commit and optimize Solr changes
+        self.solr().commit()
+        self.solr().optimize()
+