销售和排程添加消息推送

sf_message/models/sf_message_sale.py

@@ -1,11 +1,43 @@
+# -*- coding: utf-8 -*-
 from odoo import models, fields, api, _
 
 
 class SFMessageSale(models.Model):
     _name = 'sale.order'
-    _inherit = ['sale.order', 'sf.message.template']
+    _inherit = ['sale.order', 'jikimo.message.dispatch']
 
-    # def create(self):
-    #     res = super(SFMessageSale, self).create()
-    #     if res is True:
+    def create(self, vals_list):
+        res = super(SFMessageSale, self).create(vals_list)
+        if res:
+            try:
+                res.add_queue('待接单')
+            except Exception as e:
+                logging.info('add_queue error:%s' % e)
+        return res
 
+    # 确认接单
+    def action_confirm(self):
+        res = super(SFMessageSale, self).action_confirm()
+        if res is True:
+            try:
+                self.add_queue('确认接单')
+            except Exception as e:
+                logging.info('add_queue error:%s' % e)
+        return res
+
+    # 继承并重写jikimo.message.dispatch的_get_message()
+    def _get_message(self, message_queue_ids):
+        if message_queue_ids.message_template_id.bussiness_node_id.name == '确认接单':
+            # sale_order = self.env['sale.order'].search([('id', '=', message_queue_ids.model.res_id)])
+            sale_order_line = self.env['sale.order.line'].search([('order_id', '=', int(message_queue_ids.res_id))])
+            if len(sale_order_line) == 1:
+                product = sale_order_line[0].product_id.name
+            elif len(sale_order_line) > 1:
+                product = '%s...' % sale_order_line[0].product_id.name
+        res = super(SFMessageSale, self)._get_message(message_queue_ids)
+        if res:
+            try:
+                res.add_queue('待排程')
+            except Exception as e:
+                logging.info('_get_message error:%s' % e)
+        return res