导入媒体增加factoryCode

haoban-manage3-wx/src/main/java/com/gic/haoban/manage/web/controller/content/ContentAIController.java

@@ -139,6 +139,8 @@ public class ContentAIController {
     @ResponseBody
     public RestResponse<String> importMaterial(@RequestBody CmeImportMaterialQO materialQO){
         CmeImportMaterial cmeImportMaterial = EntityUtil.changeEntityNew(CmeImportMaterial.class, materialQO);
+        EnterpriseDTO enterpriseDTO = enterpriseService.getEnterpriseById(materialQO.getEnterpriseId());
+        cmeImportMaterial.setFactoryCode(enterpriseDTO.getFactoryCode());
         try {
             String materialId = CmeUtil.importMaterial(cmeImportMaterial);
             return RestResponse.successResult(materialId);

haoban-manage3-wx/src/main/java/com/gic/haoban/manage/web/qo/content/ai/CmeImportMaterialQO.java

@@ -15,6 +15,11 @@ public class CmeImportMaterialQO {
      */
     private String mediaKey;
 
+    /**
+     * 企业id
+     */
+    private String enterpriseId;
+
     public String getName() {
         return name;
     }
@@ -30,4 +35,12 @@ public class CmeImportMaterialQO {
     public void setMediaKey(String mediaKey) {
         this.mediaKey = mediaKey;
     }
+
+    public String getEnterpriseId() {
+        return enterpriseId;
+    }
+
+    public void setEnterpriseId(String enterpriseId) {
+        this.enterpriseId = enterpriseId;
+    }
 }