Add sync to solr on model

- promotion program
- promotion program line

4 files changed, 127 insertions, 2 deletions

diff --git a/indoteknik_custom/models/__init__.py b/indoteknik_custom/models/__init__.py
index 95defed6..2eb4b9b9 100755
--- a/indoteknik_custom/models/__init__.py
+++ b/indoteknik_custom/models/__init__.py
@@ -81,11 +81,11 @@ from . import po_sync_price
 from . import base_import_import
 from . import product_attribute
 from . import mrp_production
+from . import promotion
 from . import solr
 from . import cost_centre
 from . import account_account
 from . import account_move_line
 from . import stock_scheduler_compute
-from . import promotion
 from . import sale_orders_multi_update
 from . import quotation_so_multi_update
\ No newline at end of file
diff --git a/indoteknik_custom/models/solr/__init__.py b/indoteknik_custom/models/solr/__init__.py
index f2d13116..606c0035 100644
--- a/indoteknik_custom/models/solr/__init__.py
+++ b/indoteknik_custom/models/solr/__init__.py
@@ -8,4 +8,6 @@ from . import website_categories_homepage
 from . import x_manufactures
 from . import x_banner_banner
 from . import product_public_category
-from . import x_banner_category
\ No newline at end of file
+from . import x_banner_category
+from . import promotion_program
+from . import promotion_program_line
\ No newline at end of file
diff --git a/indoteknik_custom/models/solr/promotion_program.py b/indoteknik_custom/models/solr/promotion_program.py
new file mode 100644
index 00000000..c288c165
--- /dev/null
+++ b/indoteknik_custom/models/solr/promotion_program.py
@@ -0,0 +1,61 @@
+from odoo import models, api
+from datetime import datetime
+from pytz import timezone
+from typing import Type
+import pysolr
+
+
+class PromotionProgram(models.Model):
+    _inherit = 'promotion.program'
+    _solr_schema = 'promotion_programs'
+
+    def solr(self) -> Type[pysolr.Solr]:
+        return self.env['apache.solr'].connect(self._solr_schema)
+
+    def _create_solr_queue(self, function_name: str):
+        for rec in self:
+            self.env['apache.solr.queue'].create_unique({
+                'res_model': self._name,
+                'res_id': rec.id,
+                'function_name': function_name
+            })
+
+    def _sync_to_solr(self):
+        ir_attachment = self.env['ir.attachment']
+        solr_model = self.env['apache.solr']
+
+        for rec in self:
+            document = solr_model.get_doc(self._solr_schema, rec.id)
+            document.update({
+                'id': rec.id,
+                'name_s': rec.name,
+                'banner_s': ir_attachment.api_image(self._name, 'banner', rec.id) if rec.banner else '',
+                'keywords': [x.name for x in rec.keyword_ids],
+                'line_ids': [x.id for x in rec.program_line],
+                'start_time_s': self._time_format(rec.start_time),
+                'end_time_s': self._time_format(rec.end_time),
+                'applies_to_s': rec.applies_to,
+                'icon_s': ir_attachment.api_image(self._name, 'icon', rec.id) if rec.icon else '',
+                'icon_top_s': ir_attachment.api_image(self._name, 'icon_top', rec.id) if rec.icon_top else '',
+                'icon_bottom_s': ir_attachment.api_image(self._name, 'icon_bottom', rec.id) if rec.icon_bottom else '',
+            })
+
+            self.solr().add([document])
+
+        self.solr().commit()
+
+    def _time_format(self, object) -> str:
+        time = ''
+        tz_jakarta = timezone('Asia/Jakarta')
+        if isinstance(object, datetime):
+            time = object.astimezone(tz_jakarta).strftime("%Y-%m-%d %H:%M:%S")
+        return time
+
+    @api.model
+    def create(self, vals):
+        self._create_solr_queue('_sync_to_solr')
+        return super(PromotionProgram, self).create(vals)
+    
+    def write(self, vals):
+        self._create_solr_queue('_sync_to_solr')
+        return super(PromotionProgram, self).write(vals)
diff --git a/indoteknik_custom/models/solr/promotion_program_line.py b/indoteknik_custom/models/solr/promotion_program_line.py
new file mode 100644
index 00000000..581783d1
--- /dev/null
+++ b/indoteknik_custom/models/solr/promotion_program_line.py
@@ -0,0 +1,62 @@
+from odoo import models, api
+from typing import Type
+import pysolr, json
+
+
+class PromotionProgramLine(models.Model):
+    _inherit = 'promotion.program.line'
+    _solr_schema = 'promotion_program_lines'
+    
+    def solr(self) -> Type[pysolr.Solr]:
+        return self.env['apache.solr'].connect(self._solr_schema)
+    
+    def _create_solr_queue(self, function_name: str):
+        for rec in self:
+            self.env['apache.solr.queue'].create_unique({
+                'res_model': self._name,
+                'res_id': rec.id,
+                'function_name': function_name
+            })
+            
+    def _sync_to_solr(self):
+        solr_model = self.env['apache.solr']
+
+        for rec in self:
+            document = solr_model.get_doc(self._solr_schema, rec.id)
+            
+            products = [{
+                'product_id': x.product_id.id,
+                'qty': x.qty
+            } for x in rec.product_ids]
+            
+            free_products = [{
+                'product_id': x.product_id.id,
+                'qty': x.qty
+            } for x in rec.free_product_ids]
+            
+            document.update({
+                'id': rec.id,
+                'program_id': rec.program_id.id,
+                'name_s': rec.name,
+                'image_s': self.env['ir.attachment'].api_image(self._name, 'image', rec.id) if rec.image else '',
+                'type_s': json.dumps(rec._res_promotion_type()),
+                'package_limit_i': rec.package_limit,
+                'package_limit_user_i': rec.package_limit_user,
+                'package_limit_trx_i': rec.package_limit_trx,
+                'price_f': rec.price,
+                'products_s': json.dumps(products),
+                'free_products_s': json.dumps(free_products)
+            })
+
+            self.solr().add([document])
+
+        self.solr().commit()
+    
+    @api.model
+    def create(self, vals):
+        self._create_solr_queue('_sync_to_solr')
+        return super(PromotionProgramLine, self).create(vals)
+    
+    def write(self, vals):
+        self._create_solr_queue('_sync_to_solr')
+        return super(PromotionProgramLine, self).write(vals)
\ No newline at end of file