托盘绑定工单，托盘生成二维码

sf_route_workcenter/models/workcenter.py

@@ -0,0 +1,70 @@
+# -*- coding: utf-8 -*-
+# Part of SmartGo. See LICENSE file for full copyright and licensing details.
+import base64
+import logging
+
+import qrcode
+
+from io import BytesIO
+from odoo import api, fields, models
+from odoo.exceptions import UserError
+
+_logger = logging.getLogger(__name__)
+
+
+class Tray(models.Model):
+    _inherit = 'sf.tray'
+    _description = '托盘'
+
+    production_id = fields.Many2one('mrp.production', string='制造订单',
+                                    related='workorder_id.production_id')
+    workorder_id = fields.Many2one('mrp.workorder', string="工单")
+    qr_image = fields.Binary(string="托盘二维码", compute='compute_qr_image')
+
+    @api.depends('code')
+    def compute_qr_image(self):
+        for item in self:
+            if not item.code:
+                item.qr_image = False
+                continue
+            qr = qrcode.QRCode(
+                version=1,
+                error_correction=qrcode.constants.ERROR_CORRECT_L,
+                box_size=10,
+                border=4,
+            )
+            qr.add_data(item.code)
+            qr.make(fit=True)
+            img = qr.make_image()
+            temp = BytesIO()
+            img.save(temp, format='PNG')
+            qr_image = base64.b64encode(temp.getvalue())
+            item.qr_image = qr_image
+
+
+'''
+工单绑定托盘信息
+'''
+
+
+class MrpWorkOrder(models.Model):
+    _inherit = 'mrp.workorder'
+    _description = '工单'
+
+    tray_id = fields.Many2one('sf.tray', string="托盘")
+    tray_code = fields.Char(
+        string='托盘编码',
+        related='tray_id.code')
+    tray_state = fields.Selection(
+        string='托盘状态',
+        related='tray_id.state')
+
+
+'''
+制造订单绑定托盘信息
+'''
+
+
+class MrpProduction(models.Model):
+    _inherit = 'mrp.production'
+    _description = "制造订单"