还原

sf_mrs_connect/models/sync_common.py

@@ -8,6 +8,9 @@ import requests
 from odoo import models
 from odoo.exceptions import ValidationError
 from odoo.addons.sf_base.commons.common import Common
+from odoo.addons.sf_mrs_connect.models.common import Common
+
+
 
 _logger = logging.getLogger(__name__)
 
@@ -3242,28 +3245,22 @@ class SyncfixtureMaterialsBasicParameters(models.Model):
     _description = 'Redis 优先同步夹具基本参数'
 
     def sync_all_fixture_materials_basic_parameters(self):
+        # 你已有的批量同步方法不变
         rc = RedisClient()
         key = 'mrs:fixture_param_all_list'
         all_list = rc.get_json(key)
-
-        # ✅ Redis 无数据时，请求 MRS 端刷新 Redis 缓存
         if not all_list:
             config = self.env['res.config.settings'].get_values()
             headers = Common.get_headers(config['token'], config['sf_secret_key'])
             refresh_url = config['mrs_url'] + '/api/refresh_redis/fixture_param'
-
             try:
                 res = requests.post(refresh_url, headers=headers, json={}, timeout=10)
                 res.raise_for_status()
             except Exception as e:
                 raise ValidationError(f"Redis 无数据，MRS 缓存刷新失败: {str(e)}")
-
-            # 再次尝试从 Redis 获取
             all_list = rc.get_json(key)
             if not all_list:
                 raise ValidationError("Redis 刷新后仍无数据，无法同步夹具基本参数")
-
-        # ✅ 同步函数作为局部函数定义在方法体内部
         def _sync_list(param_list, material_name):
             for item in param_list or []:
                 if not item or not item.get('code'):
@@ -3274,8 +3271,6 @@ class SyncfixtureMaterialsBasicParameters(models.Model):
                     record.write(vals)
                 else:
                     self.create(vals)
-
-        # ✅ 各类夹具参数同步调用
         _sync_list(all_list.get('zero_chuck_all_list'), '零点卡盘')
         _sync_list(all_list.get('zero_tray_all_list'), '零点托盘')
         _sync_list(all_list.get('pneumatic_fixture_all_list'), '气动夹具')
@@ -3286,6 +3281,52 @@ class SyncfixtureMaterialsBasicParameters(models.Model):
         _sync_list(all_list.get('air_tray_all_list'), '气吹托盘')
         _sync_list(all_list.get('magnet_tray_all_list'), '磁吸托盘')
 
+    def sync_from_mrs(self, code):
+        rc = RedisClient()
+        key = 'mrs:fixture_param_all_list'
+        all_list = rc.get_json(key) or {}
+        if not all_list:
+            # Redis没数据，刷新一次
+            config = self.env['res.config.settings'].get_values()
+            headers = Common.get_headers(config['token'], config['sf_secret_key'])
+            refresh_url = config['mrs_url'] + '/api/refresh_redis/fixture_param'
+            try:
+                res = requests.post(refresh_url, headers=headers, json={}, timeout=10)
+                res.raise_for_status()
+            except Exception as e:
+                raise ValidationError(f"Redis 无数据，MRS 缓存刷新失败: {str(e)}")
+            all_list = rc.get_json(key) or {}
+            if not all_list:
+                raise ValidationError("刷新后仍无 Redis 数据，无法同步夹具基本参数")
+
+        # 遍历所有分类列表查找code匹配项
+        for material_name, param_list in all_list.items():
+            for item in param_list or []:
+                if item.get('code') == code:
+                    vals = self._get_basic_parameters_list(item, material_name)
+                    record = self.search([('code', '=', code)], limit=1)
+                    if record:
+                        record.write(vals)
+                    else:
+                        self.create(vals)
+                    return True
+        return False
+
+    def _get_basic_parameters_list(self, item, material_name):
+        # 请根据实际字段完善，这里给个示范
+        return {
+            'code': item.get('code'),
+            'length': item.get('length'),
+            'width': item.get('width'),
+            'height': item.get('height'),
+            'chucking_power_max': item.get('chucking_power_max'),
+            'type_of_drive': item.get('type_of_drive'),
+            'locking_method': item.get('locking_method'),
+            'max_adsorp_force': item.get('max_adsorp_force'),
+            # 你可以加上 material_name 做记录或其它业务字段
+            # 'material_name': material_name,
+        }
+
 
 class SyncFixtureModel(models.Model):
     _inherit = 'sf.fixture.model'
@@ -3295,39 +3336,32 @@ class SyncFixtureModel(models.Model):
         rc = RedisClient()
         key = 'mrs:fixture_model_all_list'
         all_list = rc.get_json(key)
-
-        # ✅ Redis 没数据时，请求 MRS 接口刷新
         if not all_list:
             config = self.env['res.config.settings'].get_values()
             headers = Common.get_headers(config['token'], config['sf_secret_key'])
             refresh_url = config['mrs_url'] + '/api/refresh_redis/fixture_model'
-
             try:
                 res = requests.post(refresh_url, headers=headers, json={}, timeout=10)
                 res.raise_for_status()
             except Exception as e:
                 raise ValidationError(f"Redis 无数据，MRS 刷新失败: {str(e)}")
-
-            # 再次尝试获取 Redis
             all_list = rc.get_json(key)
             if not all_list:
                 raise ValidationError("刷新后仍无 Redis 数据，无法同步夹具型号")
 
-        # ✅ 开始同步夹具型号
         for item in all_list:
             if not item or not item.get('code'):
                 continue
-
             record = self.search([('code', '=', item['code'])], limit=1)
             vals = {
                 'name': item['name'],
                 'code': item['code'],
                 'fixture_material_id': self.env['sf.fixture.material']
-                    .search([('code', '=', item['fixture_material_code'])], limit=1).id,
+                    .search([('code', '=', item.get('fixture_material_code'))], limit=1).id or False,
                 'multi_mounting_type_id': self.env['sf.multi_mounting.type']
-                    .search([('code', '=', item['multi_mounting_type_code'])], limit=1).id,
+                    .search([('code', '=', item.get('multi_mounting_type_code'))], limit=1).id or False,
                 'brand_id': self.env['sf.machine.brand']
-                    .search([('code', '=', item['brand_code'])], limit=1).id,
+                    .search([('code', '=', item.get('brand_code'))], limit=1).id or False,
                 'model_file': base64.b64decode(item['model_file']) if item.get('model_file') else False,
                 'status': item['status'],
                 'active': item['active'],
@@ -3335,4 +3369,44 @@ class SyncFixtureModel(models.Model):
             if record:
                 record.write(vals)
             else:
-                self.create(vals)
+                self.create(vals)
+
+    def sync_from_mrs(self, code):
+        rc = RedisClient()
+        key = 'mrs:fixture_model_all_list'
+        all_list = rc.get_json(key) or []
+        if not all_list:
+            config = self.env['res.config.settings'].get_values()
+            headers = Common.get_headers(config['token'], config['sf_secret_key'])
+            refresh_url = config['mrs_url'] + '/api/refresh_redis/fixture_model'
+            try:
+                res = requests.post(refresh_url, headers=headers, json={}, timeout=10)
+                res.raise_for_status()
+            except Exception as e:
+                raise ValidationError(f"Redis 无数据，MRS 刷新失败: {str(e)}")
+            all_list = rc.get_json(key) or []
+            if not all_list:
+                raise ValidationError("刷新后仍无 Redis 数据，无法同步夹具型号")
+
+        for item in all_list:
+            if item.get('code') == code:
+                record = self.search([('code', '=', code)], limit=1)
+                vals = {
+                    'name': item['name'],
+                    'code': item['code'],
+                    'fixture_material_id': self.env['sf.fixture.material']
+                        .search([('code', '=', item.get('fixture_material_code'))], limit=1).id or False,
+                    'multi_mounting_type_id': self.env['sf.multi_mounting.type']
+                        .search([('code', '=', item.get('multi_mounting_type_code'))], limit=1).id or False,
+                    'brand_id': self.env['sf.machine.brand']
+                        .search([('code', '=', item.get('brand_code'))], limit=1).id or False,
+                    'model_file': base64.b64decode(item['model_file']) if item.get('model_file') else False,
+                    'status': item['status'],
+                    'active': item['active'],
+                }
+                if record:
+                    record.write(vals)
+                else:
+                    self.create(vals)
+                return True
+        return False