优化批量提单

ccs_connect_tiktok/models/cc_bill_loading.py

@@ -247,46 +247,47 @@ class CcBl(models.Model):
 
     # 定义一个方法, 获取提单下的所有未同步的小包,并回传小包状态
     def callback_track(self):
-        ship_packages = self.env['cc.ship.package'].search([('bl_id', 'in', self.ids), ('is_sync', '=', False)])
-        # for package in ship_packages:
-        #     package.callback_track()
-        tt_api_obj = self.env["ao.tt.api"].sudo()
-        async def perform_requests():
-            ssl_context = ssl.create_default_context(cafile=certifi.where())
-            async with aiohttp.ClientSession(connector=aiohttp.TCPConnector(ssl=ssl_context), timeout=aiohttp.ClientTimeout(total=60)) as session:
-                tasks = []
-                for index, package in enumerate(ship_packages):
-                    if not package.is_sync and package.state and package.state.tk_code:
-                        data = package.get_callback_track_data()
-                        tasks.append(tt_api_obj.async_callback_track(session, data, package.id))
-                responses = await asyncio.gather(*tasks)
-                return responses
-
-        # 在 Odoo 中运行异步任务
-        responses = asyncio.run(perform_requests())
-        for response_item in responses:
-            response_data = response_item[0]
-            logging.info('response_data response:%s' % response_data)
-            data = response_item[1]
-            package_id = response_item[2]
-            package_order = self.env['cc.ship.package'].sudo().browse(package_id)
-            if response_data['code'] != 0:
-                package_order.is_sync = False
-                error_msg = response_data['msg']
-                request_id = response_data['requestID']
-                code = response_data['code']
-                self.env['ao.tt.api.log'].sudo().create_api_log(
-                    package_order.tracking_no or '', '小包状态轨迹回传:' + error_msg, data, code, request_id,
-                    source='推出')
-            else:
-                # 回传成功
-                package_order.is_sync = True
-                self.env['cc.ship.package.sync.log'].sudo().create_sync_log(
-                    package_order.id, 'Tiktok', package_order.state.tk_code, package_order.state_explain,
-                    package_order.process_time.strftime('%Y-%m-%d %H:%M:%S'))
-                request_id = response_data['requestID']
-                self.env['ao.tt.api.log'].sudo().create_api_log(
-                    package_order.tracking_no or '', '', data, 0, request_id, source='推出')
+        for item in self:
+            ship_packages = self.env['cc.ship.package'].search([('bl_id', '=', item.id), ('is_sync', '=', False)])
+            # for package in ship_packages:
+            #     package.callback_track()
+            tt_api_obj = self.env["ao.tt.api"].sudo()
+            async def perform_requests():
+                ssl_context = ssl.create_default_context(cafile=certifi.where())
+                async with aiohttp.ClientSession(connector=aiohttp.TCPConnector(ssl=ssl_context), timeout=aiohttp.ClientTimeout(total=60)) as session:
+                    tasks = []
+                    for index, package in enumerate(ship_packages):
+                        if not package.is_sync and package.state and package.state.tk_code:
+                            data = package.get_callback_track_data()
+                            tasks.append(tt_api_obj.async_callback_track(session, data, package.id))
+                    responses = await asyncio.gather(*tasks)
+                    return responses
+
+            # 在 Odoo 中运行异步任务
+            responses = asyncio.run(perform_requests())
+            for response_item in responses:
+                response_data = response_item[0]
+                logging.info('response_data response:%s' % response_data)
+                data = response_item[1]
+                package_id = response_item[2]
+                package_order = self.env['cc.ship.package'].sudo().browse(package_id)
+                if response_data['code'] != 0:
+                    package_order.is_sync = False
+                    error_msg = response_data['msg']
+                    request_id = response_data['requestID']
+                    code = response_data['code']
+                    self.env['ao.tt.api.log'].sudo().create_api_log(
+                        package_order.tracking_no or '', '小包状态轨迹回传:' + error_msg, data, code, request_id,
+                        source='推出')
+                else:
+                    # 回传成功
+                    package_order.is_sync = True
+                    self.env['cc.ship.package.sync.log'].sudo().create_sync_log(
+                        package_order.id, 'Tiktok', package_order.state.tk_code, package_order.state_explain,
+                        package_order.process_time.strftime('%Y-%m-%d %H:%M:%S'))
+                    request_id = response_data['requestID']
+                    self.env['ao.tt.api.log'].sudo().create_api_log(
+                        package_order.tracking_no or '', '', data, 0, request_id, source='推出')
 
     def deal_ship_package_state(self):
         for item in self: