[fix] 修改问题

jzjx_project/controllers/inherit_product_task.py

+from urllib import parse
+
 from odoo import models, fields, api, http, SUPERUSER_ID, _
 from odoo import models, fields, api, http, SUPERUSER_ID, _
 from odoo.addons.roke_workstation_api.controllers.product_task import RokeWorkstationProductionTask
 from odoo.addons.roke_workstation_api.controllers.product_task import RokeWorkstationProductionTask
 from odoo.addons.roke_workstation_api.controllers.product_task import RokeWorkstationTaskDispatch
 from odoo.addons.roke_workstation_api.controllers.product_task import RokeWorkstationTaskDispatch
@@ -183,4 +185,56 @@ class InheritRokeWorkstationTaskDispatch(RokeWorkstationTaskDispatch):
                 break
                 break
             _self.jsonrequest["work_order_id"] = new_work_order_id
             _self.jsonrequest["work_order_id"] = new_work_order_id
         res = super(InheritRokeWorkstationTaskDispatch, self).product_task_get_document_list()
         res = super(InheritRokeWorkstationTaskDispatch, self).product_task_get_document_list()
+        if work_order.type == "返修":
+            scrap_order_line = _self.env["roke.repair.order.line"].sudo().search([
+                ("repair_work_order_id", "=", work_order.id)
+            ], limit=1)
+            record_id = scrap_order_line.wr_id
+        elif work_order.type == "补件":
+            scrap_order_line = _self.env["roke.scrap.order.line"].sudo().search([("scrap_work_order_ids", "=", work_order.id)],
+                                                                         limit=1)
+            record_id = scrap_order_line.wr_id
+        else:
+            return res
+        attachment_ids = _self.env['ir.attachment'].sudo().search([
+            ("res_model", "=", "roke.work.record"), ("res_id", "=", record_id.id)
+        ])
+        base_url = _self.env['ir.config_parameter'].sudo().get_param('web.base.url')
+        for attachment in attachment_ids:
+            if not attachment.access_token:
+                attachment.generate_access_token()
+            name = f"{attachment.name}"
+            if attachment.mimetype == "application/pdf":
+                # pdf 预览
+                content_url = parse.quote(
+                    "/web/content/%s?access_token=%s" % (str(attachment.id), attachment.sudo().access_token))
+                url = f"{base_url}/web/static/lib/pdfjs/web/viewer.html?file={content_url}"
+                res["data"].append({
+                    "id": attachment.id,
+                    "name": name,
+                    "mimetype": attachment.mimetype,
+                    "type": "url",
+                    "data": url
+                })
+            else:
+                if attachment.index_content == 'image':
+                    url = f"{base_url}/web/image/{attachment.id}?access_token={attachment.access_token}" \
+                          f"&unique={str(fields.Datetime.now().timestamp())}"
+                    res["data"].append({
+                        "id": attachment.id,
+                        "name": name,
+                        "mimetype": attachment.mimetype,
+                        "type": "url",
+                        "data": url
+                    })
+                else:
+                    url = f"{base_url}/web/content/{attachment.id}?access_token={attachment.access_token}" \
+                          f"&unique={str(fields.Datetime.now().timestamp())}"
+                    res["data"].append({
+                        "id": attachment.id,
+                        "name": name,
+                        "mimetype": attachment.mimetype,
+                        "type": "url",
+                        "data": url,
+                    })
         return res
         return res