同步机床类型对象

sf_mrs_connect/models/sync_common.py

@@ -1086,3 +1086,64 @@ class sfProductionProcessParameter(models.Model):
                         })
         else:
             raise ValidationError("认证未通过")
+
+class MachineToolCategory(models.Model):
+    _inherit = 'sf.machine_tool.category'
+    _description = '机床类型'
+    url = '/api/machine_tool_category/list'
+
+    # 定时同步机床类型
+    def sync_machine_tool_category(self):
+        sf_sync_config = self.env['res.config.settings'].get_values()
+        token = sf_sync_config['token']
+        sf_secret_key = sf_sync_config['sf_secret_key']
+        headers = Common.get_headers(self, token, sf_secret_key)
+
+        strUrl = sf_sync_config['sf_url'] + self.url
+        r = requests.post(strUrl, json={}, data=None, headers=headers)
+        r = r.json()
+        result = json.loads(r['result'])
+        if result['status'] == 1:
+            for item in result['machine_tool_category_yesterday_list']:
+                brand = self.env['sf.machine_tool.category'].search(
+                    [("code", '=', item['code'])])
+                if brand:
+                    brand.name = item['name'],
+                    brand.code = item['code'],
+                    brand.category = item['category'],
+                    brand.remark = item['remark']
+
+                else:
+                    brand.name = item['name'],
+                    brand.code = item['code'],
+                    brand.category = item['category'],
+                    brand.remark = item['remark']
+        else:
+            raise ValidationError("认证未通过")
+
+    # 同步所有机床类型
+    def sync_all_machine_tool_category(self):
+        sf_sync_config = self.env['res.config.settings'].get_values()
+        token = sf_sync_config['token']
+        sf_secret_key = sf_sync_config['sf_secret_key']
+        headers = Common.get_headers(self, token, sf_secret_key)
+
+        strUrl = sf_sync_config['sf_url'] + self.url
+        r = requests.post(strUrl, json={}, data=None, headers=headers)
+        r = r.json()
+        result = json.loads(r['result'])
+        if result['status'] == 1:
+            for item in result['machine_tool_category_all_list']:
+                brand = self.env['sf.machine_tool.category'].search(
+                    [("code", '=', item['code'])])
+                if not brand:
+                    self.env['sf.machine_tool.category'].create({
+                        "name": item['name'],
+                        "code": item['code'],
+                        "category": item['category'],
+                        "remark": item['remark'],
+
+                    })
+        else:
+            raise ValidationError("认证未通过")
+