From 8bb101c6b29f8cc2e0e6eca9cfca93b334423fcc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=83=A1=E5=B0=A7?= <huyao@jikimo.com>
Date: Thu, 24 Apr 2025 11:08:02 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4=E6=89=93=E5=8D=B0=E4=BB=A3?=
 =?UTF-8?q?=E7=A0=81=E7=BB=93=E6=9E=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 jikimo_printing/models/__init__.py            |  1 +
 jikimo_printing/models/jikimo_printing.py     | 56 +++++++++++++++++++
 .../models/maintenance_printing.py            |  7 +--
 jikimo_printing/models/workorder_printing.py  |  4 +-
 4 files changed, 60 insertions(+), 8 deletions(-)
 create mode 100644 jikimo_printing/models/jikimo_printing.py

diff --git a/jikimo_printing/models/__init__.py b/jikimo_printing/models/__init__.py
index dd9d68dd..79d24db8 100644
--- a/jikimo_printing/models/__init__.py
+++ b/jikimo_printing/models/__init__.py
@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
+from . import jikimo_printing
 from . import maintenance_printing
 from . import workorder_printing
 
diff --git a/jikimo_printing/models/jikimo_printing.py b/jikimo_printing/models/jikimo_printing.py
new file mode 100644
index 00000000..738c90b5
--- /dev/null
+++ b/jikimo_printing/models/jikimo_printing.py
@@ -0,0 +1,56 @@
+from io import BytesIO
+import qrcode
+from reportlab.pdfgen import canvas
+from reportlab.lib.pagesizes import A4
+from PIL import Image
+from reportlab.lib.utils import ImageReader
+from odoo import models, fields, api
+
+class JikimoPrinting(models.AbstractModel):
+    _name = 'jikimo.printing'
+
+    def print_qr_code(self, data):
+        """
+            打印二维码
+        """
+        # 生成二维码
+        qr = qrcode.QRCode(version=1, box_size=10, border=5)
+        qr.add_data(data)
+        qr.make(fit=True)
+        qr_image = qr.make_image(fill_color="black", back_color="white")
+
+        # 将PIL Image转换为reportlab可用的格式
+        temp_image = BytesIO()
+        qr_image.save(temp_image, format="PNG")
+        temp_image.seek(0)
+
+        # 创建PDF
+        pdf_buffer = BytesIO()
+        c = canvas.Canvas(pdf_buffer, pagesize=A4)
+
+        # 计算位置
+        a4_width, a4_height = A4
+        qr_width = 200
+        qr_height = 200
+        x = (a4_width - qr_width) / 2
+        y = (a4_height - qr_height) / 2
+
+        # 直接从BytesIO绘制图片
+        c.drawImage(ImageReader(Image.open(temp_image)), x, y, width=qr_width, height=qr_height)
+        c.save()
+
+        # 获取PDF内容并打印
+        pdf_content = pdf_buffer.getvalue()
+        printer = self.env['printing.printer'].get_default()
+        printer.print_document(report=None, content=pdf_content, doc_format='pdf')
+
+        # 清理资源
+        pdf_buffer.close()
+        temp_image.close()
+
+    def print_pdf(self, pdf_data):
+        """
+            打印PDF
+        """
+        printer = self.env['printing.printer'].get_default()
+        printer.print_document(report=None, content = pdf_data, doc_format='pdf') 
\ No newline at end of file
diff --git a/jikimo_printing/models/maintenance_printing.py b/jikimo_printing/models/maintenance_printing.py
index befafccf..641c0af4 100644
--- a/jikimo_printing/models/maintenance_printing.py
+++ b/jikimo_printing/models/maintenance_printing.py
@@ -7,9 +7,6 @@ class MaintenancePrinting(models.Model):
 
         print('self.name========== %s' % self.name)
         self.ensure_one()
-        qr_code_data = self.qr_code_image
-        if not qr_code_data:
-            raise UserError("没有找到二维码数据。")
         # maintenance_equipment_id = self.id
         # # host = "192.168.50.110"  # 可以根据实际情况修改
         # # port = 9100  # 可以根据实际情况修改
@@ -23,9 +20,9 @@ class MaintenancePrinting(models.Model):
         # self.print_qr_code(maintenance_equipment_id, host, port)
 
         # 切换成A4打印机
-        printer = self.env['printing.printer'].get_default()
+        
         try:
-            printer.print_document(report=None, content = qr_code_data) 
+            self.env['jikimo.printing'].print_qr_code(self.id)
         except Exception as e:
             raise UserError(f"打印失败: {str(e)}")
 
diff --git a/jikimo_printing/models/workorder_printing.py b/jikimo_printing/models/workorder_printing.py
index 344e0faf..97c857ca 100644
--- a/jikimo_printing/models/workorder_printing.py
+++ b/jikimo_printing/models/workorder_printing.py
@@ -19,9 +19,7 @@ class MrpWorkorder(models.Model):
                     if pdf_data:
                         try:
                             # 执行打印
-                            printer = self.env['printing.printer'].get_default()
-                            printer.print_document(report=None, content = pdf_data, doc_format='pdf') 
-
+                            self.env['jikimo.printing'].print_pdf(pdf_data)
                             wo.production_id.product_id.is_print_program = True
                             _logger.info(f"工单 {wo.name} 的PDF已成功打印")